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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
|
import copy
import os
import re
import subprocess
import tempfile
# We test the compiler indirectly, rather than reaching into the ipdl/
# module, to make the testing framework as general as possible.
class IPDLCompile:
def __init__(self, specfilename, ipdlargv=["python", "ipdl.py"]):
self.argv = copy.deepcopy(ipdlargv)
self.specfilename = specfilename
self.stdout = None
self.stderr = None
self.returncode = None
def run(self):
"""Run |self.specfilename| through the IPDL compiler."""
assert self.returncode is None
tmpoutdir = tempfile.mkdtemp(prefix="ipdl_unit_test")
try:
self.argv.extend(["-d", tmpoutdir, self.specfilename])
proc = subprocess.Popen(
args=self.argv,
stdout=subprocess.PIPE,
stderr=subprocess.PIPE,
universal_newlines=True,
)
self.stdout, self.stderr = proc.communicate()
self.returncode = proc.returncode
assert self.returncode is not None
finally:
for root, dirs, files in os.walk(tmpoutdir, topdown=0):
for name in files:
os.remove(os.path.join(root, name))
for name in dirs:
os.rmdir(os.path.join(root, name))
os.rmdir(tmpoutdir)
if proc.returncode is None:
proc.kill()
def completed(self):
return (
self.returncode is not None
and isinstance(self.stdout, str)
and isinstance(self.stderr, str)
)
def error(self, expectedError):
"""Return True iff compiling self.specstring resulted in an
IPDL compiler error."""
assert self.completed()
errorRe = re.compile(re.escape(expectedError))
return None is not re.search(errorRe, self.stderr)
def exception(self):
"""Return True iff compiling self.specstring resulted in a Python
exception being raised."""
assert self.completed()
return None is not re.search(r"Traceback (most recent call last):", self.stderr)
def ok(self):
"""Return True iff compiling self.specstring was successful."""
assert self.completed()
return (
not self.exception() and not self.error("error:") and (0 == self.returncode)
)
|
