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',
)
