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)