fix: fixed tests

creyD · Jan 24, 2025 · 5b0cc0d · 5b0cc0d
1 parent ecfc0fc
commit 5b0cc0d
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/test.py b/test.py
@@ -7,9 +7,7 @@
 from sqlalchemy.orm import sessionmaker
 
 from creyPY.fastapi.app import generate_unique_id
-from creyPY.fastapi.crud import (
-    get_object_or_404,
-)
+from creyPY.fastapi.crud import get_object_or_404
 from creyPY.fastapi.models.base import Base
 
 
@@ -65,7 +63,7 @@ def test_generate_unique_id(self):
     def test_get_object_or_404_existing_object(self):
         # Arrange
         obj_id = UUID("123e4567-e89b-12d3-a456-426614174000")
-        obj = MockDBClass(obj_id)
+        obj = MockDBClass(id=obj_id)
         self.db.add(obj)
         self.db.commit()