Merge pull request #52 from PythonFloripa/51-atualizar-endpoint-de-likes

Feature/#51

Author: dan5e3s6ares

app/routers/news/routes.py

@@ -28,10 +28,6 @@ class NewsLikeResponse(BaseModel):
     total_likes: int | None
 
 
-class LikeRequest(BaseModel):
-    email: str
-
-
 def encode_email(email: str) -> str:
     """Encodes the email to be safely stored in database."""
     return jwt.encode({"email": email}, SECRET_KEY, algorithm=ALGORITHM)
@@ -107,13 +103,12 @@ async def post_like(
             DBCommunity, Depends(get_current_active_community)
         ],
         news_id: str,
-        body: LikeRequest,
         user_email: str = Header(..., alias="user-email"),
     ):
         """
         News endpoint where user can set like to news item.
         """
-        encoded_email = encode_email(body.email)
+        encoded_email = encode_email(user_email)
         total_likes = await orm_news.like_news(
             session=request.app.db_session_factory,
             news_id=news_id,
@@ -134,13 +129,12 @@ async def delete_like(
             DBCommunity, Depends(get_current_active_community)
         ],
         news_id: str,
-        email: str,
         user_email: str = Header(..., alias="user-email"),
     ):
         """
         News endpoint where user can set like to news item.
         """
-        encoded_email = encode_email(email)
+        encoded_email = encode_email(user_email)
         total_likes = await orm_news.delete_like(
             session=request.app.db_session_factory,
             news_id=news_id,

tests/test_news.py

@@ -318,7 +318,7 @@ async def test_news_likes_endpoint(
     response = await async_client.post(
         f"/api/news/{stored_news.id}/like",
         json={"email": emails[0]},
-        headers=valid_auth_headers,
+        headers={**valid_auth_headers, "user-email": emails[0]},
     )
     assert response.status_code == status.HTTP_200_OK
     statement = select(News).where(News.title == news_data["title"])
@@ -329,8 +329,7 @@ async def test_news_likes_endpoint(
 
     response = await async_client.post(
         f"/api/news/{stored_news.id}/like",
-        json={"email": emails[1]},
-        headers=valid_auth_headers,
+        headers={**valid_auth_headers, "user-email": emails[1]},
     )
     assert response.status_code == status.HTTP_200_OK
     statement = select(News).where(News.title == news_data["title"])
@@ -345,8 +344,7 @@ async def test_news_likes_endpoint(
     # Remove likes
     response = await async_client.delete(
         f"/api/news/{stored_news.id}/like",
-        params={"email": emails[0]},
-        headers=valid_auth_headers,
+        headers={**valid_auth_headers, "user-email": emails[0]},
     )
     assert response.status_code == status.HTTP_200_OK
     statement = select(News).where(News.title == news_data["title"])
@@ -357,8 +355,7 @@ async def test_news_likes_endpoint(
 
     response = await async_client.delete(
         f"/api/news/{stored_news.id}/like",
-        params={"email": emails[1]},
-        headers=valid_auth_headers,
+        headers={**valid_auth_headers, "user-email": emails[1]},
     )
     assert response.status_code == status.HTTP_200_OK
     statement = select(News).where(News.title == news_data["title"])