-
Notifications
You must be signed in to change notification settings - Fork 1.9k
Expand file tree
/
Copy pathtest_io_error.py
More file actions
45 lines (37 loc) · 1.15 KB
/
test_io_error.py
File metadata and controls
45 lines (37 loc) · 1.15 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
import sys
import os.path
import shutil
import unittest
from contextlib import contextmanager
import semmle.populator
from tests import test_utils
import subprocess
if sys.version_info < (3,0):
from StringIO import StringIO
else:
from io import StringIO
ALL_ACCESS = int("777", base=8)
@contextmanager
def discard_output():
new_out, new_err = StringIO(), StringIO()
old_out, old_err = sys.stdout, sys.stderr
try:
sys.stdout, sys.stderr = new_out, new_err
yield
finally:
sys.stdout, sys.stderr = old_out, old_err
class SingleThreadedTest(test_utils.ExtractorTest):
def __init__(self, name):
super(SingleThreadedTest, self).__init__(name)
def test_ioerror(self):
if os.name == "nt":
return
try:
os.chmod(self.trap_path, 0)
with discard_output():
try:
self.run_extractor("-z1", "-y", "package.sub", "mod1", "package.x", "package.sub.a")
except subprocess.CalledProcessError as ex:
self.assertEqual(ex.returncode, 1)
finally:
os.chmod(self.trap_path, ALL_ACCESS)