Тесты для бекенда

backend/app/tests/conftest.py

@@ -0,0 +1,66 @@
+import os
+import pytest
+from sqlalchemy import create_engine
+from sqlalchemy.orm import sessionmaker
+from fastapi.testclient import TestClient
+from ..database import Base, get_db
+from ..main import app
+from ..utils.jwt import create_and_save_token
+from ..crud import employees
+
+# Получаем URL базы данных из переменной окружения или используем значение по умолчанию
+SQLALCHEMY_DATABASE_URL = os.getenv(
+    "DATABASE_URL",
+    "postgresql://postgres:postgres@localhost:5432/support_test"
+)
+
+engine = create_engine(SQLALCHEMY_DATABASE_URL)
+TestingSessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
+
+@pytest.fixture(scope="function")
+def test_db():
+    # Создаем таблицы
+    Base.metadata.create_all(bind=engine)
+    
+    # Создаем сессию
+    db = TestingSessionLocal()
+    try:
+        yield db
+    finally:
+        db.close()
+        # Очищаем таблицы после каждого теста
+        Base.metadata.drop_all(bind=engine)
+
+@pytest.fixture(scope="function")
+def test_employee(test_db):
+    employee_data = {
+        "first_name": "Test",
+        "last_name": "User",
+        "department": "IT",
+        "office": "101",
+        "password": "testpass123"
+    }
+    employee = employees.create_employee(test_db, employee_data)
+    return employee
+
+@pytest.fixture(scope="function")
+def test_token(test_db, test_employee):
+    return create_and_save_token(test_employee.id, test_db)
+
+@pytest.fixture(scope="function")
+def admin_token(test_db):
+    return create_and_save_token(-1, test_db)  # -1 для админа
+
+@pytest.fixture(scope="function")
+def test_employee_id(test_employee):
+    return test_employee.id
+
+# Переопределяем зависимость для получения БД
+def override_get_db():
+    db = TestingSessionLocal()
+    try:
+        yield db
+    finally:
+        db.close()
+
+app.dependency_overrides[get_db] = override_get_db 

backend/app/tests/test_auth.py

@@ -0,0 +1,83 @@
+import pytest
+from fastapi.testclient import TestClient
+from sqlalchemy.orm import Session
+from ..main import app
+from ..crud import employees
+from ..utils.auth import verify_password, get_password_hash
+
+client = TestClient(app)
+
+def test_login_success(test_db: Session):
+    # Создаем тестового сотрудника
+    hashed_password = get_password_hash("testpass123")
+    employee = employees.create_employee(
+        test_db,
+        {
+            "first_name": "Test",
+            "last_name": "User",
+            "department": "IT",
+            "office": "101",
+            "password": "testpass123"
+        }
+    )
+    
+    response = client.post(
+        "/api/auth/login",
+        data={
+            "username": "User",
+            "password": "testpass123"
+        }
+    )
+    
+    assert response.status_code == 200
+    assert "access_token" in response.json()
+    assert response.json()["token_type"] == "bearer"
+
+def test_login_wrong_password(test_db: Session):
+    response = client.post(
+        "/api/auth/login",
+        data={
+            "username": "User",
+            "password": "wrongpass"
+        }
+    )
+    
+    assert response.status_code == 401
+    assert "detail" in response.json()
+
+def test_login_nonexistent_user(test_db: Session):
+    response = client.post(
+        "/api/auth/login",
+        data={
+            "username": "NonExistent",
+            "password": "testpass123"
+        }
+    )
+    
+    assert response.status_code == 401
+    assert "detail" in response.json()
+
+def test_admin_login_success():
+    response = client.post(
+        "/api/auth/admin/login",
+        data={
+            "username": "admin",
+            "password": "admin123"
+        }
+    )
+    
+    assert response.status_code == 200
+    assert "access_token" in response.json()
+    assert response.json()["token_type"] == "bearer"
+
+def test_admin_login_wrong_password():
+    response = client.post(
+        "/api/auth/admin/login",
+        data={
+            "username": "admin",
+            "password": "wrongpass"
+        }
+    )
+    
+    assert response.status_code == 401
+    assert "detail" in response.json() 

backend/app/tests/test_employees.py

@@ -0,0 +1,150 @@
+import pytest
+from fastapi.testclient import TestClient
+from sqlalchemy.orm import Session
+from ..main import app
+from ..crud import employees
+from ..utils.auth import verify_password
+
+client = TestClient(app)
+
+def test_create_employee(test_db: Session, admin_token: str):
+    employee_data = {
+        "first_name": "John",
+        "last_name": "Doe",
+        "department": "IT",
+        "office": "101",
+        "password": "testpass123"
+    }
+    
+    response = client.post(
+        "/api/employees/",
+        json=employee_data,
+        headers={"Authorization": f"Bearer {admin_token}"}
+    )
+    
+    assert response.status_code == 200
+    data = response.json()
+    assert data["first_name"] == employee_data["first_name"]
+    assert data["last_name"] == employee_data["last_name"]
+    assert data["department"] == employee_data["department"]
+    assert data["office"] == employee_data["office"]
+    assert "id" in data
+
+def test_create_employee_unauthorized():
+    employee_data = {
+        "first_name": "John",
+        "last_name": "Doe",
+        "department": "IT",
+        "office": "101",
+        "password": "testpass123"
+    }
+    
+    response = client.post(
+        "/api/employees/",
+        json=employee_data
+    )
+    
+    assert response.status_code == 401
+
+def test_get_employees(test_db: Session, admin_token: str):
+    # Создаем несколько тестовых сотрудников
+    for i in range(3):
+        employees.create_employee(
+            test_db,
+            {
+                "first_name": f"Test{i}",
+                "last_name": f"User{i}",
+                "department": "IT",
+                "office": f"10{i}",
+                "password": "testpass123"
+            }
+        )
+    
+    response = client.get(
+        "/api/employees/",
+        headers={"Authorization": f"Bearer {admin_token}"}
+    )
+    
+    assert response.status_code == 200
+    data = response.json()
+    assert len(data) >= 3
+
+def test_get_employee_by_id(test_db: Session, admin_token: str):
+    # Создаем тестового сотрудника
+    employee = employees.create_employee(
+        test_db,
+        {
+            "first_name": "Test",
+            "last_name": "User",
+            "department": "IT",
+            "office": "101",
+            "password": "testpass123"
+        }
+    )
+    
+    response = client.get(
+        f"/api/employees/{employee.id}",
+        headers={"Authorization": f"Bearer {admin_token}"}
+    )
+    
+    assert response.status_code == 200
+    data = response.json()
+    assert data["id"] == employee.id
+    assert data["first_name"] == employee.first_name
+    assert data["last_name"] == employee.last_name
+
+def test_update_employee(test_db: Session, admin_token: str):
+    # Создаем тестового сотрудника
+    employee = employees.create_employee(
+        test_db,
+        {
+            "first_name": "Test",
+            "last_name": "User",
+            "department": "IT",
+            "office": "101",
+            "password": "testpass123"
+        }
+    )
+    
+    update_data = {
+        "department": "HR",
+        "office": "202"
+    }
+    
+    response = client.put(
+        f"/api/employees/{employee.id}",
+        json=update_data,
+        headers={"Authorization": f"Bearer {admin_token}"}
+    )
+    
+    assert response.status_code == 200
+    data = response.json()
+    assert data["department"] == update_data["department"]
+    assert data["office"] == update_data["office"]
+
+def test_delete_employee(test_db: Session, admin_token: str):
+    # Создаем тестового сотрудника
+    employee = employees.create_employee(
+        test_db,
+        {
+            "first_name": "Test",
+            "last_name": "User",
+            "department": "IT",
+            "office": "101",
+            "password": "testpass123"
+        }
+    )
+    
+    response = client.delete(
+        f"/api/employees/{employee.id}",
+        headers={"Authorization": f"Bearer {admin_token}"}
+    )
+    
+    assert response.status_code == 200
+    
+    # Проверяем, что сотрудник удален
+    get_response = client.get(
+        f"/api/employees/{employee.id}",
+        headers={"Authorization": f"Bearer {admin_token}"}
+    )
+    assert get_response.status_code == 404 

backend/app/tests/test_requests.py

@@ -0,0 +1,114 @@
+import pytest
+from fastapi.testclient import TestClient
+from sqlalchemy.orm import Session
+from ..main import app
+from ..crud import requests, employees
+from ..models.request import RequestStatus
+
+client = TestClient(app)
+
+def test_create_request(test_db: Session, test_token: str):
+    request_data = {
+        "title": "Test Request",
+        "description": "Test Description",
+        "priority": "low",
+        "status": "new"
+    }
+    
+    response = client.post(
+        "/api/requests/",
+        json=request_data,
+        headers={"Authorization": f"Bearer {test_token}"}
+    )
+    
+    assert response.status_code == 200
+    data = response.json()
+    assert data["title"] == request_data["title"]
+    assert data["description"] == request_data["description"]
+    assert data["priority"] == request_data["priority"]
+    assert data["status"] == RequestStatus.NEW.value
+
+def test_create_request_unauthorized():
+    request_data = {
+        "title": "Test Request",
+        "description": "Test Description",
+        "priority": "low"
+    }
+    
+    response = client.post(
+        "/api/requests/",
+        json=request_data
+    )
+    
+    assert response.status_code == 401
+
+def test_get_employee_requests(test_db: Session, test_token: str, test_employee_id: int):
+    # Создаем несколько тестовых заявок
+    for i in range(3):
+        requests.create_request(
+            test_db,
+            {
+                "title": f"Test Request {i}",
+                "description": f"Test Description {i}",
+                "priority": "low",
+                "status": "new"
+            },
+            test_employee_id
+        )
+    
+    response = client.get(
+        "/api/requests/",
+        headers={"Authorization": f"Bearer {test_token}"}
+    )
+    
+    assert response.status_code == 200
+    data = response.json()
+    assert len(data) == 3
+    assert all(req["employee_id"] == test_employee_id for req in data)
+
+def test_update_request_status(test_db: Session, admin_token: str):
+    # Создаем тестовую заявку
+    employee = employees.create_employee(
+        test_db,
+        {
+            "first_name": "Test",
+            "last_name": "User",
+            "department": "IT",
+            "office": "101",
+            "password": "testpass123"
+        }
+    )
+    
+    request = requests.create_request(
+        test_db,
+        {
+            "title": "Test Request",
+            "description": "Test Description",
+            "priority": "low",
+            "status": "new"
+        },
+        employee.id
+    )
+    
+    response = client.put(
+        f"/api/requests/{request.id}",
+        json={"status": "in_progress"},
+        headers={"Authorization": f"Bearer {admin_token}"}
+    )
+    
+    assert response.status_code == 200
+    assert response.json()["status"] == RequestStatus.IN_PROGRESS.value
+
+def test_get_request_statistics(test_db: Session, admin_token: str):
+    response = client.get(
+        "/api/requests/statistics",
+        headers={"Authorization": f"Bearer {admin_token}"}
+    )
+    
+    assert response.status_code == 200
+    data = response.json()
+    assert "total" in data
+    assert "new" in data
+    assert "in_progress" in data
+    assert "completed" in data
+    assert "rejected" in data