from __future__ import annotations

from sqlalchemy import create_engine, func, select
from sqlalchemy.orm import Session, sessionmaker
from sqlalchemy.pool import StaticPool

from app.db.base import Base
from app.models.employee import Employee
from app.models.employee_change_log import EmployeeChangeLog
from app.models.organization import OrganizationUnit
from app.models.role import Role
from app.services.employee import EmployeeService


def build_session() -> Session:
    engine = create_engine(
        "sqlite+pysqlite:///:memory:",
        connect_args={"check_same_thread": False},
        poolclass=StaticPool,
    )
    Base.metadata.create_all(bind=engine)
    session_factory = sessionmaker(bind=engine, autoflush=False, autocommit=False)
    return session_factory()


def test_employee_directory_seeds_rich_employee_data() -> None:
    with build_session() as db:
        service = EmployeeService(db)

        employees = service.list_employees()
        meta = service.get_employee_meta()

        assert len(employees) == 30
        assert meta.totalEmployees == 30
        assert any(item.status == "试用中" for item in employees)
        assert any(item.status == "停用" for item in employees)
        assert any("审批负责人" in item.roles for item in employees)
        assert any(item.permissions for item in employees)
        assert any(item.history for item in employees)

        role_count = db.scalar(select(func.count()).select_from(Role))
        org_count = db.scalar(select(func.count()).select_from(OrganizationUnit))
        employee_count = db.scalar(select(func.count()).select_from(Employee))
        history_count = db.scalar(select(func.count()).select_from(EmployeeChangeLog))

        assert role_count == 6
        assert org_count == 10
        assert employee_count == 30
        assert history_count and history_count >= 30


def test_employee_detail_contains_department_and_roles() -> None:
    with build_session() as db:
        service = EmployeeService(db)
        employee = service.list_employees()[0]
        detail = service.get_employee(employee.id)

        assert detail is not None
        assert detail.department
        assert detail.manager
        assert detail.organization is not None
        assert detail.roles