def test_command_line(self):
output = "/tmp"
args = ["analysis.py", "-o", output]
mock_init = mock.Mock()
mock_init.return_value = None
with mock.patch('tlsfuzzer.analysis.Analysis.generate_report') as mock_report:
with mock.patch('tlsfuzzer.analysis.Analysis.__init__', mock_init):
with mock.patch("sys.argv", args):
main()
mock_report.assert_called_once()
mock_init.assert_called_once_with(output, True, True, True)
def test_call_with_no_plots(self):
output = "/tmp"
args = ["analysis.py", "-o", output, "--no-ecdf-plot",
"--no-scatter-plot", "--no-conf-interval-plot"]
mock_init = mock.Mock()
mock_init.return_value = None
with mock.patch('tlsfuzzer.analysis.Analysis.generate_report') as mock_report:
with mock.patch('tlsfuzzer.analysis.Analysis.__init__', mock_init):
with mock.patch("sys.argv", args):
main()
mock_report.assert_called_once()
mock_init.assert_called_once_with(
output, False, False, False)