Добавления тестов бекенда

backend/tests/test_requests.py

@@ -1,75 +1,333 @@
-"""Request management endpoint tests"""
 import pytest
-from unittest.mock import patch
+from app.models.request import Request, RequestStatus, RequestPriority
+from app.models.employee import Employee
+from app.utils.auth import get_password_hash
+from datetime import datetime, timedelta
 
-def test_create_request(client, test_db, test_employee, test_request):
-    """Test request creation"""
-    # Create test employee first
-    employee_response = client.post(
-        "/api/employees",
-        json={
-            "first_name": test_employee["first_name"],
-            "last_name": test_employee["last_name"],
-            "department": test_employee["department"],
-            "office": test_employee["office"],
-            "password": test_employee["password"]
-        }
+def test_create_request(client, db_session):
+    """Test creating a new request"""
+    # Create test employee
+    hashed_password = get_password_hash("test123")
+    employee = Employee(
+        first_name="Test",
+        last_name="User",
+        department="IT",
+        office="A101",
+        password=hashed_password
     )
-    assert employee_response.status_code == 200
-    employee_data = employee_response.json()
-    test_request["employee_id"] = employee_data["id"]
-    
+    db_session.add(employee)
+    db_session.commit()
+    employee_id = employee.id
+
+    # Login as employee
+    login_response = client.post("/api/auth/login", json={
+        "last_name": "User",
+        "password": "test123"
+    })
+    assert login_response.status_code == 200
+    token = login_response.json()["access_token"]
+
     # Create request
-    with patch('app.bot.notifications.send_notification'):  # Mock notification
-        response = client.post(
-            "/api/requests",
-            json=test_request
-        )
+    request_data = {
+        "title": "Test Request",
+        "description": "This is a test request",
+        "priority": RequestPriority.MEDIUM.value
+    }
+    headers = {"Authorization": f"Bearer {token}"}
+    response = client.post("/api/requests/", json=request_data, headers=headers)
     
     assert response.status_code == 200
     data = response.json()
-    assert data["employee_id"] == test_request["employee_id"]
-    assert data["status"] == "new"
+    assert data["title"] == request_data["title"]
+    assert data["description"] == request_data["description"]
+    assert data["priority"] == request_data["priority"]
+    assert data["status"] == RequestStatus.NEW.value
+    assert data["employee_id"] == employee_id
 
-def test_create_request_invalid_employee(client, test_request):
-    """Test creating request with invalid employee ID"""
-    test_request["employee_id"] = 999  # Non-existent ID
-    
-    response = client.post(
-        "/api/requests",
-        json=test_request
+def test_get_employee_requests(client, db_session):
+    """Test getting employee's requests"""
+    # Create test employee
+    hashed_password = get_password_hash("test123")
+    employee = Employee(
+        first_name="Test",
+        last_name="User",
+        department="IT",
+        office="A101",
+        password=hashed_password
     )
-    
-    assert response.status_code == 404
-    assert "не найден" in response.json()["detail"]
+    db_session.add(employee)
+    db_session.commit()
+    employee_id = employee.id
 
-def test_create_request_invalid_priority(client, test_db, test_employee):
-    """Test creating request with invalid priority"""
-    # Create test employee first
-    employee_response = client.post(
-        "/api/employees",
-        json={
-            "first_name": test_employee["first_name"],
-            "last_name": test_employee["last_name"],
-            "department": test_employee["department"],
-            "office": test_employee["office"],
-            "password": test_employee["password"]
-        }
+    # Create test request and save its data
+    request = Request(
+        title="Test Request",
+        description="This is a test request",
+        priority=RequestPriority.MEDIUM.value,
+        status=RequestStatus.NEW.value,
+        employee_id=employee_id
     )
-    assert employee_response.status_code == 200
-    employee_data = employee_response.json()
+    db_session.add(request)
+    db_session.commit()
     
-    invalid_request = {
-        "employee_id": employee_data["id"],
-        "department": "general",
-        "request_type": "hardware",
-        "priority": "invalid",  # Invalid priority
-        "description": "Test request"
+    # Сохраняем данные для сравнения
+    expected_data = {
+        "title": request.title,
+        "description": request.description,
+        "priority": request.priority,
+        "status": request.status,
+        "employee_id": request.employee_id
     }
+
+    # Login as employee
+    login_response = client.post("/api/auth/login", json={
+        "last_name": "User",
+        "password": "test123"
+    })
+    assert login_response.status_code == 200
+    token = login_response.json()["access_token"]
+
+    # Get requests
+    headers = {"Authorization": f"Bearer {token}"}
+    response = client.get("/api/requests/my", headers=headers)
     
-    response = client.post(
-        "/api/requests",
-        json=invalid_request
+    assert response.status_code == 200
+    data = response.json()
+    assert len(data) == 1
+    assert data[0]["title"] == expected_data["title"]
+    assert data[0]["description"] == expected_data["description"]
+    assert data[0]["priority"] == expected_data["priority"]
+    assert data[0]["status"] == expected_data["status"]
+    assert data[0]["employee_id"] == expected_data["employee_id"]
+
+def test_update_request_status(client, db_session):
+    """Test updating request status"""
+    # Create test employee and request
+    hashed_password = get_password_hash("test123")
+    employee = Employee(
+        first_name="Test",
+        last_name="User",
+        department="IT",
+        office="A101",
+        password=hashed_password
     )
+    db_session.add(employee)
+    db_session.commit()
+    employee_id = employee.id
+
+    request = Request(
+        title="Test Request",
+        description="This is a test request",
+        priority=RequestPriority.MEDIUM.value,
+        status=RequestStatus.NEW.value,
+        employee_id=employee_id
+    )
+    db_session.add(request)
+    db_session.commit()
+    request_id = request.id
+
+    # Login as admin
+    admin_response = client.post("/api/auth/admin", json={
+        "username": "admin",
+        "password": "admin123"
+    })
+    assert admin_response.status_code == 200
+    admin_token = admin_response.json()["access_token"]
+
+    # Update request status
+    update_data = {"status": RequestStatus.IN_PROGRESS.value}
+    headers = {"Authorization": f"Bearer {admin_token}"}
+    response = client.patch(f"/api/requests/{request_id}/status", json=update_data, headers=headers)
+
+    assert response.status_code == 200
+    data = response.json()
+    assert data["status"] == RequestStatus.IN_PROGRESS.value
+
+def test_get_all_requests_admin(client, db_session):
+    """Test getting all requests as admin"""
+    # Create test employees and requests
+    hashed_password = get_password_hash("test123")
+    employee1 = Employee(
+        first_name="Test1",
+        last_name="User1",
+        department="IT",
+        office="A101",
+        password=hashed_password
+    )
+    employee2 = Employee(
+        first_name="Test2",
+        last_name="User2",
+        department="HR",
+        office="B202",
+        password=hashed_password
+    )
+    db_session.add_all([employee1, employee2])
+    db_session.commit()
+
+    request1 = Request(
+        title="Test Request 1",
+        description="This is test request 1",
+        priority=RequestPriority.HIGH.value,
+        status=RequestStatus.NEW.value,
+        employee_id=employee1.id
+    )
+    request2 = Request(
+        title="Test Request 2",
+        description="This is test request 2",
+        priority=RequestPriority.MEDIUM.value,
+        status=RequestStatus.IN_PROGRESS.value,
+        employee_id=employee2.id
+    )
+    db_session.add_all([request1, request2])
+    db_session.commit()
     
-    assert response.status_code == 422
+    # Сохраняем данные для сравнения
+    expected_titles = {request1.title, request2.title}
+
+    # Login as admin
+    admin_response = client.post("/api/auth/admin", json={
+        "username": "admin",
+        "password": "admin123"
+    })
+    assert admin_response.status_code == 200
+    admin_token = admin_response.json()["access_token"]
+
+    # Get all requests
+    headers = {"Authorization": f"Bearer {admin_token}"}
+    response = client.get("/api/requests/admin", headers=headers)
+
+    assert response.status_code == 200
+    data = response.json()
+    assert len(data) == 2
+    received_titles = {r["title"] for r in data}
+    assert received_titles == expected_titles
+
+def test_get_requests_by_status(client, db_session):
+    """Test filtering requests by status"""
+    # Create test employee and requests
+    hashed_password = get_password_hash("test123")
+    employee = Employee(
+        first_name="Test",
+        last_name="User",
+        department="IT",
+        office="A101",
+        password=hashed_password
+    )
+    db_session.add(employee)
+    db_session.commit()
+
+    request1 = Request(
+        title="New Request",
+        description="This is a new request",
+        priority=RequestPriority.HIGH.value,
+        status=RequestStatus.NEW.value,
+        employee_id=employee.id
+    )
+    request2 = Request(
+        title="In Progress Request",
+        description="This is an in progress request",
+        priority=RequestPriority.MEDIUM.value,
+        status=RequestStatus.IN_PROGRESS.value,
+        employee_id=employee.id
+    )
+    db_session.add_all([request1, request2])
+    db_session.commit()
+    
+    # Сохраняем данные для сравнения
+    expected_data = {
+        "title": request1.title,
+        "status": request1.status
+    }
+
+    # Login as admin
+    admin_response = client.post("/api/auth/admin", json={
+        "username": "admin",
+        "password": "admin123"
+    })
+    assert admin_response.status_code == 200
+    admin_token = admin_response.json()["access_token"]
+
+    # Get requests filtered by status
+    headers = {"Authorization": f"Bearer {admin_token}"}
+    response = client.get(f"/api/requests/admin?status={RequestStatus.NEW.value}", headers=headers)
+
+    assert response.status_code == 200
+    data = response.json()
+    assert len(data) == 1
+    assert data[0]["title"] == expected_data["title"]
+    assert data[0]["status"] == expected_data["status"]
+
+def test_get_request_statistics(client, db_session):
+    """Test getting request statistics"""
+    # Create test employee and requests
+    hashed_password = get_password_hash("test123")
+    employee = Employee(
+        first_name="Test",
+        last_name="User",
+        department="IT",
+        office="A101",
+        password=hashed_password
+    )
+    db_session.add(employee)
+    db_session.commit()
+
+    # Create requests with different statuses
+    requests_data = [
+        {"status": RequestStatus.NEW.value, "priority": RequestPriority.HIGH.value},
+        {"status": RequestStatus.IN_PROGRESS.value, "priority": RequestPriority.MEDIUM.value},
+        {"status": RequestStatus.COMPLETED.value, "priority": RequestPriority.LOW.value},
+        {"status": RequestStatus.NEW.value, "priority": RequestPriority.HIGH.value}
+    ]
+
+    for i, data in enumerate(requests_data):
+        request = Request(
+            title=f"Request {i+1}",
+            description=f"This is request {i+1}",
+            priority=data["priority"],
+            status=data["status"],
+            employee_id=employee.id
+        )
+        db_session.add(request)
+    db_session.commit()
+
+    # Login as admin
+    admin_response = client.post("/api/auth/admin", json={
+        "username": "admin",
+        "password": "admin123"
+    })
+    assert admin_response.status_code == 200
+    admin_token = admin_response.json()["access_token"]
+
+    # Get statistics
+    headers = {"Authorization": f"Bearer {admin_token}"}
+    response = client.get("/api/requests/statistics", headers=headers)
+
+    assert response.status_code == 200
+    data = response.json()
+    
+    # Проверяем статистику
+    assert "total_requests" in data
+    assert data["total_requests"] == 4
+    assert "by_status" in data
+    assert data["by_status"]["new"] == 2
+    assert data["by_status"]["in_progress"] == 1
+    assert data["by_status"]["completed"] == 1
+    assert "by_priority" in data
+    assert data["by_priority"]["high"] == 2
+    assert data["by_priority"]["medium"] == 1
+    assert data["by_priority"]["low"] == 1
+
+def test_create_request_unauthorized(client):
+    """Test creating request without authorization"""
+    request_data = {
+        "title": "Test Request",
+        "description": "This is a test request",
+        "priority": RequestPriority.MEDIUM.value
+    }
+    response = client.post("/api/requests/", json=request_data)
+    assert response.status_code == 401
+
+def test_get_requests_unauthorized(client):
+    """Test getting requests without authorization"""
+    response = client.get("/api/requests/my")
+    assert response.status_code == 401