ruff fixes

src/chat/routers.py

@@ -1,12 +1,8 @@
-from typing import Annotated
 import uuid
+from typing import Annotated
+from uuid import UUID
 
-
-from motor.motor_asyncio import AsyncIOMotorClient
-
-from src.database import mongo
 from bson.binary import Binary
-
 from fastapi import APIRouter, Depends
 from fastapi.websockets import WebSocket
 from orjson import JSONEncodeError, orjson
@@ -24,6 +20,7 @@
     update_message,
     ws_manager,
 )
+from src.database import mongo
 
 ws_router = APIRouter(
     prefix="/chat",
@@ -74,7 +71,7 @@ async def websocket_chat(
 
 
 @ws_router.get("/messages/{match_id}")
-async def get_latest_messages(match_id):
+async def get_latest_messages(match_id: UUID):
 
     collection = mongo.collection
 

src/chat/utils.py

@@ -169,5 +169,4 @@ async def update_message(ws_msg: WSMessageRequest, ws: WebSocket, user: AuthUser
 
     await send_ws_message(ws, msg, WSAction.UPDATE)
 
-
 ws_manager = WebSocketConnectionManager()

tests/test_acceptance.py

@@ -1,13 +1,14 @@
+import uuid
+
 import orjson
 from async_asgi_testclient import TestClient
+from bson.binary import Binary
 from dirty_equals import IsStr, IsUUID
 from fastapi import status
-from src.mongodb.mongodb import Mongo
-import uuid
-from bson.binary import Binary
 
 from src.chat.schemas import MessageStatus, WSAction, WSStatus
 from src.chat.utils import orjson_dumps
+from src.mongodb.mongodb import Mongo
 
 
 class TestAcceptance:
@@ -470,9 +471,9 @@ async def test_get_latest_messages(self, mongo: Mongo, async_client: TestClient)
             "from_id": Binary(from_id.bytes, subtype=3),
             "to_id": Binary(to_id.bytes, subtype=3),
             "text": text,
-            "created_at": '2023-01-01T00:00:00',
-            "updated_at": '2023-01-01T00:00:00',
-            "status": "sent"
+            "created_at": "2023-01-01T00:00:00",
+            "updated_at": "2023-01-01T00:00:00",
+            "status": "sent",
         }
 
         collection.insert_one(message)
@@ -493,4 +494,4 @@ async def test_get_latest_messages(self, mongo: Mongo, async_client: TestClient)
         assert result[0]["updated_at"] == message["updated_at"]
         assert result[0]["status"] == message["status"]
 
-        collection.delete_one({'match_id': match_id})
+        collection.delete_one({"match_id": match_id})