def test_inspect_scan_error(self, mock_scan):
"""Test scan flow with mocked manager and failure."""
scanner = InspectTaskRunner(self.scan_job, self.inspect_scan_task,
self.inspect_results)
scan_task_status = scanner.run()
mock_scan.assert_called_with()
self.assertEqual(scan_task_status, ScanTask.FAILED)
def test_inspect_scan_error(self, mock_scan):
"""Test scan flow with mocked manager and failure."""
scanner = InspectTaskRunner(
self.scan_job, self.scan_task)
scan_task_status = scanner.run(Value('i', ScanJob.JOB_RUN))
mock_scan.assert_called_with(self.host_list)
self.assertEqual(scan_task_status[1], ScanTask.FAILED)
def test_inspect_scan_fail_no_facts(self, mock_run):
"""Test running a inspect scan with mocked connection."""
expected = ([('1.2.3.4', {'name': 'cred1'})], [])
mock_run.return_value = expected
with requests_mock.Mocker() as mocker:
mocker.post(self.fact_endpoint, status_code=201, json={'id': 1})
scanner = InspectTaskRunner(self.scan_job, self.scan_task)
scan_task_status = scanner.run()
mock_run.assert_called_with(ANY)
self.assertEqual(scan_task_status[1], ScanTask.FAILED)
def test_inspect_scan_fail_no_facts(self, mock_run):
"""Test running a inspect scan with mocked connection."""
mock_run.return_value.status = 'successful'
with requests_mock.Mocker() as mocker:
mocker.post(self.fact_endpoint, status_code=201, json={'id': 1})
scanner = InspectTaskRunner(self.scan_job, self.scan_task)
scanner.connect_scan_task = self.connect_scan_task
scan_task_status = scanner.run(Value('i', ScanJob.JOB_RUN))
mock_run.assert_called()
self.assertEqual(scan_task_status[1], ScanTask.FAILED)
def test_no_reachable_host(self, discovery):
"""Test no reachable host."""
discovery.return_value = [], [], [], []
scanner = InspectTaskRunner(self.scan_job, self.scan_task)
scan_task_status = scanner.run(Value('i', ScanJob.JOB_RUN))
self.assertEqual(scan_task_status[1], ScanTask.FAILED)
def test_run_manager_interupt(self):
"""Test manager interupt for inspect run method."""
scanner = InspectTaskRunner(self.scan_job, self.scan_task)
terminate = Value('i', ScanJob.JOB_TERMINATE_CANCEL)
scan_task_status = scanner.run(terminate)
self.assertEqual(scan_task_status[1], ScanTask.CANCELED)
