63 lines
2.2 KiB
Python
63 lines
2.2 KiB
Python
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
|