68 lines
2.3 KiB
Python
68 lines
2.3 KiB
Python
from pathlib import Path
|
|
import tempfile
|
|
import unittest
|
|
|
|
from fastapi import FastAPI
|
|
from fastapi.testclient import TestClient
|
|
|
|
from web.static_files import NoCacheStaticFiles
|
|
|
|
|
|
ROOT = Path(__file__).resolve().parents[1]
|
|
STATIC_DIR = ROOT / "web" / "static"
|
|
DEV_HTML = STATIC_DIR / "dev.html"
|
|
JS_DIR = STATIC_DIR / "js"
|
|
VENDOR_DIR = STATIC_DIR / "vendor" / "markdown"
|
|
|
|
|
|
def _frontend_source() -> str:
|
|
"""dev.html + 拆出的 ES module 源(路径 1 模块化后逻辑散落 js/*.js)合并文本。"""
|
|
parts = [DEV_HTML.read_text(encoding="utf-8")]
|
|
parts += [p.read_text(encoding="utf-8") for p in sorted(JS_DIR.glob("*.js"))]
|
|
return "\n".join(parts)
|
|
|
|
|
|
class StaticVendorTests(unittest.TestCase):
|
|
def test_dev_html_uses_local_markdown_vendor_assets(self) -> None:
|
|
html = DEV_HTML.read_text(encoding="utf-8")
|
|
|
|
self.assertNotIn("cdn.jsdelivr.net", html)
|
|
self.assertIn('src="vendor/markdown/marked.umd.js"', html)
|
|
self.assertIn('src="vendor/markdown/purify.min.js"', html)
|
|
self.assertIn('src="vendor/markdown/highlight.min.js"', html)
|
|
self.assertIn('href="vendor/markdown/github.min.css"', html)
|
|
|
|
for name in (
|
|
"marked.umd.js",
|
|
"purify.min.js",
|
|
"highlight.min.js",
|
|
"github.min.css",
|
|
):
|
|
path = VENDOR_DIR / name
|
|
self.assertTrue(path.exists(), f"missing vendored asset: {path}")
|
|
self.assertGreater(path.stat().st_size, 0, f"empty vendored asset: {path}")
|
|
|
|
def test_frontend_surfaces_context_and_cache_stats(self) -> None:
|
|
src = _frontend_source()
|
|
|
|
self.assertIn("formatContextStats", src)
|
|
self.assertIn("context_original_chars", src)
|
|
self.assertIn("cache_hit_tokens", src)
|
|
|
|
def test_static_js_is_served_with_revalidation_header(self) -> None:
|
|
with tempfile.TemporaryDirectory() as tmp:
|
|
static_dir = Path(tmp)
|
|
(static_dir / "app.js").write_text("export const ok = true;\n", encoding="utf-8")
|
|
app = FastAPI()
|
|
app.mount("/static", NoCacheStaticFiles(directory=str(static_dir)), name="static")
|
|
client = TestClient(app)
|
|
|
|
resp = client.get("/static/app.js")
|
|
|
|
self.assertEqual(resp.status_code, 200)
|
|
self.assertEqual(resp.headers.get("cache-control"), "no-cache")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|