pytest

Author: dmsnback

tests/conftest.py

@@ -1,12 +1,10 @@
 import pytest
 import pytest_asyncio
+from db.models import Base, News
 from sqlalchemy import create_engine
-from sqlalchemy.ext.asyncio import create_async_engine, async_sessionmaker
+from sqlalchemy.ext.asyncio import async_sessionmaker, create_async_engine
 from sqlalchemy.orm import sessionmaker
 
-from db.models import Base, News
-
-
 DATABASE_SYNC_URL = "sqlite:///:memory:"
 DATABASE_ASYNC_URL = "sqlite+aiosqlite:///:memory:"
 
@@ -35,18 +33,28 @@ async def async_session():
 @pytest.fixture
 def news():
     test_news = {
-        'url': 'https://panorama.pub/test-news',
-        'title': 'Test Title',
-        'text': 'Test Text',
-        'image': None
+        "url": "https://panorama.pub/test-news",
+        "title": "Test Title",
+        "text": "Test Text",
+        "image": None,
     }
     return test_news
 
 
 @pytest_asyncio.fixture
 async def news_list(async_session):
-    first_news = News(url="https://panorama.pub/test-news_1", title="Test Title 1", image=None, text="Ttest Text_1")
-    second_news = News(url="https://panorama.pub/test-news_2", title="Test Title 2", image=None, text="Ttest Text_2")
+    first_news = News(
+        url="https://panorama.pub/test-news_1",
+        title="Test Title 1",
+        image=None,
+        text="Ttest Text_1",
+    )
+    second_news = News(
+        url="https://panorama.pub/test-news_2",
+        title="Test Title 2",
+        image=None,
+        text="Ttest Text_2",
+    )
 
     async_session.add_all([first_news, second_news])
     await async_session.commit()

tests/test_bot_storage.py

@@ -1,6 +1,10 @@
 import pytest
-
-from bot.bot_storage import get_last_sent_id, save_last_sent_news_id, get_news_after_id, get_all_users
+from bot.bot_storage import (
+    get_all_users,
+    get_last_sent_id,
+    get_news_after_id,
+    save_last_sent_news_id,
+)
 
 
 @pytest.mark.asyncio
@@ -9,29 +13,43 @@ async def test_get_last_sent_id(async_session):
     chat_id = 100
 
     last_id = await get_last_sent_id(chat_id, async_session)
-    assert last_id == 0, "id последней отпрвленной новости должен быть равен 0, если нет отправленных новостей"
+    assert (
+        last_id == 0
+    ), "id последней отпрвленной новости должен быть равен 0, если нет отправленных новостей"
 
-    await save_last_sent_news_id(chat_id=chat_id, last_id=10, session=async_session)
+    await save_last_sent_news_id(
+        chat_id=chat_id, last_id=10, session=async_session
+    )
     last_id = await get_last_sent_id(chat_id, async_session)
-    assert last_id == 10, "id последней отпрвленной новости должен быть равен 10"
+    assert (
+        last_id == 10
+    ), "id последней отпрвленной новости должен быть равен 10"
 
 
 @pytest.mark.asyncio
 async def test_get_news_after_id(async_session, news_list):
     """Получение новостей"""
     news = await get_news_after_id(last_id=0, session=async_session)
     assert len(news) == 2, "Должно быть получено 2 новости"
-    assert news[0].title == "Test Title 1", "Заголовок первой новости должен быть 'Test Title 1'"
-    assert news[1].title == "Test Title 2", "Заголовок второй новости должен быть 'Test Title 2'"
+    assert (
+        news[0].title == "Test Title 1"
+    ), "Заголовок первой новости должен быть 'Test Title 1'"
+    assert (
+        news[1].title == "Test Title 2"
+    ), "Заголовок второй новости должен быть 'Test Title 2'"
 
 
 @pytest.mark.asyncio
 async def test_get_all_users(async_session):
     """Получение пользователей"""
-    await save_last_sent_news_id(chat_id=111, last_id=11, session=async_session)
-    await save_last_sent_news_id(chat_id=222, last_id=22, session=async_session)
+    await save_last_sent_news_id(
+        chat_id=111, last_id=11, session=async_session
+    )
+    await save_last_sent_news_id(
+        chat_id=222, last_id=22, session=async_session
+    )
 
     users = await get_all_users(async_session)
     assert len(users) == 2, "Должно быть 2 пользователя в базе"
     assert users[0] == 111, "id первого юзера должен быть 111"
-    assert users[1] == 222,  "id второго юзеера должен быть 222"
+    assert users[1] == 222, "id второго юзеера должен быть 222"

tests/test_spider_storage.py

@@ -5,6 +5,16 @@ def test_save_and_check_news(session, news):
     """Проверяет что новостей нет в базе,
     сохранение новости
     и новость появилась в базе после сохранения."""
-    assert not is_news_exists(news['url'], session), "Перед сохраением новой новости её не должно быть в базе"
-    save_news(news["url"], news["title"], news['image'], news["text"], session=session)
-    assert is_news_exists(news["url"], session), "Новость должна появиться в базе после сохранения"
+    assert not is_news_exists(
+        news["url"], session
+    ), "Перед сохраением новой новости её не должно быть в базе"
+    save_news(
+        news["url"],
+        news["title"],
+        news["image"],
+        news["text"],
+        session=session,
+    )
+    assert is_news_exists(
+        news["url"], session
+    ), "Новость должна появиться в базе после сохранения"