From a9dd90645f50af0fc878e169f2ad830744edec47 Mon Sep 17 00:00:00 2001 From: Maxence Date: Wed, 29 Jan 2025 17:13:29 +0000 Subject: [PATCH] fix friend test --- app/main.py | 2 +- tests/test_friends.py | 3 +-- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/app/main.py b/app/main.py index 94c1956..895ad96 100644 --- a/app/main.py +++ b/app/main.py @@ -215,7 +215,7 @@ async def get_friend(id: str, current_user: User = Depends(get_current_user)): if friend is None: friend_not_found() - return serializers.friend_serialize(friend) + return serializers.friend_serialize(friend,False if friend['user_id']==current_user._id else True) @app.post( path="/friend/add", diff --git a/tests/test_friends.py b/tests/test_friends.py index bc3d75c..0996fe7 100644 --- a/tests/test_friends.py +++ b/tests/test_friends.py @@ -21,7 +21,7 @@ def test_list_friends(token): response = client.get("/friends", headers={"Authorization": f"Bearer {token}"}) assert response.status_code == 200 data = response.json() - assert isinstance(data, dict) + assert isinstance(data, list) def test_get_friend(token): friend_id = str(db["friends"].find_one({})["_id"]) @@ -29,7 +29,6 @@ def test_get_friend(token): assert response.status_code == 200 data = response.json() assert "id" in data - assert "user_id" in data assert "friend_user_id" in data assert "status" in data