tests user_with_load

tests/unit_tests/test_repo_db.py

@@ -1,5 +1,8 @@
+import json
+from datetime import datetime
 from typing import TYPE_CHECKING
 
+from src.models.users import UsersORM
 from src.schemas.users import User
 
 if TYPE_CHECKING:
@@ -29,3 +32,24 @@ async def test_user_crud(db: "TestDBManager"):
     await db.user.delete_one(id=updated_user.id)
     delete_user = await db.user.get_one_or_none(id=new_user.id)
     assert not delete_user
+
+
+async def test_tasks_user(db: "TestDBManager"):
+    with open("tests/mock_tasks.json") as jsonfile:
+        data = json.load(jsonfile)
+    admin_user: UsersORM | None = await db.user.get_one_or_none(id=1)
+    assert admin_user
+    data = [
+        {
+            **item,
+            "user_id": admin_user.id,
+            "due_date": datetime.strptime(item["due_date"], "%Y-%m-%d"),
+        }
+        for item in data
+    ]
+    result = await db.task.create_bulk(data)
+    await db.commit()
+    assert result
+    user_with_tasks = await db.user.get_one_with_load(user_id=admin_user.id)
+    assert user_with_tasks
+    assert user_with_tasks.tasks