From d17c07632e762871ed269776de717e2a653823eb Mon Sep 17 00:00:00 2001 From: Alexis Feron Date: Wed, 29 Jan 2025 16:29:25 +0100 Subject: [PATCH] =?UTF-8?q?=F0=9F=94=92=EF=B8=8F=20Assign=20pin=20to=20a?= =?UTF-8?q?=20user?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/dto/PinDTO.py | 3 ++- app/models/pin.py | 3 ++- app/serializers/pin_serializer.py | 3 ++- tests/test_pins.py | 10 ++++++---- 4 files changed, 12 insertions(+), 7 deletions(-) diff --git a/app/dto/PinDTO.py b/app/dto/PinDTO.py index e7b32d4..45ad39b 100644 --- a/app/dto/PinDTO.py +++ b/app/dto/PinDTO.py @@ -4,4 +4,5 @@ class PinDTO(BaseModel): title: str description: str location: list - files: list \ No newline at end of file + files: list + user_id: str \ No newline at end of file diff --git a/app/models/pin.py b/app/models/pin.py index 33b2d1c..9374a25 100644 --- a/app/models/pin.py +++ b/app/models/pin.py @@ -6,4 +6,5 @@ class Pin(BaseModel): title: str description: str location: list - files: Optional[list] \ No newline at end of file + files: Optional[list] + user_id: str \ No newline at end of file diff --git a/app/serializers/pin_serializer.py b/app/serializers/pin_serializer.py index 190c833..bd6473d 100644 --- a/app/serializers/pin_serializer.py +++ b/app/serializers/pin_serializer.py @@ -7,7 +7,8 @@ def pin_serialize(pin: list) -> Pin: "title": pin["title"], "description": pin["description"], "location": pin["location"], - "files": pin["files"] + "files": pin["files"], + "user_id": pin["user_id"] }) def pins_serialize(pins: list) -> list: diff --git a/tests/test_pins.py b/tests/test_pins.py index ba3f316..b42b033 100644 --- a/tests/test_pins.py +++ b/tests/test_pins.py @@ -1,7 +1,7 @@ from test_main import * def test_add_pin(token): - response = client.post("/pin/add", json={"title": "Test Pin", "description": "Test Description", "location": [0,0], "files": ["Test file 1"]}, headers={"Authorization": f"Bearer {token}"}) + response = client.post("/pin/add", json={"title": "Test Pin", "description": "Test Description", "location": [0,0], "files": ["Test file 1"], "user_id": "6798e24b9d212d0f6a6b0abd"}, headers={"Authorization": f"Bearer {token}"}) assert response.status_code == 200 data = response.json() assert "id" in data @@ -13,7 +13,7 @@ def test_list_pins(token): assert isinstance(data, list) def test_get_pin(token): - add_pin_response = client.post("/pin/add", json={"title": "Test Pin", "description": "Test Description", "location": [0,0], "files": ["Test file 1"]}, headers={"Authorization": f"Bearer {token}"}) + add_pin_response = client.post("/pin/add", json={"title": "Test Pin", "description": "Test Description", "location": [0,0], "files": ["Test file 1"], "user_id": "6798e24b9d212d0f6a6b0abd"}, headers={"Authorization": f"Bearer {token}"}) pin_id = add_pin_response.json()["id"] response = client.get(f"/pin/{pin_id}", headers={"Authorization": f"Bearer {token}"}) @@ -23,6 +23,7 @@ def test_get_pin(token): assert data["description"] == "Test Description" assert data["location"] == [0,0] assert data["files"] == ["Test file 1"] + assert data["user_id"] == "6798e24b9d212d0f6a6b0abd" def test_get_pin_wrong_format(token): @@ -36,10 +37,10 @@ def test_get_pin_unknown_id(token): def test_update_pin(token): - add_pin_response = client.post("/pin/add", json={"title": "Test Pin", "description": "Test Description", "location": [0,0], "files": ["Test file 1"]}, headers={"Authorization": f"Bearer {token}"}) + add_pin_response = client.post("/pin/add", json={"title": "Test Pin", "description": "Test Description", "location": [0,0], "files": ["Test file 1"], "user_id": "6798e24b9d212d0f6a6b0abd"}, headers={"Authorization": f"Bearer {token}"}) pin_id = add_pin_response.json()["id"] - update_response = client.patch(f"/pin/{pin_id}", json={"title": "Updated Pin", "description": "Updated Description", "location": [0,1], "files": ["Updated file 1"]}, headers={"Authorization": f"Bearer {token}"}) + update_response = client.patch(f"/pin/{pin_id}", json={"title": "Updated Pin", "description": "Updated Description", "location": [0,1], "files": ["Updated file 1"], "user_id": "6798e24b9d212d0f6a6b0abd"}, headers={"Authorization": f"Bearer {token}"}) assert update_response.status_code == 200 get_pin_response = client.get(f"/pin/{pin_id}", headers={"Authorization": f"Bearer {token}"}) @@ -48,6 +49,7 @@ def test_update_pin(token): assert data["description"] == "Updated Description" assert data["location"] == [0,1] assert data["files"] == ["Updated file 1"] + assert data["user_id"] == "6798e24b9d212d0f6a6b0abd" def test_update_wrong_format(token):