diff --git a/surfsense_backend/tests/integration/document_upload/test_document_upload.py b/surfsense_backend/tests/integration/document_upload/test_document_upload.py index 87f953a51..13ceae828 100644 --- a/surfsense_backend/tests/integration/document_upload/test_document_upload.py +++ b/surfsense_backend/tests/integration/document_upload/test_document_upload.py @@ -335,53 +335,3 @@ class TestDocumentSearchability: assert doc_ids[0] in result_ids, ( f"Uploaded document {doc_ids[0]} not found in search results: {result_ids}" ) - - -# --------------------------------------------------------------------------- -# Test L: Status polling returns correct structure -# --------------------------------------------------------------------------- - - -class TestStatusPolling: - """Verify the status endpoint returns well-formed responses.""" - - async def test_status_endpoint_returns_items( - self, - client: httpx.AsyncClient, - headers: dict[str, str], - search_space_id: int, - cleanup_doc_ids: list[int], - ): - resp = await upload_file( - client, headers, "sample.txt", search_space_id=search_space_id - ) - doc_ids = resp.json()["document_ids"] - cleanup_doc_ids.extend(doc_ids) - - status_resp = await client.get( - "/api/v1/documents/status", - headers=headers, - params={ - "search_space_id": search_space_id, - "document_ids": ",".join(str(d) for d in doc_ids), - }, - ) - assert status_resp.status_code == 200 - - body = status_resp.json() - assert "items" in body - assert len(body["items"]) == len(doc_ids) - for item in body["items"]: - assert "id" in item - assert "status" in item - assert "state" in item["status"] - assert item["status"]["state"] in { - "pending", - "processing", - "ready", - "failed", - } - - await poll_document_status( - client, headers, doc_ids, search_space_id=search_space_id - )