#!/usr/bin/env python3
"""FastAPI smoke tests for the TTT v1.4 Space app."""
from fastapi.testclient import TestClient
import app as ttt_app
from ttt_codec_v14 import HEADER, MAGIC, NAME_TO_CODEC


def main() -> int:
    client = TestClient(ttt_app.app)
    r = client.get("/health")
    assert r.status_code == 200 and r.json()["packet_format"] == "TTT4"
    print("PASS GET /health")

    original = "TTT v1.4 API — exact bytes 🌏".encode("utf-8")
    encoded = client.post("/encode?mode=auto", content=original)
    assert encoded.status_code == 200, encoded.text
    assert encoded.headers.get("x-ttt-version") == "1.4.0"
    print(f"PASS POST /encode ({len(original)} -> {len(encoded.content)} bytes)")

    decoded = client.post("/decode", content=encoded.content)
    assert decoded.status_code == 200 and decoded.content == original
    print("PASS POST /decode exact byte match")

    bad = client.post("/decode", content=b"not a packet")
    assert bad.status_code == 400
    malformed_gzip = HEADER.pack(MAGIC, NAME_TO_CODEC["gzip"], 12, 0) + b"not gzip"
    bad_payload = client.post("/decode", content=malformed_gzip)
    assert bad_payload.status_code == 400, bad_payload.text
    print("PASS POST /decode rejects invalid/corrupt packets")

    root = client.get("/", follow_redirects=False)
    assert root.status_code in (307, 308) and root.headers.get("location", "").startswith("/ui"), root.status_code
    ui = client.get("/ui/")
    assert ui.status_code == 200, ui.status_code
    print("PASS root redirect and mounted Gradio UI")
    print("ALL TTT v1.4 API TESTS PASSED")
    return 0

if __name__ == "__main__":
    raise SystemExit(main())