Create tests for PUT news endpoint

- Create test_put_news_endpoint
- Add publish to test_insert_news

Author: arthur-mesquita

tests/test_news.py

@@ -72,21 +72,22 @@ async def test_insert_news(
     await session.commit()
     statement = select(News).where(News.title == "Test news")
     result = await session.exec(statement)
-    found_news = result.first()
-    assert found_news is not None
-    assert found_news.title == news_list[0].title
-    assert found_news.content == news_list[0].content
-    assert found_news.category == news_list[0].category
-    assert found_news.user_email == news_list[0].user_email
-    assert found_news.source_url == news_list[0].source_url
-    assert found_news.tags == news_list[0].tags
-    assert found_news.social_media_url == news_list[0].social_media_url
-    assert found_news.likes == 0
-    assert found_news.community_id == community.id
-    assert isinstance(found_news.created_at, datetime)
-    assert isinstance(found_news.updated_at, datetime)
-    assert found_news.created_at <= datetime.now()
-    assert found_news.updated_at >= found_news.created_at
+    stored_news = result.first()
+    assert stored_news is not None
+    assert stored_news.title == news_list[0].title
+    assert stored_news.content == news_list[0].content
+    assert stored_news.category == news_list[0].category
+    assert stored_news.user_email == news_list[0].user_email
+    assert stored_news.source_url == news_list[0].source_url
+    assert stored_news.tags == news_list[0].tags
+    assert stored_news.social_media_url == news_list[0].social_media_url
+    assert stored_news.likes == 0
+    assert stored_news.community_id == community.id
+    assert isinstance(stored_news.created_at, datetime)
+    assert isinstance(stored_news.updated_at, datetime)
+    assert stored_news.created_at <= datetime.now()
+    assert stored_news.updated_at >= stored_news.created_at
+    assert stored_news.publish is False
 
 
 @pytest.mark.asyncio
@@ -221,6 +222,7 @@ async def test_get_news_by_id(
     statement = select(News).where(News.title == "Test news")
     result = await session.exec(statement)
     stored_news = result.first()
+    assert stored_news is not None
 
     response = await async_client.get(
         "/api/news",
@@ -288,6 +290,51 @@ async def test_news_integration(
     assert data["news_list"][0]["likes"] == 0
 
 
+@pytest.mark.asyncio
+async def test_put_news_endpoint(
+    session: AsyncSession,
+    async_client: AsyncClient,
+    community: Community,
+    valid_auth_headers: Mapping[str, str],
+):
+    session.add_all(news_list)
+    await session.commit()
+
+    statement = select(News).where(News.title == "Test news")
+    result = await session.exec(statement)
+    stored_news = result.first()
+    assert stored_news is not None
+
+    response = await async_client.put(
+        "/api/news",
+        params={
+            "title": "updated title",
+            "content": "updated content",
+            "category": "updated_category",
+            "user_email": "updated_email@test.com",
+            "source_url": "https://updated_url.com",
+            "tags": "test_tag_updated",
+            "user_email_list": "updated_email@test.com",
+            "social_media_url": "https://updated_social_media_url.com",
+            "likes": 42,
+        },
+        headers=valid_auth_headers,
+        json={"publish": True},
+    )
+    data = response.json()
+    assert response.status_code == status.HTTP_200_OK
+    assert data["tittle"] == "updated title"
+    assert data["content"] == "updated content"
+    assert data["category"] == "updated_category"
+    assert data["user_email"] == "updated_email@test.com"
+    assert data["source_url"] == "https://updated_url.com"
+    assert data["tags"] == "test_tag_updated"
+    assert data["user_email_list"] == "updated_email@test.com"
+    assert data["social_media_url"] == "https://updated_social_media_url.com"
+    assert data["likes"] == 42
+    assert data["publish"] is True
+
+
 @pytest.mark.asyncio
 async def test_news_likes_endpoint(
     session: AsyncSession,
@@ -325,6 +372,7 @@ async def test_news_likes_endpoint(
     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 == 1
     assert stored_news.user_email_list == f"['{encode_email(emails[0])}']"
 
@@ -336,6 +384,7 @@ async def test_news_likes_endpoint(
     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 == 2
     assert (
         stored_news.user_email_list
@@ -351,6 +400,7 @@ async def test_news_likes_endpoint(
     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 == 1
     assert stored_news.user_email_list == f"['{encode_email(emails[1])}']"
 
@@ -362,5 +412,6 @@ async def test_news_likes_endpoint(
     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
     assert stored_news.user_email_list == "[]"