def test_process_vts_not_found(self, mock_kb):
w = DummyDaemon()
logging.Logger.warning = Mock()
vts = {
'1.3.6.1.4.1.25623.1.0.100065': {'3': 'new value'},
'vt_groups': ['family=debian', 'family=general'],
}
p = PreferenceHandler('1234-1234', mock_kb, w.scan_collection, w.nvti)
w.nvti.get_nvt_metadata.return_value = None
p._process_vts(vts)
assert_called_once(logging.Logger.warning)
def test_process_vts_not_found(self, mock_kb):
w = DummyDaemon()
logging.Logger.warning = Mock()
vts = {
'1.3.6.1.4.1.25623.1.0.100065': {
'3': 'new value'
},
'vt_groups': ['family=debian', 'family=general'],
}
w.load_vts()
temp_vts = w.vts
p = PreferenceHandler('1234-1234', mock_kb, w.scan_collection,
temp_vts)
p._process_vts(vts)
assert_called_once(logging.Logger.warning)
def test_process_vts_bad_param_id(self):
w = DummyDaemon()
vts = {
'1.3.6.1.4.1.25623.1.0.100061': {'3': 'new value'},
'vt_groups': ['family=debian', 'family=general'],
}
p = PreferenceHandler('1234-1234', None, w.scan_collection, w.nvti)
ret = p._process_vts(vts)
self.assertFalse(ret[1])
def test_not_append_notus_oids(self):
dummy = DummyDaemon()
vts = {
'1.3.6.1.4.1.25623.1.0.100061': {'1': 'new value'},
'vt_groups': ['family=debian', 'family=general'],
}
p_handler = PreferenceHandler(
'1234-1234', None, dummy.scan_collection, dummy.nvti, lambda _: True
)
re = p_handler._process_vts(vts) # pylint: disable = protected-access
self.assertEqual(re[0], [])
self.assertEqual(re[1], {})
def test_process_vts_bad_param_id(self):
dummy = DummyDaemon()
vts = {
'1.3.6.1.4.1.25623.1.0.100061': {'3': 'new value'},
'vt_groups': ['family=debian', 'family=general'],
}
p_handler = PreferenceHandler(
'1234-1234', None, dummy.scan_collection, dummy.nvti
)
ret = p_handler._process_vts(vts) # pylint: disable = protected-access
self.assertFalse(ret[1])
def test_process_vts(self):
w = DummyDaemon()
vts = {
'1.3.6.1.4.1.25623.1.0.100061': {'1': 'new value'},
'vt_groups': ['family=debian', 'family=general'],
}
vt_out = (
['1.3.6.1.4.1.25623.1.0.100061'],
{'1.3.6.1.4.1.25623.1.0.100061:1:entry:Data length :': 'new value'},
)
p = PreferenceHandler('1234-1234', None, w.scan_collection, w.nvti)
ret = p._process_vts(vts)
self.assertEqual(ret, vt_out)
def test_process_vts(self):
dummy = DummyDaemon()
vts = {
'1.3.6.1.4.1.25623.1.0.100061': {'1': 'new value'},
'vt_groups': ['family=debian', 'family=general'],
}
vt_out = (
['1.3.6.1.4.1.25623.1.0.100061'],
{'1.3.6.1.4.1.25623.1.0.100061:1:entry:Data length :': 'new value'},
)
p_handler = PreferenceHandler(
'1234-1234', None, dummy.scan_collection, dummy.nvti
)
ret = p_handler._process_vts(vts) # pylint: disable = protected-access
self.assertEqual(ret, vt_out)
