def add_vt_to_cache(self, vt_id: str, vt: List[str]):
if not vt_id:
raise RequiredArgument('add_vt_to_cache', 'vt_id')
if not vt:
raise RequiredArgument('add_vt_to_cache', 'vt')
if not isinstance(vt, list) or len(vt) != 15:
raise OspdOpenvasError('Error trying to load the VT'
' {} in cache'.format(vt))
OpenvasDB.add_single_list(self.ctx, vt_id, vt)
def test_add_single_list(self, mock_redis):
ctx = mock_redis.return_value
pipeline = ctx.pipeline.return_value
pipeline.delete.return_value = None
pipeline.execute.return_value = (None, 0)
OpenvasDB.add_single_list(ctx, 'a', ['12', '11', '12'])
pipeline.delete.assert_called_once_with('a')
pipeline.rpush.assert_called_once_with('a', '12', '11', '12')
assert_called(pipeline.execute)
def add_vt_to_cache(self, vt_id: str, vt: List[str]):
if not vt_id:
raise RequiredArgument('add_vt_to_cache', 'vt_id')
if not vt:
raise RequiredArgument('add_vt_to_cache', 'vt')
if not isinstance(vt, list) or len(vt) != 15:
raise OspdOpenvasError(
f'Error trying to load the VT {vt} in cache')
OpenvasDB.add_single_list(self.ctx, vt_id, vt)
OpenvasDB.add_single_item(self.ctx, f'filename:{vt[0]}', [int(time())])
