diff --git a/src/repository/users.py b/src/repository/users.py index 8ba75d8..021785c 100644 --- a/src/repository/users.py +++ b/src/repository/users.py @@ -1,7 +1,7 @@ from datetime import date from typing import Optional -from sqlalchemy import func, select, update +from sqlalchemy import func, select from sqlalchemy.orm import selectinload from src.models import UsersORM diff --git a/tests/unit_tests/test_repo_db.py b/tests/unit_tests/test_repo_db.py index b26cbb5..1c3e4c8 100644 --- a/tests/unit_tests/test_repo_db.py +++ b/tests/unit_tests/test_repo_db.py @@ -55,3 +55,32 @@ async def test_tasks_user(db: "TestDBManager"): user_with_tasks = await db.user.get_one_with_load(user_id=admin_user.id) assert user_with_tasks assert user_with_tasks.tasks + + +async def test_tasks_crud(db: "TestDBManager"): + data = { + "title": "test_tasks_crud", + "description": "test", + "due_date": datetime.now(), + "status": "open", + "priority": "medium", + "user_id": 1, + } + task = await db.task.create_one(data) + assert task + assert task.title == data["title"] + assert task.description == data["description"] + assert task.due_date == data["due_date"].date() + assert task.status == data["status"] + assert task.priority == data["priority"] + assert task.user_id == data["user_id"] + assert task.created_at is not None + find_task = await db.task.get_filtered(title=data["title"]) + assert find_task + assert find_task[0].title == task.title + data["title"] = "test2" + task = await db.task.update_one(id=task.id, data=data) + assert task.title == data["title"] + await db.task.delete_one(id=task.id) + task = await db.task.get_one_or_none(id=task.id) + assert not task \ No newline at end of file