def test_get_filenames_and_oids(self, mock_redis):
ctx = mock_redis.return_value
ctx.keys.return_value = ['nvt:1', 'nvt:2']
ctx.lindex.side_effect = ['aa', 'ab']
ret = OpenvasDB.get_filenames_and_oids(ctx)
self.assertEqual(list(ret), [('aa', '1'), ('ab', '2')])
def get_oids(self) -> Iterator[Tuple[str, str]]:
"""Get the list of NVT file names and OIDs.
Returns:
A i. Each single list contains the filename
as first element and the oid as second one.
"""
return OpenvasDB.get_filenames_and_oids(self.ctx)
def get_oids(self) -> Iterator[Tuple[str, str]]:
"""Get the list of NVT file names and OIDs.
Returns:
An iterable of tuples of file name and oid.
"""
if self.notus:
for f, oid in self.notus.get_filenames_and_oids():
yield (f, oid)
if self.ctx:
for f, oid in OpenvasDB.get_filenames_and_oids(self.ctx):
if not self.notus or not self.notus.exists(oid):
yield (f, oid)
def test_get_filenames_and_oids_error(self, mock_redis):
ctx = mock_redis.return_value
with self.assertRaises(RequiredArgument):
OpenvasDB.get_filenames_and_oids(None)
