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)