def test_get_list_item(self, mock_redis):
ctx = mock_redis.return_value
ctx.lrange.return_value = ['1234']
ret = OpenvasDB.get_list_item(ctx, 'name')
self.assertEqual(ret, ['1234'])
assert_called(ctx.lrange)
def test_parse_metadata_tag_missing_value(self, MockOpenvasDB):
logging.Logger.error = Mock()
tags = 'tag1'
ret = NVTICache._parse_metadata_tags( # pylint: disable=protected-access
tags, '1.2.3')
self.assertEqual(ret, {})
assert_called(logging.Logger.error)
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 test_set_single_item(self, mock_redis):
ctx = mock_redis.return_value
pipeline = ctx.pipeline.return_value
pipeline.delete.return_value = None
pipeline.rpush.return_value = None
pipeline.execute.return_value = None
OpenvasDB.set_single_item(ctx, 'foo', ['bar'])
pipeline.delete.assert_called_once_with('foo')
pipeline.rpush.assert_called_once_with('foo', 'bar')
assert_called(pipeline.execute)
def test_pop_list_items_no_results(self, mock_redis):
ctx = mock_redis.return_value
pipeline = ctx.pipeline.return_value
pipeline.lrange.return_value = None
pipeline.delete.return_value = None
pipeline.execute.return_value = (None, 0)
ret = OpenvasDB.pop_list_items(ctx, 'foo')
self.assertEqual(ret, [])
pipeline.lrange.assert_called_once_with('foo', 0, -1)
pipeline.delete.assert_called_once_with('foo')
assert_called(pipeline.execute)
def test_pop_list_items_with_results(self, mock_redis):
ctx = mock_redis.return_value
pipeline = ctx.pipeline.return_value
pipeline.lrange.return_value = None
pipeline.delete.return_value = None
pipeline.execute.return_value = [['c', 'b', 'a'], 2]
ret = OpenvasDB.pop_list_items(ctx, 'results')
# reversed list
self.assertEqual(ret, ['a', 'b', 'c'])
pipeline.lrange.assert_called_once_with('results', 0, -1)
pipeline.delete.assert_called_once_with('results')
assert_called(pipeline.execute)
