def test_no_errors_are_raised___completed_is_incremented(
self, input_dir, output_dir, settings, do_il, do_ri,
initial_complete, initial_failed):
client = OasisAPIClient('http://localhost:8001')
client.upload_inputs_from_directory = Mock(
return_value='input_location')
client.run_analysis_and_poll = Mock()
counter = Counter({
'completed': initial_complete,
'failed': initial_failed,
})
TestModelApiCmd().run_analysis(
(client, input_dir, output_dir, settings, do_il, do_ri, counter,
[], 1))
self.assertEqual(initial_complete + 1, counter['completed'])
self.assertEqual(initial_failed, counter['failed'])
client.upload_inputs_from_directory.assert_called_once_with(
input_dir,
bin_directory=ANY,
do_il=do_il,
do_ri=do_ri,
do_build=True)
client.run_analysis_and_poll.assert_called_once_with(
settings, 'input_location', output_dir)
def test_run_and_poll_raises_an_error___failed_counter_is_incremented(self, input_dir, output_dir, settings, do_il, initial_complete, initial_failed):
client = OasisAPIClient('http://localhost:8001')
client.upload_inputs_from_directory = Mock()
client.run_analysis_and_poll = Mock(side_effect=OasisException())
counter = Counter({
'completed': initial_complete,
'failed': initial_failed,
})
TestModelApiCmd().run_analysis((client, input_dir, output_dir, settings, do_il, counter))
self.assertEqual(initial_complete, counter['completed'])
self.assertEqual(initial_failed + 1, counter['failed'])