def test_select_database(self, mock_redis):
mock_redis.execute_command.return_value = mock_redis
OpenvasDB.select_database(mock_redis, 1)
mock_redis.execute_command.assert_called_with('SELECT 1')
def test_select_database_error(self, mock_redis):
with self.assertRaises(RequiredArgument):
OpenvasDB.select_database(None, 1)
with self.assertRaises(RequiredArgument):
OpenvasDB.select_database(mock_redis, None)
