def do_dot_test(self, expected, **kwargs):
graph.Process._id_gen = 0
tmpdir = Path.tempdir(prefix='rpz_testgraph_')
target = tmpdir / 'graph.dot'
try:
graph.generate(target, self._trace / 'config.yml',
self._trace / 'trace.sqlite3', **kwargs)
if expected is False:
self.fail("DOT generation didn't fail as expected")
with target.open('r') as fp:
self.assertEqual(expected, fp.read())
except UsageError:
if expected is not False:
raise
finally:
tmpdir.rmtree()
def do_dot_test(self, expected, **kwargs):
graph.Process._id_gen = 0
fd, target = Path.tempfile(prefix='rpz_testgraph_', suffix='.dot')
os.close(fd)
try:
graph.generate(target, self._trace / 'config.yml',
self._trace / 'trace.sqlite3', **kwargs)
if expected is False:
self.fail("DOT generation didn't fail as expected")
with target.open('r') as fp:
self.assertEqual(expected, fp.read())
except UsageError:
if expected is not False:
raise
finally:
target.remove()
def do_dot_test(self, expected, **kwargs):
graph.Process._id_gen = 0
tmpdir = Path.tempdir(prefix='rpz_testgraph_')
target = tmpdir / 'graph.dot'
try:
graph.generate(target,
self._trace / 'config.yml',
self._trace / 'trace.sqlite3',
**kwargs)
if expected is False:
self.fail("DOT generation didn't fail as expected")
with target.open('r') as fp:
self.assertEqual(expected, fp.read())
except UsageError:
if expected is not False:
raise
finally:
tmpdir.rmtree()
def do_dot_test(self, expected, **kwargs):
graph.Process._id_gen = 0
fd, target = Path.tempfile(prefix='rpz_testgraph_', suffix='.dot')
os.close(fd)
try:
graph.generate(target,
self._trace / 'config.yml',
self._trace / 'trace.sqlite3',
**kwargs)
if expected is False:
self.fail("DOT generation didn't fail as expected")
with target.open('r') as fp:
self.assertEqual(expected, fp.read())
except UsageError:
if expected is not False:
raise
finally:
target.remove()
def do_json_test(self, expected, **kwargs):
graph.Process._id_gen = 0
tmpdir = Path.tempdir(prefix='rpz_testgraph_')
target = tmpdir / 'graph.json'
try:
graph.generate(target,
self._trace / 'config.yml',
self._trace / 'trace.sqlite3',
graph_format='json', **kwargs)
if expected is False:
self.fail("JSON generation didn't fail as expected")
with target.open('r', encoding='utf-8') as fp:
obj = json.load(fp)
self.assertEqual(expected, obj)
except SystemExit:
if expected is not False:
raise
finally:
tmpdir.rmtree()
def do_json_test(self, expected, **kwargs):
graph.Process._id_gen = 0
tmpdir = Path.tempdir(prefix='rpz_testgraph_')
target = tmpdir / 'graph.json'
try:
graph.generate(target,
self._trace / 'config.yml',
self._trace / 'trace.sqlite3',
graph_format='json',
**kwargs)
if expected is False:
self.fail("JSON generation didn't fail as expected")
with target.open('r', encoding='utf-8') as fp:
obj = json.load(fp)
self.assertEqual(expected, obj)
except SystemExit:
if expected is not False:
raise
finally:
tmpdir.rmtree()