AdministrationItDepartmens/backend/tests/test_employees.py

"""Employee tests."""
import pytest
from fastapi.testclient import TestClient
from sqlalchemy.orm import Session
from app.models.employee import Employee

def test_create_employee(client: TestClient, admin_token: str, db: Session):
    """Тест создания сотрудника."""
    response = client.post(
        "/api/employees",
        headers={"Authorization": f"Bearer {admin_token}"},
        json={
            "first_name": "New",
            "last_name": "Employee",
            "department": "IT",
            "office": "102",
            "password": "newpassword"
        }
    )
    assert response.status_code == 201
    data = response.json()
    assert data["first_name"] == "New"
    assert data["last_name"] == "Employee"
    assert data["department"] == "IT"
    assert data["office"] == "102"
    assert "id" in data

def test_create_employee_unauthorized(client: TestClient):
    """Тест создания сотрудника без авторизации."""
    response = client.post(
        "/api/employees",
        json={
            "first_name": "New",
            "last_name": "Employee",
            "department": "IT",
            "office": "102",
            "password": "newpassword"
        }
    )
    assert response.status_code == 401
    assert response.json()["detail"] == "Not authenticated"

def test_get_employees(client: TestClient, admin_token: str, test_employee: Employee, db: Session):
    """Тест получения списка сотрудников."""
    response = client.get(
        "/api/employees",
        headers={"Authorization": f"Bearer {admin_token}"}
    )
    assert response.status_code == 200
    data = response.json()
    assert isinstance(data, list)
    assert len(data) > 0
    assert "first_name" in data[0]
    assert "last_name" in data[0]
    assert "department" in data[0]
    assert "office" in data[0]

def test_get_employee_by_id(client: TestClient, admin_token: str, test_employee: Employee, db: Session):
    """Тест получения сотрудника по ID."""
    response = client.get(
        f"/api/employees/{test_employee.id}",
        headers={"Authorization": f"Bearer {admin_token}"}
    )
    assert response.status_code == 200
    data = response.json()
    assert data["first_name"] == test_employee.first_name
    assert data["last_name"] == test_employee.last_name
    assert data["department"] == test_employee.department
    assert data["office"] == test_employee.office

def test_get_nonexistent_employee(client: TestClient, admin_token: str):
    """Тест получения несуществующего сотрудника."""
    response = client.get(
        "/api/employees/999",
        headers={"Authorization": f"Bearer {admin_token}"}
    )
    assert response.status_code == 404
    assert response.json()["detail"] == "Employee not found"

def test_update_employee(client: TestClient, admin_token: str, test_employee: Employee, db: Session):
    """Тест обновления данных сотрудника."""
    response = client.put(
        f"/api/employees/{test_employee.id}",
        headers={"Authorization": f"Bearer {admin_token}"},
        json={
            "first_name": "Updated",
            "last_name": "Name",
            "department": "HR",
            "office": "103"
        }
    )
    assert response.status_code == 200
    data = response.json()
    assert data["first_name"] == "Updated"
    assert data["last_name"] == "Name"
    assert data["department"] == "HR"
    assert data["office"] == "103"

def test_delete_employee(client: TestClient, admin_token: str, test_employee: Employee, db: Session):
    """Тест удаления сотрудника."""
    response = client.delete(
        f"/api/employees/{test_employee.id}",
        headers={"Authorization": f"Bearer {admin_token}"}
    )
    assert response.status_code == 200
    data = response.json()
    assert data["first_name"] == test_employee.first_name
    assert data["last_name"] == test_employee.last_name
    assert data["department"] == test_employee.department
    assert data["office"] == test_employee.office

def test_employee_me(client: TestClient, employee_token: str, test_employee: Employee, db: Session):
    """Тест получения информации о текущем сотруднике."""
    response = client.get(
        "/api/employees/me",
        headers={"Authorization": f"Bearer {employee_token}"}
    )
    assert response.status_code == 200
    data = response.json()
    assert data["first_name"] == test_employee.first_name
    assert data["last_name"] == test_employee.last_name
    assert data["department"] == test_employee.department
    assert data["office"] == test_employee.office

def test_update_me(client: TestClient, employee_token: str, test_employee: Employee, db: Session):
    """Тест обновления информации о текущем сотруднике."""
    response = client.put(
        "/api/employees/me",
        headers={"Authorization": f"Bearer {employee_token}"},
        json={
            "first_name": "Updated",
            "last_name": "Name",
            "department": "Support",
            "office": "104"
        }
    )
    assert response.status_code == 200
    data = response.json()
    assert data["first_name"] == "Updated"
    assert data["last_name"] == "Name"
    assert data["department"] == "Support"
    assert data["office"] == "104"