Create like endpoint

Author: arthur-mesquita

app/routers/news/routes.py

@@ -7,7 +7,11 @@
 from app.routers.authentication import get_current_active_community
 from app.schemas import News
 from app.services.database.models import Community as DBCommunity
-from app.services.database.orm.news import create_news, get_news_by_query_params
+from app.services.database.orm.news import (
+    create_news,
+    get_news_by_query_params,
+    like_news,
+)
 
 
 class NewsPostResponse(BaseModel):
@@ -19,6 +23,10 @@ class NewsGetResponse(BaseModel):
     news_list: list = []
 
 
+class NewsLikeResponse(BaseModel):
+    total_likes: int | None
+
+
 def setup():
     router = APIRouter(prefix="/news", tags=["news"])
 
@@ -76,4 +84,29 @@ async def get_news(
         )
         return NewsGetResponse(news_list=news_list)
 
+    @router.post(
+        path="/{news_id}/like",
+        response_model=NewsPostResponse,
+        status_code=status.HTTP_200_OK,
+        summary="News like endpoint",
+        description="Allows user to like a news item",
+    )
+    async def post_like(
+        request: Request,
+        current_community: Annotated[
+            DBCommunity, Depends(get_current_active_community)
+        ],
+        news_id,
+        user_email: str = Header(..., alias="user-email"),
+    ):
+        """
+        News endpoint where user can set like to news item.
+        """
+        total_likes = await like_news(
+            session=request.app.db_session_factory,
+            news_id=news_id,
+            user_email=user_email,
+        )
+        return NewsLikeResponse(total_likes=total_likes)
+
     return router

app/services/database/orm/news.py

@@ -42,3 +42,18 @@ async def get_news_by_query_params(
     statement = select(News).where(*filters)
     results = await session.exec(statement)
     return results.all()
+
+
+async def like_news(
+    session: AsyncSession, news_id: str, user_email: str
+) -> int | None:
+    statement = select(News).where(News.id == news_id)
+    results = await session.exec(statement)
+    news_item = results.first()
+    if news_item:
+        news_item.likes += 1
+        session.add(news_item)
+        await session.commit()
+        await session.refresh(news_item)
+        return news_item.likes
+    return None

tests/test_news.py

@@ -284,3 +284,40 @@ async def test_news_integration(
         == news_data["social_media_url"]
     )
     assert data["news_list"][0]["likes"] == 0
+
+
+@pytest.mark.asyncio
+async def test_insert_news_likes_endpoint(
+    session: AsyncSession,
+    async_client: AsyncClient,
+    community: Community,
+    valid_auth_headers: Mapping[str, str],
+):
+    news_data = {
+        "title": "Test News",
+        "content": "Test news content.",
+        "category": "test_category",
+        "tags": "test_tag",
+        "source_url": "https://example.com/test-news",
+        "social_media_url": "https://test.com/test_news",
+    }
+    response = await async_client.post(
+        "/api/news", json=news_data, headers=valid_auth_headers
+    )
+    assert response.status_code == status.HTTP_200_OK
+    statement = select(News).where(News.title == news_data["title"])
+    result = await session.exec(statement)
+    stored_news = result.first()
+    assert stored_news is not None
+    assert stored_news.likes == 0
+
+    response = await async_client.post(
+        f"/api/news/{stored_news.id}/like",
+        json=news_data,
+        headers=valid_auth_headers,
+    )
+    assert response.status_code == status.HTTP_200_OK
+    statement = select(News).where(News.title == news_data["title"])
+    result = await session.exec(statement)
+    stored_news = result.first()
+    assert stored_news.likes == 1