def test_get_keys_by_pattern_error(self, mock_redis):
ctx = mock_redis.return_value
with self.assertRaises(RequiredArgument):
OpenvasDB.get_keys_by_pattern(None, 'a')
with self.assertRaises(RequiredArgument):
OpenvasDB.get_keys_by_pattern(ctx, None)
def test_get_keys_by_pattern(self, mock_redis):
ctx = mock_redis.return_value
ctx.keys.return_value = ['nvt:2', 'nvt:1']
ret = OpenvasDB.get_keys_by_pattern(ctx, 'nvt:*')
# Return sorted list
self.assertEqual(ret, ['nvt:1', 'nvt:2'])
