import os import uuid from datetime import date, timedelta import pytest from fastapi.testclient import TestClient os.environ.setdefault("SECRET_KEY", "x" * 32) os.environ.setdefault("DATABASE_URL", "sqlite:////tmp/delphi_test.sqlite") from app.main import app # noqa: E402 from app.auth.security import get_current_user # noqa: E402 class _User: def __init__(self): self.id = "test" self.username = "tester" self.is_admin = True self.is_active = True @pytest.fixture() def client(): app.dependency_overrides[get_current_user] = lambda: _User() try: yield TestClient(app) finally: app.dependency_overrides.pop(get_current_user, None) def _create_customer(client: TestClient) -> str: cid = f"FSSR-{uuid.uuid4().hex[:8]}" resp = client.post("/api/customers/", json={"id": cid, "last": "SearchSort", "email": f"{cid}@example.com"}) assert resp.status_code == 200 return cid def _create_file(client: TestClient, file_no: str, owner_id: str, regarding: str, opened: date): payload = { "file_no": file_no, "id": owner_id, "regarding": regarding, "empl_num": "E01", "file_type": "CIVIL", "opened": opened.isoformat(), "status": "ACTIVE", "rate_per_hour": 100.0, "memo": "test search/sort", } resp = client.post("/api/files/", json=payload) assert resp.status_code == 200 def test_files_tokenized_search_sort_and_pagination(client: TestClient): owner_id = _create_customer(client) base_day = date.today() f1 = f"FS-{uuid.uuid4().hex[:6]}" f2 = f"FS-{uuid.uuid4().hex[:6]}" # f1 contains both tokens across a single field _create_file(client, f1, owner_id, regarding="Alpha project Beta milestone", opened=base_day - timedelta(days=1)) # f2 contains only one token _create_file(client, f2, owner_id, regarding="Only Alpha token here", opened=base_day) # Tokenized search: both tokens required (AND-of-OR across fields) resp = client.get("/api/files/", params={"search": "Alpha Beta"}) assert resp.status_code == 200 items = resp.json() file_nos = {it["file_no"] for it in items} assert f1 in file_nos and f2 not in file_nos # Sorting by opened desc should put f2 first if both were present; we restrict to both-token result (just f1) resp = client.get("/api/files/", params={"search": "Alpha Beta", "sort_by": "opened", "sort_dir": "desc"}) assert resp.status_code == 200 items = resp.json() assert len(items) >= 1 and items[0]["file_no"] == f1 # Pagination over a broader query (single-token) to verify skip/limit resp = client.get( "/api/files/", params={"search": "Alpha", "sort_by": "file_no", "sort_dir": "asc", "limit": 1, "skip": 0}, ) assert resp.status_code == 200 first_page = resp.json() assert len(first_page) == 1 resp = client.get( "/api/files/", params={"search": "Alpha", "sort_by": "file_no", "sort_dir": "asc", "limit": 1, "skip": 1}, ) second_page = resp.json() assert len(second_page) >= 0 # may be 0 or 1 depending on other fixtures