166 lines
5.7 KiB
Python
166 lines
5.7 KiB
Python
"""
|
|
Tests for Document Category API Endpoints.
|
|
|
|
TDD tests for category filtering and management in document endpoints.
|
|
"""
|
|
|
|
import pytest
|
|
from datetime import datetime
|
|
from unittest.mock import MagicMock, patch
|
|
from uuid import UUID, uuid4
|
|
|
|
from fastapi import FastAPI
|
|
from fastapi.testclient import TestClient
|
|
|
|
|
|
# Test constants
|
|
TEST_DOC_UUID = "550e8400-e29b-41d4-a716-446655440000"
|
|
TEST_TOKEN = "test-admin-token-12345"
|
|
|
|
|
|
class TestGetCategoriesEndpoint:
|
|
"""Tests for GET /admin/documents/categories endpoint."""
|
|
|
|
def test_categories_endpoint_returns_list(self):
|
|
"""Test categories endpoint returns list of available categories."""
|
|
from inference.web.schemas.admin import DocumentCategoriesResponse
|
|
|
|
# Test schema exists and works
|
|
response = DocumentCategoriesResponse(
|
|
categories=["invoice", "letter", "receipt"],
|
|
total=3,
|
|
)
|
|
assert response.categories == ["invoice", "letter", "receipt"]
|
|
assert response.total == 3
|
|
|
|
def test_categories_response_schema(self):
|
|
"""Test DocumentCategoriesResponse schema structure."""
|
|
from inference.web.schemas.admin import DocumentCategoriesResponse
|
|
|
|
assert "categories" in DocumentCategoriesResponse.model_fields
|
|
assert "total" in DocumentCategoriesResponse.model_fields
|
|
|
|
|
|
class TestDocumentListFilterByCategory:
|
|
"""Tests for filtering documents by category."""
|
|
|
|
@pytest.fixture
|
|
def mock_admin_db(self):
|
|
"""Create mock AdminDB."""
|
|
db = MagicMock()
|
|
db.is_valid_admin_token.return_value = True
|
|
|
|
# Mock documents with different categories
|
|
invoice_doc = MagicMock()
|
|
invoice_doc.document_id = uuid4()
|
|
invoice_doc.category = "invoice"
|
|
invoice_doc.filename = "invoice1.pdf"
|
|
|
|
letter_doc = MagicMock()
|
|
letter_doc.document_id = uuid4()
|
|
letter_doc.category = "letter"
|
|
letter_doc.filename = "letter1.pdf"
|
|
|
|
db.get_documents.return_value = ([invoice_doc], 1)
|
|
db.get_document_categories.return_value = ["invoice", "letter", "receipt"]
|
|
return db
|
|
|
|
def test_list_documents_accepts_category_filter(self, mock_admin_db):
|
|
"""Test list documents endpoint accepts category query parameter."""
|
|
# The endpoint should accept ?category=invoice parameter
|
|
# This test verifies the schema/query parameter exists
|
|
from inference.web.schemas.admin import DocumentListResponse
|
|
|
|
# Schema should work with category filter applied
|
|
assert DocumentListResponse is not None
|
|
|
|
def test_get_document_categories_from_db(self, mock_admin_db):
|
|
"""Test fetching unique categories from database."""
|
|
categories = mock_admin_db.get_document_categories()
|
|
assert "invoice" in categories
|
|
assert "letter" in categories
|
|
assert len(categories) == 3
|
|
|
|
|
|
class TestDocumentUploadWithCategory:
|
|
"""Tests for uploading documents with category."""
|
|
|
|
def test_upload_request_accepts_category(self):
|
|
"""Test upload request can include category field."""
|
|
# When uploading via form data, category should be accepted
|
|
# This is typically a form field, not a schema
|
|
pass
|
|
|
|
def test_upload_response_includes_category(self):
|
|
"""Test upload response includes the category that was set."""
|
|
from inference.web.schemas.admin import DocumentUploadResponse
|
|
|
|
response = DocumentUploadResponse(
|
|
document_id=TEST_DOC_UUID,
|
|
filename="test.pdf",
|
|
file_size=1024,
|
|
page_count=1,
|
|
status="pending",
|
|
category="letter", # Custom category
|
|
message="Upload successful",
|
|
)
|
|
assert response.category == "letter"
|
|
|
|
def test_upload_defaults_to_invoice_category(self):
|
|
"""Test upload defaults to 'invoice' if no category specified."""
|
|
from inference.web.schemas.admin import DocumentUploadResponse
|
|
|
|
response = DocumentUploadResponse(
|
|
document_id=TEST_DOC_UUID,
|
|
filename="test.pdf",
|
|
file_size=1024,
|
|
page_count=1,
|
|
status="pending",
|
|
message="Upload successful",
|
|
# No category specified - should default to "invoice"
|
|
)
|
|
assert response.category == "invoice"
|
|
|
|
|
|
class TestAdminDBCategoryMethods:
|
|
"""Tests for AdminDB category-related methods."""
|
|
|
|
def test_get_document_categories_method_exists(self):
|
|
"""Test AdminDB has get_document_categories method."""
|
|
from inference.data.admin_db import AdminDB
|
|
|
|
db = AdminDB()
|
|
assert hasattr(db, "get_document_categories")
|
|
|
|
def test_get_documents_accepts_category_filter(self):
|
|
"""Test get_documents_by_token method accepts category parameter."""
|
|
from inference.data.admin_db import AdminDB
|
|
import inspect
|
|
|
|
db = AdminDB()
|
|
# Check the method exists and accepts category parameter
|
|
method = getattr(db, "get_documents_by_token", None)
|
|
assert callable(method)
|
|
|
|
# Check category is in the method signature
|
|
sig = inspect.signature(method)
|
|
assert "category" in sig.parameters
|
|
|
|
|
|
class TestUpdateDocumentCategory:
|
|
"""Tests for updating document category."""
|
|
|
|
def test_update_document_category_method_exists(self):
|
|
"""Test AdminDB has method to update document category."""
|
|
from inference.data.admin_db import AdminDB
|
|
|
|
db = AdminDB()
|
|
assert hasattr(db, "update_document_category")
|
|
|
|
def test_update_request_schema(self):
|
|
"""Test DocumentUpdateRequest can update category."""
|
|
from inference.web.schemas.admin import DocumentUpdateRequest
|
|
|
|
request = DocumentUpdateRequest(category="receipt")
|
|
assert request.category == "receipt"
|