diff --git a/tests/driver/test_falkordb_driver.py b/tests/driver/test_falkordb_driver.py index 6cca9e74..1220101a 100644 --- a/tests/driver/test_falkordb_driver.py +++ b/tests/driver/test_falkordb_driver.py @@ -209,9 +209,12 @@ class TestFalkorDriver: async def test_delete_all_indexes(self): """Test delete_all_indexes method.""" with patch.object(self.driver, 'execute_query', new_callable=AsyncMock) as mock_execute: + # Return None to simulate no indexes found + mock_execute.return_value = None + await self.driver.delete_all_indexes() - mock_execute.assert_called_once_with('CALL db.indexes() YIELD name DROP INDEX name') + mock_execute.assert_called_once_with('CALL db.indexes()') class TestFalkorDriverSession: