95 lines
3.1 KiB
Python
95 lines
3.1 KiB
Python
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
|
|
|
|
|