85967e14ca
- Add GET /setup serving install.html (CLI + Claude Code setup page) - Add GET /install → 301 redirect to /setup for backwards compat - Move first-time setup wizard from /setup to /first-time-setup - Update nav link: href=/setup, label 'Setup local agent', active on both /setup and /install paths - Update page <title> to 'Setup local agent — …' - Update /dashboard and /setup comment in _claude_setup_instructions.jinja - Update tests and OpenAPI snapshot accordingly
87 lines
3.6 KiB
Python
87 lines
3.6 KiB
Python
"""Tests for the SelectiveGZipMiddleware path-skip logic in app/main.py.
|
|
|
|
Key property: parquet-serving endpoints must not be gzipped on the wire,
|
|
but JSON / HTML endpoints above the minimum-size threshold must be.
|
|
"""
|
|
|
|
import pytest
|
|
from fastapi.testclient import TestClient
|
|
|
|
|
|
@pytest.fixture
|
|
def isolated_client(tmp_path, monkeypatch):
|
|
"""Fresh FastAPI app with its own tmp DATA_DIR so DuckDB locks don't
|
|
collide with a concurrently-running dev container."""
|
|
monkeypatch.setenv("DATA_DIR", str(tmp_path))
|
|
monkeypatch.setenv("TESTING", "1")
|
|
monkeypatch.setenv("JWT_SECRET_KEY", "test-secret-key-min-32-characters!!")
|
|
(tmp_path / "state").mkdir()
|
|
(tmp_path / "analytics").mkdir()
|
|
(tmp_path / "extracts").mkdir()
|
|
from src.db import close_system_db
|
|
close_system_db()
|
|
from app.main import create_app
|
|
yield TestClient(create_app())
|
|
close_system_db()
|
|
|
|
|
|
def test_parquet_path_is_not_gzipped(isolated_client, tmp_path, monkeypatch):
|
|
"""/cli/wheel/... must return the raw bytes without Content-Encoding: gzip."""
|
|
wheel = tmp_path / "agnes_fake-1.0-py3-none-any.whl"
|
|
wheel.write_bytes(b"PK\x03\x04" + b"x" * 4096)
|
|
monkeypatch.setenv("AGNES_CLI_DIST_DIR", str(tmp_path))
|
|
|
|
resp = isolated_client.get(
|
|
f"/cli/wheel/{wheel.name}",
|
|
headers={"Accept-Encoding": "gzip"},
|
|
)
|
|
assert resp.status_code == 200
|
|
assert "gzip" not in resp.headers.get("content-encoding", "")
|
|
assert resp.content.startswith(b"PK")
|
|
|
|
|
|
def test_install_page_is_gzipped(isolated_client):
|
|
"""/setup is HTML above the threshold — gzip should kick in when the
|
|
client advertises gzip support. TestClient may decompress transparently,
|
|
so we accept either the header or readable body as proof that the
|
|
middleware decided to handle the response (i.e. did not skip)."""
|
|
resp = isolated_client.get("/setup", headers={"Accept-Encoding": "gzip"})
|
|
assert resp.status_code == 200
|
|
enc = resp.headers.get("content-encoding", "")
|
|
# Either we see the encoding on the wire OR TestClient auto-decoded it.
|
|
assert "gzip" in enc or "setup" in resp.text.lower()
|
|
|
|
|
|
def test_no_accept_encoding_means_no_gzip_anywhere(isolated_client):
|
|
"""Client that doesn't advertise gzip gets uncompressed body."""
|
|
resp = isolated_client.get("/setup", headers={"Accept-Encoding": "identity"})
|
|
assert resp.status_code == 200
|
|
assert "gzip" not in resp.headers.get("content-encoding", "")
|
|
|
|
|
|
def test_selective_gzip_wrapper_dispatches_on_prefix():
|
|
"""Direct unit test of the wrapper's path-based branch without standing up
|
|
the whole FastAPI app — verifies the skip list is honoured."""
|
|
from app.main import _SelectiveGZipMiddleware
|
|
|
|
calls = {"raw": 0, "gzip": 0}
|
|
|
|
async def raw_app(scope, receive, send):
|
|
calls["raw"] += 1
|
|
|
|
wrapper = _SelectiveGZipMiddleware(raw_app, minimum_size=10, skip_prefixes=("/api/data/",))
|
|
# Monkey-patch the gzip inner so we can count hits without running middleware.
|
|
async def stub_gzip(scope, receive, send):
|
|
calls["gzip"] += 1
|
|
wrapper._gzip = stub_gzip
|
|
|
|
import asyncio
|
|
# Path that matches the skip prefix → raw app
|
|
asyncio.run(wrapper({"type": "http", "path": "/api/data/orders/download"}, None, None))
|
|
assert calls == {"raw": 1, "gzip": 0}
|
|
# Path that does not → gzip app
|
|
asyncio.run(wrapper({"type": "http", "path": "/api/sync/manifest"}, None, None))
|
|
assert calls == {"raw": 1, "gzip": 1}
|
|
# Non-http scope (websocket, lifespan) → gzip app (it handles lifespan as pass-through)
|
|
asyncio.run(wrapper({"type": "lifespan"}, None, None))
|
|
assert calls == {"raw": 1, "gzip": 2}
|