| import os |
|
|
| os.environ["OCR_CPU_AUTH_TOKEN"] = "test-token" |
|
|
| import pytest |
| from fastapi.testclient import TestClient |
| from ocr.config import Settings |
| from ocr.worker_api import create_app |
|
|
|
|
| class MockBackend: |
| class Health: |
| ready = True |
|
|
| async def health(self): |
| return self.Health() |
|
|
| async def capabilities(self): |
| return {"models": []} |
|
|
| async def recognize(self, p, opts): |
| return {} |
|
|
|
|
| @pytest.fixture(scope="module") |
| def client(): |
| settings = Settings() |
| settings.worker_token = "test-token" |
| app = create_app(settings) |
| with TestClient(app) as c: |
| app.state.backend = MockBackend() |
| yield c |
|
|
|
|
| def test_health_public(client): |
| response = client.get("/health") |
| assert response.status_code == 200 |
| assert response.json() == {"status": "ok"} |
|
|
|
|
| def test_ready_unauthorized(client): |
| response = client.get("/ready") |
| assert response.status_code == 401 |
|
|
|
|
| def test_v1_models_authorized(client): |
| response = client.get("/v1/models", headers={"Authorization": "Bearer test-token"}) |
| assert response.status_code == 200 |
| assert response.json()["data"][0]["id"] == "ocr-cpu" |
|
|
|
|
| def test_v1_chat_completions_invalid_model(client): |
| req = { |
| "model": "gpt-4", |
| "messages": [ |
| { |
| "role": "user", |
| "content": [ |
| { |
| "type": "image_url", |
| "image_url": {"url": "data:image/png;base64,iVBORw0KGgo="}, |
| } |
| ], |
| } |
| ], |
| } |
| response = client.post( |
| "/v1/chat/completions", headers={"Authorization": "Bearer test-token"}, json=req |
| ) |
| assert response.status_code == 422 |
| assert response.json()["code"] == "invalid_request" |
|
|
|
|
| def test_v1_chat_completions_invalid_url(client): |
| req = { |
| "model": "ocr-cpu", |
| "messages": [ |
| { |
| "role": "user", |
| "content": [ |
| {"type": "image_url", "image_url": {"url": "https://example.com/image.png"}} |
| ], |
| } |
| ], |
| } |
| response = client.post( |
| "/v1/chat/completions", headers={"Authorization": "Bearer test-token"}, json=req |
| ) |
| assert response.status_code == 422 |
| assert response.json()["code"] == "only_image_data_urls_supported" |
|
|
|
|
| def test_v1_ocr_page_wrong_sensitivity(client): |
| files = {"file": ("test.png", b"fake-image-data", "image/png")} |
| data = {"sensitivity": "D1"} |
| response = client.post( |
| "/v1/ocr/page", headers={"Authorization": "Bearer test-token"}, files=files, data=data |
| ) |
| assert response.status_code == 422 |
| assert response.json()["code"] == "invalid_worker_options" |
|
|
