def test_pretty_results(fake_results):
"""Ensure pretty results is calling everything it should."""
with patch.object(formatting, "pretty_header") as patched_header:
with patch.object(formatting, "_pretty_result") as patched_result:
with patch.object(formatting, "write") as patched_write:
formatting.pretty_results(fake_results)
assert patched_header.called
assert patched_result.called
assert patched_write.called
def test_pretty_results(fake_results):
"""Ensure pretty results is calling everything it should."""
with patch.object(formatting, "pretty_header") as patched_header:
with patch.object(formatting, "_pretty_result") as patched_result:
with patch.object(formatting, "write") as patched_write:
formatting.pretty_results(fake_results)
assert patched_header.called
assert patched_result.called
assert patched_write.called
def cmdline_exit(results, options):
"""A buffer for selecting the correct output function and exiting.
Args::
results: the results dictionary from Bladerunner.run
options: the options dictionary, uses 'style' and 'stacked' keys
"""
if options.get("stacked"):
stacked_results(results, options)
elif options["style"] < 0 or options["style"] > 3:
csv_results(results, options)
else:
pretty_results(results, options)
raise SystemExit
def cmdline_exit(results, options):
"""A buffer for selecting the correct output function and exiting.
Args::
results: the results dictionary from Bladerunner.run
options: the options dictionary, uses 'style' key only
"""
if options["style"] < 0 or options["style"] > 3:
csv_results(results, options)
else:
try:
pretty_results(results, options)
except UnicodeEncodeError:
csv_results(results, options)
raise SystemExit
def cmdline_exit(results, options):
"""A buffer for selecting the correct output function and exiting.
Args::
results: the results dictionary from Bladerunner.run
options: the options dictionary, uses 'style' and 'stacked' keys
"""
if options.get("stacked"):
stacked_results(results, options)
elif options["style"] < 0 or options["style"] > 3:
csv_results(results, options)
else:
pretty_results(results, options)
raise SystemExit
