tests/conftest.py - main - evlx/task_tracker

tests/conftest.py

84 строки2 KB

evlx

добавлено сохранение тестовой БД в директорию тестов

29 май 2024, 18:01

29 май 2024, 18:019a535a3

import atexit
import os
import uuid

import pytest
from fastapi.testclient import TestClient
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker

from main import app
from src.db_connect import get_db
from src.employee.model import Base

TEST_DB_DIR = os.path.join(os.path.dirname(__file__), 'test')
if not os.path.exists(TEST_DB_DIR):
    os.mkdir(TEST_DB_DIR)
SQLALCHEMY_DATABASE_URL = f"sqlite:///{os.path.join(TEST_DB_DIR, 'test.db')}"

engine = create_engine(
    SQLALCHEMY_DATABASE_URL,
    connect_args={"check_same_thread": False}
)
TestingSessionLocal = sessionmaker(
    autocommit=False,
    autoflush=False,
    bind=engine)
Base.metadata.create_all(bind=engine)


def override_get_db():
    try:
        db = TestingSessionLocal()
        yield db
    finally:
        db.close()


app.dependency_overrides[get_db] = override_get_db

client = TestClient(app)


@pytest.fixture
def create_test_task():
    unique_name = f"Test Task {uuid.uuid4()}"
    test_task_payload = {
        "name": unique_name,
        "content": "This is a test task",
        "period_of_execution": "2023-12-31",
        "status": 0,
        "employee_id": None,
        "parent_id": None
    }
    response = client.post("/tasks/create/", json=test_task_payload)
    assert response.status_code == 201
    return response.json()["task"]["id"]


@pytest.fixture
def create_test_employee():
    unique_name = f"TestEmployee{uuid.uuid4()}"
    test_employee_payload = {
        "first_name": "Test",
        "last_name": "Employee",
        "name": unique_name,
        "email": f"{unique_name}@example.com"
    }

    response = client.post("/employees/create", json=test_employee_payload)
    assert response.status_code == 201
    return response.json()["employee"]["id"]


# Удаление файла test.db посл завершении тестов
test_db_file = "test.db"


def delete_test_db_file_after_tests():
    os.system(f'del test.db')


if __name__ == '__main__':
    atexit.register(delete_test_db_file_after_tests)
    pytest.main()