test: Настройка отображения покрытия кода

Author: A-V-tor

.github/workflows/test_runner.yaml

@@ -0,0 +1,69 @@
+name: run tests
+
+on:
+  push:
+    branches: [main]
+  pull_request:
+    branches: [main]
+
+jobs:
+  build:
+    runs-on: ubuntu-latest
+
+    services:
+      postgres:
+        image: postgres:13
+        ports:
+          - 5432:5432
+        env:
+          POSTGRES_USER: a-v-tor
+          POSTGRES_PASSWORD: postgres
+          POSTGRES_DB: test_bot_secretar
+
+    steps:
+      - name: Checkout repository
+        uses: actions/checkout@v2
+
+      - name: Set up Python
+        uses: actions/setup-python@v2
+        with:
+          python-version: "3.11"
+
+      - name: Install Poetry
+        run: |
+          pip install poetry
+
+      - name: Install dependencies
+        run: |
+          poetry install
+
+      - name: Wait for PostgreSQL to start
+        run: |
+          sudo apt-get update
+          sudo apt-get install -y postgresql-client
+          until pg_isready -h localhost -p 5432 -U a-v-tor; do
+            sleep 1
+          done
+
+      - name: Environment variables setup
+        run: |
+          echo "DATABASE_URL=postgresql+psycopg2://a-v-tor:postgres@localhost:5432/test_bot_secretar" >> $GITHUB_ENV
+
+      - name: Debug environment
+        run: |
+          printenv | grep DATABASE_URL
+
+      - name: Apply migrations
+        env:
+          DATABASE_URL: ${{ env.DATABASE_URL }}
+        run: |
+          poetry run alembic upgrade head
+
+      - name: Run tests
+        run: |
+          poetry run pytest --cov=./ --cov-report=xml
+
+      - name: Upload coverage to Codecov
+        uses: codecov/codecov-action@v1
+        with:
+          token: ${{ secrets.CODECOV_TOKEN }}

README.md

@@ -1,4 +1,7 @@
 <h1 align="center">Личный бот - секретарь</h1>
+<a href="https://codecov.io/github/A-V-tor/Bot-secretar" >
+ <img src="https://codecov.io/github/A-V-tor/Bot-secretar/graph/badge.svg?token=65PRUK4GYD"/>
+ </a>
 <br>
 <div id="header" align="center">
 <img src='https://media.giphy.com/media/wlR4kWTnwEyY8RwHKM/giphy.gif' width="100"/>

env.example

@@ -8,7 +8,7 @@ DB_USER=a-v-tor
 DB_PASS=postgres
 DB_HOST=localhost
 DB_PORT=5432
-DB_NAME=bot_secretar
+DB_NAME=test_bot_secretar
 
 URL_ADMIN=https://localhost/login
 

pyproject.toml

@@ -36,6 +36,9 @@ loguru = "^0.7.3"
 ipykernel = "^6.19.4"
 pre-commit = "^3.6.0"
 ruff = "^0.11.0"
+pytest-cov = "^6.1.1"
+pytest-asyncio = "^0.26.0"
+pytest-dotenv = "^0.5.2"
 
 [build-system]
 requires = ["poetry-core"]

pytest.ini

@@ -0,0 +1,3 @@
+[pytest]
+pythonpath = . src
+env_files = env.example

tests/conftest.py

@@ -0,0 +1,31 @@
+import os
+
+import pytest
+from dotenv import load_dotenv
+from sqlalchemy import create_engine
+from sqlalchemy.orm import sessionmaker
+
+from config import DevelopConfig
+from src.database.base import Base
+
+
+@pytest.fixture(scope='session', autouse=True)
+def load_env():
+    load_dotenv(dotenv_path='env.example')
+
+
+@pytest.fixture(name='get_db', scope='session')
+def get_test_db():
+    # SQLALCHEMY_DATABASE_URI = 'postgresql+psycopg2://a-v-tor:postgres@localhost:5432/test_bot_secretar'
+    # engine = create_engine(SQLALCHEMY_DATABASE_URI)
+
+    settings = DevelopConfig()
+    engine = create_engine(settings.SQLALCHEMY_DATABASE_URI)
+
+    Base.metadata.drop_all(bind=engine)
+    Base.metadata.create_all(bind=engine)
+    session_factory = sessionmaker(
+        engine,
+        expire_on_commit=False,
+    )
+    yield session_factory

tests/test_bot_handlers.py

@@ -0,0 +1,51 @@
+import re
+from unittest.mock import AsyncMock, create_autospec
+
+import pytest
+from aiogram import types
+
+from src.telegram.handlers.base import start_command
+from src.telegram.keyboards.base_kb import start_kb
+
+
+@pytest.mark.asyncio
+async def test_start_command_with_create_user(get_db):
+    message_mock = create_autospec(types.Message)
+    message_mock.text = '/start'
+
+    message_mock.chat = create_autospec(types.Chat, instance=True)
+    message_mock.chat.id = 123456789
+    message_mock.chat.username = 'test_username'
+    message_mock.chat.first_name = 'first_name'
+    message_mock.chat.last_name = 'last_name'
+
+    message_mock.reply = AsyncMock()
+
+    await start_command(message=message_mock)
+
+    message_mock.reply.assert_called()
+    actual_msg = message_mock.reply.call_args[0][0]
+    assert re.search(r'Вы добавлены в пользователи бота', actual_msg)
+
+    # Проверка других аргументов (reply_markup и parse_mode)
+    actual_reply_markup = message_mock.reply.call_args[1]['reply_markup']
+    assert actual_reply_markup == await start_kb()
+    assert message_mock.reply.call_args[1]['parse_mode'] == 'HTML'
+
+
+@pytest.mark.asyncio
+async def test_start_command_with_existing_user(get_db):
+    message_mock = create_autospec(types.Message)
+    message_mock.text = '/start'
+
+    message_mock.chat = create_autospec(types.Chat, instance=True)
+    message_mock.chat.id = 123456789
+    message_mock.chat.username = 'test_username'
+    message_mock.chat.first_name = 'first_name'
+    message_mock.chat.last_name = 'last_name'
+
+    message_mock.reply = AsyncMock()
+
+    await start_command(message=message_mock)
+
+    message_mock.reply.assert_called_with('Главное меню', reply_markup=await start_kb(), parse_mode='HTML')