def test_report_openvas_scan_status(self, mock_db, mock_sort_host_finished, mock_set_scan_progress_batch): w = DummyDaemon() mock_set_scan_progress_batch.return_value = None mock_sort_host_finished.return_value = None mock_db.get_scan_status.return_value = [ '192.168.0.1/15/1000', '192.168.0.2/15/0', '192.168.0.3/15/-1', '192.168.0.4/1500/1500', ] target_element = w.create_xml_target() targets = OspRequest.process_target_element(target_element) w.create_scan('123-456', targets, None, []) w.report_openvas_scan_status(mock_db, '123-456') mock_set_scan_progress_batch.assert_called_with( '123-456', host_progress={ '192.168.0.1': 1, '192.168.0.3': -1, '192.168.0.4': 100, }, ) mock_sort_host_finished.assert_called_with( '123-456', ['192.168.0.3', '192.168.0.4'])
def test_get_openvas_result_host_deny( self, mock_add_scan_error_to_list, MockDBClass ): w = DummyDaemon() target_element = w.create_xml_target() targets = OspRequest.process_target_element(target_element) w.create_scan('123-456', targets, None, []) results = [ "ERRMSG|||127.0.0.1|||localhost|||||||||Host access denied.", ] MockDBClass.get_result.return_value = results mock_add_scan_error_to_list.return_value = None w.report_openvas_results(MockDBClass, '123-456') mock_add_scan_error_to_list.assert_called_with( host='127.0.0.1', hostname='localhost', name='', port='', test_id='', uri='', value='Host access denied.', )
def test_get_openvas_result(self, mock_add_scan_log_to_list, MockDBClass): w = DummyDaemon() target_element = w.create_xml_target() targets = OspRequest.process_target_element(target_element) w.create_scan('123-456', targets, None, []) results = [ "LOG|||192.168.0.1|||localhost|||general/Host_Details||||||Host" " dead", ] MockDBClass.get_result.return_value = results mock_add_scan_log_to_list.return_value = None w.report_openvas_results(MockDBClass, '123-456') mock_add_scan_log_to_list.assert_called_with( host='192.168.0.1', hostname='localhost', name='', port='general/Host_Details', qod='', test_id='', uri='', value='Host dead', )
def test_update_progress(self, mock_ospd, mock_nvti, mock_db): msg = '0/-1' w = DummyDaemon(mock_nvti, mock_db) target_list = w.create_xml_target() targets = w.process_targets_element(target_list) w.create_scan('123-456', targets, None, []) mock_ospd.return_value = None w.update_progress('123-456', 'localhost', 'localhost', msg) mock_ospd.assert_called_with('123-456', 'localhost', 'localhost', 100)
def test_get_openvas_timestamp_scan_host_end(self, mock_nvti, mock_db): mock_db.get_host_scan_scan_end_time.return_value = '12345' w = DummyDaemon(mock_nvti, mock_db) target_list = w.create_xml_target() targets = w.process_targets_element(target_list) w.create_scan('123-456', targets, None, []) w.get_openvas_timestamp_scan_host('123-456', '192.168.0.1') for result in w.scan_collection.results_iterator('123-456', False): self.assertEqual(result.get('value'), '12345')
def test_update_progress(self, mock_set_scan_host_progress): w = DummyDaemon() mock_set_scan_host_progress.return_value = None msg = '0/-1' target_element = w.create_xml_target() targets = OspRequest.process_target_element(target_element) w.create_scan('123-456', targets, None, []) w.update_progress('123-456', 'localhost', msg) mock_set_scan_host_progress.assert_called_with('123-456', 'localhost', 100)
def test_get_openvas_result_dead_hosts(self, MockDBClass): w = DummyDaemon() target_element = w.create_xml_target() targets = OspRequest.process_target_element(target_element) w.create_scan('123-456', targets, None, []) results = ["DEADHOST||| ||| ||| |||4", None] mock_db = MockDBClass.return_value mock_db.get_result.side_effect = results w.scan_collection.set_amount_dead_hosts = MagicMock() w.report_openvas_results(mock_db, '123-456', 'localhost') w.scan_collection.set_amount_dead_hosts.assert_called_with( '123-456', total_dead=4, )
def test_result_without_vt_oid(self, mock_add_scan_alarm_to_list, MockDBClass): w = DummyDaemon() logging.Logger.warning = Mock() target_element = w.create_xml_target() targets = OspRequest.process_target_element(target_element) w.create_scan('123-456', targets, None, []) w.scan_collection.scans_table['123-456']['results'] = list() results = ["ALARM||| ||| ||| ||| |||some alarm|||path", None] MockDBClass.get_result.return_value = results mock_add_scan_alarm_to_list.return_value = None w.report_openvas_results(MockDBClass, '123-456') assert_called_once(logging.Logger.warning)
def test_get_openvas_result_hosts_count(self, MockDBClass): w = DummyDaemon() target_element = w.create_xml_target() targets = OspRequest.process_target_element(target_element) w.create_scan('123-456', targets, None, []) results = [ "HOSTS_COUNT||| ||| ||| ||| |||4", ] MockDBClass.get_result.return_value = results w.set_scan_total_hosts = MagicMock() w.report_openvas_results(MockDBClass, '123-456') w.set_scan_total_hosts.assert_called_with( '123-456', 4, )
def test_get_openvas_result_host_start(self, mock_add_scan_log_to_list, MockDBClass): w = DummyDaemon() target_element = w.create_xml_target() targets = OspRequest.process_target_element(target_element) w.create_scan('123-456', targets, None, []) results = [ "HOST_START|||192.168.10.124||| ||| ||||||today 1", ] MockDBClass.get_result.return_value = results mock_add_scan_log_to_list.return_value = None w.report_openvas_results(MockDBClass, '123-456') mock_add_scan_log_to_list.assert_called_with( host='192.168.10.124', name='HOST_START', value='today 1', )
def test_get_openvas_result(self, mock_add_scan_log_to_list, MockDBClass): w = DummyDaemon() target_element = w.create_xml_target() targets = OspRequest.process_target_element(target_element) w.create_scan('123-456', targets, None, []) results = ["LOG||| |||general/Host_Details||| |||Host dead", None] mock_db = MockDBClass.return_value mock_db.get_result.side_effect = results mock_add_scan_log_to_list.return_value = None w.load_vts() w.report_openvas_results(mock_db, '123-456', 'localhost') mock_add_scan_log_to_list.assert_called_with( host='localhost', hostname='', name='', port='general/Host_Details', qod='', test_id='', value='Host dead', )