def test_get_vt_iterator_with_filter_no_vt(self):
w = DummyDaemon()
vthelper = VtHelper(w.nvti)
w.nvti.get_nvt_metadata.return_value = None
vt = ["1.3.6.1.4.1.25623.1.0.100065"]
for _, values in vthelper.get_vt_iterator(vt_selection=vt):
self.assertIs(values, None)
def test_get_vt_iterator(self):
w = DummyDaemon()
vthelper = VtHelper(w.nvti)
vt = ["1.3.6.1.4.1.25623.1.0.100061"]
for key, _ in vthelper.get_vt_iterator():
self.assertIn(key, vt)
def test_get_vt_iterator_with_filter(self):
w = DummyDaemon()
vthelper = VtHelper(w.nvti)
vt = ["1.3.6.1.4.1.25623.1.0.100061"]
vtout = w.VTS["1.3.6.1.4.1.25623.1.0.100061"]
for key, vt_dict in vthelper.get_vt_iterator(vt_selection=vt):
self.assertIn(key, vt)
for key in vtout:
self.assertIn(key, vt_dict)
def get_vt_iterator(self,
vt_selection: List[str] = None,
details: bool = True) -> Iterator[Tuple[str, Dict]]:
vthelper = VtHelper(self.nvti)
return vthelper.get_vt_iterator(vt_selection, details)