339 lines
9.6 KiB
Python
339 lines
9.6 KiB
Python
# tests/conftest.py
|
||
"""
|
||
Общие pytest fixtures для всех тестов проекта.
|
||
|
||
Этот файл содержит переиспользуемые фикстуры для:
|
||
- Создания пользователей, департаментов, ролей
|
||
- API клиентов (аутентифицированных и нет)
|
||
- Тестовых данных (картинки, файлы)
|
||
- Вспомогательных функций (извлечение результатов пагинации и т.д.)
|
||
"""
|
||
|
||
import itertools
|
||
|
||
import pytest
|
||
from django.contrib.auth import get_user_model
|
||
from django.contrib.auth.models import Permission
|
||
from django.contrib.contenttypes.models import ContentType
|
||
from django.core.files.base import ContentFile
|
||
from django.db import connections
|
||
from rest_framework.test import APIClient
|
||
|
||
from employees.models import Department, DepartmentRole, EmployeeDepartment
|
||
from tests.test_config import (
|
||
DEFAULT_PASSWORD,
|
||
DEFAULT_TEST_PHONE_PREFIX,
|
||
TEST_EMAIL_DOMAIN,
|
||
TEST_IMAGE_DATA_URI,
|
||
)
|
||
|
||
pytestmark = pytest.mark.django_db
|
||
|
||
User = get_user_model()
|
||
|
||
# -------------------------
|
||
# Counters & generators
|
||
# -------------------------
|
||
|
||
_phone_seq = itertools.count(1000)
|
||
_email_seq = itertools.count(1)
|
||
|
||
|
||
def _unique_phone() -> str:
|
||
"""Генерирует уникальный валидный E.164 номер телефона для тестов."""
|
||
return f"{DEFAULT_TEST_PHONE_PREFIX}{next(_phone_seq):04d}"
|
||
|
||
|
||
def _unique_email(prefix: str = "user") -> str:
|
||
"""Генерирует уникальный email для тестов."""
|
||
return f"{prefix}{next(_email_seq)}@{TEST_EMAIL_DOMAIN}"
|
||
|
||
|
||
# -------------------------
|
||
# Speed & isolation
|
||
# -------------------------
|
||
|
||
|
||
@pytest.fixture(scope="session", autouse=True)
|
||
def _flush_postgres_legacy_test_tables(django_db_setup):
|
||
"""Flush FK-linked historical tables only inside disposable test databases.
|
||
|
||
Old notification migrations left tables absent from Django's current model
|
||
state. PostgreSQL otherwise refuses every transactional test's TRUNCATE.
|
||
Keep normal post_migrate/permission recreation, unlike available_apps.
|
||
This changes test cleanup only; no production migrations or tables change.
|
||
"""
|
||
with pytest.MonkeyPatch.context() as patcher:
|
||
for test_connection in connections.all(initialized_only=True):
|
||
if test_connection.vendor != "postgresql":
|
||
continue
|
||
original_flush = test_connection.ops.sql_flush
|
||
|
||
def isolated_flush(
|
||
style,
|
||
tables,
|
||
*,
|
||
reset_sequences=False,
|
||
allow_cascade=False,
|
||
_connection=test_connection,
|
||
_original=original_flush,
|
||
):
|
||
database_name = str(_connection.settings_dict.get("NAME", ""))
|
||
if not database_name.startswith("test_"):
|
||
raise RuntimeError(
|
||
"Refusing cascading test cleanup outside a test_ database"
|
||
)
|
||
return _original(
|
||
style,
|
||
tables,
|
||
reset_sequences=reset_sequences,
|
||
allow_cascade=True,
|
||
)
|
||
|
||
patcher.setattr(test_connection.ops, "sql_flush", isolated_flush)
|
||
yield
|
||
|
||
|
||
@pytest.fixture(autouse=True)
|
||
def _fast_hashers_and_email(settings, tmp_path):
|
||
# быстрые хеши паролей и локальная почта
|
||
settings.PASSWORD_HASHERS = ["django.contrib.auth.hashers.MD5PasswordHasher"]
|
||
settings.EMAIL_BACKEND = "django.core.mail.backends.locmem.EmailBackend"
|
||
# изоляция медиа
|
||
media_root = tmp_path / "test_media"
|
||
media_root.mkdir(exist_ok=True)
|
||
settings.MEDIA_ROOT = media_root
|
||
yield
|
||
|
||
|
||
# -------------------------
|
||
# API clients
|
||
# -------------------------
|
||
|
||
|
||
@pytest.fixture
|
||
def api_client() -> APIClient:
|
||
"""DRF APIClient (имеет force_authenticate)."""
|
||
return APIClient()
|
||
|
||
|
||
@pytest.fixture
|
||
def auth_client_factory():
|
||
"""Фабрика аутентифицированных клиентов: client = auth_client_factory(user)."""
|
||
|
||
def _make(user=None):
|
||
client = APIClient()
|
||
if user:
|
||
client.force_authenticate(user=user)
|
||
return client
|
||
|
||
return _make
|
||
|
||
|
||
# -------------------------
|
||
# Factories
|
||
# -------------------------
|
||
|
||
|
||
@pytest.fixture
|
||
def user_factory():
|
||
"""
|
||
Фабрика для создания пользователей.
|
||
|
||
Пример использования:
|
||
user = user_factory(email="x@example.com", staff=False, superuser=False,
|
||
verified=True, active=True, **extra)
|
||
|
||
Создаёт пользователя напрямую (без менеджера create_user, чтобы не слать письма).
|
||
"""
|
||
|
||
def _make(
|
||
email: str = None,
|
||
*,
|
||
staff: bool = False,
|
||
superuser: bool = False,
|
||
verified: bool = True,
|
||
active: bool = True,
|
||
password: str = DEFAULT_PASSWORD,
|
||
**extra,
|
||
):
|
||
if email is None:
|
||
email = _unique_email()
|
||
|
||
u = User.objects.create(
|
||
email=email,
|
||
phone_number=extra.pop("phone_number", _unique_phone()),
|
||
first_name=extra.pop("first_name", "FN"),
|
||
last_name=extra.pop("last_name", "LN"),
|
||
is_staff=staff,
|
||
is_superuser=superuser,
|
||
is_active=active,
|
||
email_verified=verified,
|
||
**extra,
|
||
)
|
||
u.set_password(password)
|
||
u.save()
|
||
return u
|
||
|
||
return _make
|
||
|
||
|
||
@pytest.fixture
|
||
def department_factory():
|
||
def _make(**kwargs) -> Department:
|
||
return Department.objects.create(**kwargs)
|
||
|
||
return _make
|
||
|
||
|
||
@pytest.fixture
|
||
def perm_for_department():
|
||
"""
|
||
p = perm_for_department("manage_department")
|
||
Гарантированно возвращает Permission для модели Department с нужным codename.
|
||
"""
|
||
|
||
def _make(codename: str) -> Permission:
|
||
ct = ContentType.objects.get_for_model(Department)
|
||
p, _ = Permission.objects.get_or_create(
|
||
content_type=ct,
|
||
codename=codename,
|
||
defaults={"name": codename},
|
||
)
|
||
return p
|
||
|
||
return _make
|
||
|
||
|
||
@pytest.fixture
|
||
def role_factory(perm_for_department):
|
||
"""
|
||
role = role_factory(dept, name="mgr", codes=["manage_department", ...])
|
||
"""
|
||
|
||
def _make(dept: Department, name: str = "role", codes=None) -> DepartmentRole:
|
||
r = DepartmentRole.objects.create(department=dept, name=name)
|
||
if codes:
|
||
perms = [perm_for_department(code) for code in codes]
|
||
r.permissions.add(*perms)
|
||
return r
|
||
|
||
return _make
|
||
|
||
|
||
@pytest.fixture
|
||
def link_factory():
|
||
"""
|
||
link = link_factory(employee, department, is_active=True, role=None, **dates)
|
||
"""
|
||
|
||
def _make(
|
||
employee, department, is_active=True, role=None, **dates
|
||
) -> EmployeeDepartment:
|
||
return EmployeeDepartment.objects.create(
|
||
employee=employee,
|
||
department=department,
|
||
is_active=is_active,
|
||
role=role,
|
||
**dates,
|
||
)
|
||
|
||
return _make
|
||
|
||
|
||
# -------------------------
|
||
# Test helpers
|
||
# -------------------------
|
||
|
||
|
||
@pytest.fixture
|
||
def extract_results():
|
||
"""
|
||
items = extract_results(resp.json()) — поддерживает DRF пагинацию и «плоский» список.
|
||
"""
|
||
|
||
def _do(payload):
|
||
if isinstance(payload, dict) and "results" in payload:
|
||
return payload["results"]
|
||
return payload
|
||
|
||
return _do
|
||
|
||
|
||
# -------------------------
|
||
# Small misc utilities
|
||
# -------------------------
|
||
|
||
|
||
@pytest.fixture
|
||
def data_uri_image(tmp_path):
|
||
"""
|
||
Возвращает data URI маленькой PNG-картинки (1x1), удобно для аватаров.
|
||
"""
|
||
return TEST_IMAGE_DATA_URI
|
||
|
||
|
||
@pytest.fixture
|
||
def content_file_from_b64():
|
||
"""
|
||
ContentFile <- base64 строки (без data: префикса).
|
||
"""
|
||
import base64
|
||
|
||
def _make(b64: str, name: str = "upload.bin") -> ContentFile:
|
||
return ContentFile(base64.b64decode(b64), name=name)
|
||
|
||
return _make
|
||
|
||
|
||
# -------------------------
|
||
# Дополнительные утилиты
|
||
# -------------------------
|
||
|
||
|
||
@pytest.fixture
|
||
def make_user():
|
||
"""
|
||
Альтернативная функция создания пользователя (прямой вызов, не через фабрику).
|
||
Используется в старых тестах для совместимости.
|
||
"""
|
||
|
||
def _make(
|
||
email: str,
|
||
*,
|
||
staff: bool = False,
|
||
superuser: bool = False,
|
||
verified: bool = True,
|
||
active: bool = True,
|
||
password: str = DEFAULT_PASSWORD,
|
||
**extra,
|
||
):
|
||
u = User.objects.create(
|
||
email=email,
|
||
phone_number=extra.pop("phone_number", _unique_phone()),
|
||
first_name=extra.pop("first_name", "FN"),
|
||
last_name=extra.pop("last_name", "LN"),
|
||
is_staff=staff,
|
||
is_superuser=superuser,
|
||
is_active=active,
|
||
email_verified=verified,
|
||
**extra,
|
||
)
|
||
u.set_password(password)
|
||
u.save()
|
||
return u
|
||
|
||
return _make
|
||
|
||
|
||
@pytest.fixture
|
||
def unique_phone():
|
||
"""Возвращает функцию для генерации уникальных номеров телефона."""
|
||
return _unique_phone
|
||
|
||
|
||
@pytest.fixture
|
||
def unique_email():
|
||
"""Возвращает функцию для генерации уникальных email."""
|
||
return _unique_email
|