restructure project

tests/web/test_admin_routes.py

@@ -12,8 +12,9 @@ from uuid import UUID
 from fastapi import HTTPException
 from fastapi.testclient import TestClient
 
-from src.data.admin_models import AdminDocument, AdminToken
-from src.web.api.v1.admin.documents import _validate_uuid, create_admin_router
+from inference.data.admin_models import AdminDocument, AdminToken
+from inference.web.api.v1.admin.documents import _validate_uuid, create_documents_router
+from inference.web.config import StorageConfig
 
 
 # Test UUID
@@ -42,13 +43,12 @@ class TestAdminRouter:
 
     def test_creates_router_with_endpoints(self):
         """Test router is created with expected endpoints."""
-        router = create_admin_router((".pdf", ".png", ".jpg"))
+        router = create_documents_router(StorageConfig())
 
         # Get route paths (include prefix from router)
         paths = [route.path for route in router.routes]
 
-        # Paths include the /admin prefix
-        assert any("/auth/token" in p for p in paths)
+        # Paths include the /admin/documents prefix
         assert any("/documents" in p for p in paths)
         assert any("/documents/stats" in p for p in paths)
         assert any("{document_id}" in p for p in paths)
@@ -66,7 +66,7 @@ class TestCreateTokenEndpoint:
 
     def test_create_token_success(self, mock_db):
         """Test successful token creation."""
-        from src.web.schemas.admin import AdminTokenCreate
+        from inference.web.schemas.admin import AdminTokenCreate
 
         request = AdminTokenCreate(name="Test Token", expires_in_days=30)