1
0
mirror of https://gitlab.com/MoonTestUse1/AdministrationItDepartmens.git synced 2025-08-14 00:25:46 +02:00

Fix tests

This commit is contained in:
MoonTestUse1
2025-01-06 23:12:12 +06:00
parent c3145d7bc4
commit dbcfa0513f
3 changed files with 31 additions and 19 deletions

View File

@@ -1,4 +1,5 @@
"""Settings configuration""" """Settings configuration"""
import os
from pydantic_settings import BaseSettings, SettingsConfigDict from pydantic_settings import BaseSettings, SettingsConfigDict
class Settings(BaseSettings): class Settings(BaseSettings):
@@ -8,7 +9,8 @@ class Settings(BaseSettings):
API_V1_STR: str = "/api" API_V1_STR: str = "/api"
# Database # Database
DATABASE_URL: str = "postgresql://postgres:postgres123@db:5432/support_db" TESTING: bool = os.getenv("TESTING", "False") == "True"
DATABASE_URL: str = "sqlite:///./test.db" if TESTING else "postgresql://postgres:postgres123@postgres:5432/support_db"
# JWT # JWT
SECRET_KEY: str = "your-secret-key" SECRET_KEY: str = "your-secret-key"

View File

@@ -2,7 +2,6 @@
from sqlalchemy import create_engine from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker from sqlalchemy.orm import sessionmaker
from .core.config import settings from .core.config import settings
from .db.base import Base
# Для создания таблиц импортируем модели # Для создания таблиц импортируем модели
from .models.employee import Employee # noqa from .models.employee import Employee # noqa
@@ -11,7 +10,13 @@ from .models.token import Token # noqa
SQLALCHEMY_DATABASE_URL = settings.DATABASE_URL SQLALCHEMY_DATABASE_URL = settings.DATABASE_URL
engine = create_engine(SQLALCHEMY_DATABASE_URL) # Для SQLite нужны специальные параметры подключения
connect_args = {"check_same_thread": False} if settings.DATABASE_URL.startswith("sqlite") else {}
engine = create_engine(
SQLALCHEMY_DATABASE_URL,
connect_args=connect_args
)
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine) SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
def get_db(): def get_db():

View File

@@ -8,16 +8,23 @@ from sqlalchemy.orm import sessionmaker
from sqlalchemy.pool import StaticPool from sqlalchemy.pool import StaticPool
from unittest.mock import Mock, patch from unittest.mock import Mock, patch
from app.database import Base, get_db from app.database import get_db
from app.main import app from app.main import app
from app.models.base import Base
from app.models.employee import Employee from app.models.employee import Employee
from app.utils.auth import get_password_hash from app.utils.auth import get_password_hash
from app.utils.jwt import create_access_token from app.utils.jwt import create_access_token
from app.core.config import settings
# Устанавливаем переменную окружения для тестов
os.environ["TESTING"] = "True"
# Создаем тестовую базу данных в памяти # Создаем тестовую базу данных в памяти
SQLALCHEMY_DATABASE_URL = "sqlite:///./test.db" SQLALCHEMY_DATABASE_URL = "sqlite:///:memory:"
engine = create_engine(SQLALCHEMY_DATABASE_URL, connect_args={"check_same_thread": False}) engine = create_engine(
SQLALCHEMY_DATABASE_URL,
connect_args={"check_same_thread": False},
poolclass=StaticPool
)
TestingSessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine) TestingSessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
class MockRedis: class MockRedis:
@@ -78,12 +85,11 @@ def client(db: TestingSessionLocal, redis_mock) -> Generator:
def test_employee(db: TestingSessionLocal) -> Employee: def test_employee(db: TestingSessionLocal) -> Employee:
"""Фикстура для создания тестового сотрудника.""" """Фикстура для создания тестового сотрудника."""
employee = Employee( employee = Employee(
email="test@example.com", first_name="Test",
full_name="Test Employee", last_name="User",
hashed_password=get_password_hash("testpassword"), department="IT",
is_active=True, office="101",
is_admin=False, hashed_password=get_password_hash("testpassword")
department="IT"
) )
db.add(employee) db.add(employee)
db.commit() db.commit()
@@ -94,12 +100,11 @@ def test_employee(db: TestingSessionLocal) -> Employee:
def test_admin(db: TestingSessionLocal) -> Employee: def test_admin(db: TestingSessionLocal) -> Employee:
"""Фикстура для создания тестового администратора.""" """Фикстура для создания тестового администратора."""
admin = Employee( admin = Employee(
email="admin@example.com", first_name="Admin",
full_name="Test Admin", last_name="User",
hashed_password=get_password_hash("adminpassword"), department="Administration",
is_active=True, office="100",
is_admin=True, hashed_password=get_password_hash("adminpassword")
department="Administration"
) )
db.add(admin) db.add(admin)
db.commit() db.commit()