mirror of
https://gitlab.com/MoonTestUse1/AdministrationItDepartmens.git
synced 2025-08-14 00:25:46 +02:00
Fix database
This commit is contained in:
@@ -7,16 +7,16 @@ from fastapi.testclient import TestClient
|
||||
from typing import Generator, Any
|
||||
|
||||
# Устанавливаем переменную окружения для тестов
|
||||
os.environ["TESTING"] = "True"
|
||||
os.environ["TESTING"] = "1"
|
||||
|
||||
from app.database import Base
|
||||
from app.main import app
|
||||
from app.core.test_config import test_settings
|
||||
from app.dependencies import get_db
|
||||
from .fixtures import * # импортируем все фикстуsры
|
||||
from .test_fixtures import * # импортируем все фикстуры
|
||||
|
||||
# Создаем тестовый движок базы данных
|
||||
engine = create_engine(test_settings.DATABASE_URL)
|
||||
engine = create_engine(test_settings.get_database_url())
|
||||
|
||||
# Создаем тестовую фабрику сессий
|
||||
TestingSessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
|
||||
@@ -25,7 +25,7 @@ TestingSessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engin
|
||||
def setup_test_db() -> Generator[None, Any, None]:
|
||||
"""Setup test database"""
|
||||
# Пробуем создать базу данных test_app
|
||||
default_engine = create_engine("postgresql://postgres:postgres@localhost:5432/postgres")
|
||||
default_engine = create_engine(f"postgresql://{test_settings.POSTGRES_USER}:{test_settings.POSTGRES_PASSWORD}@{test_settings.POSTGRES_HOST}:{test_settings.POSTGRES_PORT}/postgres")
|
||||
with default_engine.connect() as conn:
|
||||
conn.execute(text("COMMIT")) # Завершаем текущую транзакцию
|
||||
try:
|
||||
|
||||
77
backend/tests/test_fixtures.py
Normal file
77
backend/tests/test_fixtures.py
Normal file
@@ -0,0 +1,77 @@
|
||||
"""Test fixtures"""
|
||||
import pytest
|
||||
from fastapi.testclient import TestClient
|
||||
from sqlalchemy.orm import Session
|
||||
|
||||
from app.crud import employees
|
||||
from app.schemas.employee import EmployeeCreate
|
||||
from app.utils.auth import get_password_hash
|
||||
from app.models.employee import Employee
|
||||
|
||||
@pytest.fixture(scope="function")
|
||||
def test_employee(db_session: Session) -> Employee:
|
||||
"""Create test employee"""
|
||||
# Удаляем существующего сотрудника, если есть
|
||||
db_session.query(Employee).filter(
|
||||
Employee.first_name == "Test",
|
||||
Employee.last_name == "User"
|
||||
).delete()
|
||||
db_session.commit()
|
||||
|
||||
employee = EmployeeCreate(
|
||||
first_name="Test",
|
||||
last_name="User",
|
||||
department="IT",
|
||||
office="101",
|
||||
password="testpass123",
|
||||
is_admin=False
|
||||
)
|
||||
hashed_password = get_password_hash(employee.password)
|
||||
db_employee = employees.create_employee(db_session, employee, hashed_password)
|
||||
return db_employee
|
||||
|
||||
@pytest.fixture(scope="function")
|
||||
def test_admin(db_session: Session) -> Employee:
|
||||
"""Create test admin"""
|
||||
# Удаляем существующего админа, если есть
|
||||
db_session.query(Employee).filter(
|
||||
Employee.first_name == "Admin",
|
||||
Employee.last_name == "User"
|
||||
).delete()
|
||||
db_session.commit()
|
||||
|
||||
admin = EmployeeCreate(
|
||||
first_name="Admin",
|
||||
last_name="User",
|
||||
department="IT",
|
||||
office="102",
|
||||
password="adminpass123",
|
||||
is_admin=True
|
||||
)
|
||||
hashed_password = get_password_hash(admin.password)
|
||||
db_admin = employees.create_employee(db_session, admin, hashed_password)
|
||||
return db_admin
|
||||
|
||||
@pytest.fixture(scope="function")
|
||||
def employee_token(client: TestClient, test_employee: Employee) -> str:
|
||||
"""Get employee token"""
|
||||
response = client.post(
|
||||
"/api/auth/login",
|
||||
data={
|
||||
"username": f"{test_employee.first_name} {test_employee.last_name}",
|
||||
"password": "testpass123"
|
||||
}
|
||||
)
|
||||
return response.json()["access_token"]
|
||||
|
||||
@pytest.fixture(scope="function")
|
||||
def admin_token(client: TestClient, test_admin: Employee) -> str:
|
||||
"""Get admin token"""
|
||||
response = client.post(
|
||||
"/api/auth/admin/login",
|
||||
data={
|
||||
"username": f"{test_admin.first_name} {test_admin.last_name}",
|
||||
"password": "adminpass123"
|
||||
}
|
||||
)
|
||||
return response.json()["access_token"]
|
||||
Reference in New Issue
Block a user