70 lines
2.6 KiB
Python
70 lines
2.6 KiB
Python
"""
|
||
Тесты для проверки функций работы с базой данных
|
||
"""
|
||
import pytest
|
||
import time
|
||
from sqlalchemy import create_engine, Column, Integer, String, inspect
|
||
from sqlalchemy.orm import declarative_base, Session
|
||
|
||
from services.db import create_table_if_not_exists, get_column_names_without_virtual, local_session
|
||
|
||
# Создаем базовую модель для тестирования
|
||
Base = declarative_base()
|
||
|
||
class TestModel(Base):
|
||
"""Тестовая модель для проверки функций базы данных"""
|
||
__tablename__ = 'test_model'
|
||
|
||
id = Column(Integer, primary_key=True)
|
||
name = Column(String)
|
||
description = Column(String, nullable=True)
|
||
|
||
class TestDatabaseFunctions:
|
||
"""Тесты для функций работы с базой данных"""
|
||
|
||
def test_create_table_if_not_exists(self, tmp_path):
|
||
"""
|
||
Проверка создания таблицы, если она не существует
|
||
"""
|
||
# Создаем временную базу данных SQLite
|
||
db_path = tmp_path / "test.db"
|
||
engine = create_engine(f"sqlite:///{db_path}")
|
||
Base.metadata.create_all(engine)
|
||
|
||
# Создаем таблицу
|
||
create_table_if_not_exists(engine, TestModel)
|
||
|
||
# Проверяем, что таблица создана
|
||
inspector = inspect(engine)
|
||
assert inspector.has_table('test_model')
|
||
|
||
def test_get_column_names_without_virtual(self):
|
||
"""
|
||
Проверка получения имен колонок без виртуальных полей
|
||
"""
|
||
columns = get_column_names_without_virtual(TestModel)
|
||
|
||
# Ожидаем, что будут только реальные колонки
|
||
assert set(columns) == {'id', 'name', 'description'}
|
||
|
||
def test_local_session_management(self):
|
||
"""
|
||
Проверка создания и управления локальной сессией
|
||
"""
|
||
# Создаем сессию
|
||
session = local_session()
|
||
|
||
try:
|
||
# Проверяем, что сессия создана корректно
|
||
assert isinstance(session, Session)
|
||
|
||
# Проверяем, что сессия работает с существующими таблицами
|
||
# Используем Author вместо TestModel
|
||
from auth.orm import Author
|
||
authors_count = session.query(Author).count()
|
||
assert isinstance(authors_count, int)
|
||
|
||
finally:
|
||
# Всегда закрываем сессию
|
||
session.close()
|