mirror of
https://github.com/jkingsman/Remote-Terminal-for-MeshCore.git
synced 2026-03-28 17:43:05 +01:00
162 lines
5.9 KiB
Python
162 lines
5.9 KiB
Python
import logging
|
|
|
|
from fastapi import FastAPI
|
|
from fastapi.testclient import TestClient
|
|
|
|
from app.frontend_static import (
|
|
ASSET_CACHE_CONTROL,
|
|
FRONTEND_BUILD_INSTRUCTIONS,
|
|
INDEX_CACHE_CONTROL,
|
|
STATIC_FILE_CACHE_CONTROL,
|
|
register_first_available_frontend_static_routes,
|
|
register_frontend_missing_fallback,
|
|
register_frontend_static_routes,
|
|
)
|
|
|
|
|
|
def test_missing_dist_logs_error_and_keeps_app_running(tmp_path, caplog):
|
|
app = FastAPI()
|
|
missing_dist = tmp_path / "frontend" / "dist"
|
|
|
|
with caplog.at_level(logging.ERROR):
|
|
registered = register_frontend_static_routes(app, missing_dist)
|
|
|
|
assert registered is False
|
|
assert "Frontend build directory not found" in caplog.text
|
|
|
|
# Register the fallback like main.py does
|
|
register_frontend_missing_fallback(app)
|
|
|
|
with TestClient(app) as client:
|
|
resp = client.get("/")
|
|
assert resp.status_code == 404
|
|
assert FRONTEND_BUILD_INSTRUCTIONS in resp.json()["detail"]
|
|
|
|
|
|
def test_missing_index_logs_error_and_skips_frontend_routes(tmp_path, caplog):
|
|
app = FastAPI()
|
|
dist_dir = tmp_path / "frontend" / "dist"
|
|
dist_dir.mkdir(parents=True)
|
|
|
|
with caplog.at_level(logging.ERROR):
|
|
registered = register_frontend_static_routes(app, dist_dir)
|
|
|
|
assert registered is False
|
|
assert "Frontend index file not found" in caplog.text
|
|
|
|
|
|
def test_valid_dist_serves_static_and_spa_fallback(tmp_path):
|
|
app = FastAPI()
|
|
dist_dir = tmp_path / "frontend" / "dist"
|
|
assets_dir = dist_dir / "assets"
|
|
dist_dir.mkdir(parents=True)
|
|
assets_dir.mkdir(parents=True)
|
|
|
|
index_file = dist_dir / "index.html"
|
|
index_file.write_text("<html><body>index page</body></html>")
|
|
(dist_dir / "robots.txt").write_text("User-agent: *")
|
|
(assets_dir / "app.js").write_text("console.log('ok');")
|
|
|
|
registered = register_frontend_static_routes(app, dist_dir)
|
|
assert registered is True
|
|
|
|
with TestClient(app) as client:
|
|
root_response = client.get("/")
|
|
assert root_response.status_code == 200
|
|
assert "index page" in root_response.text
|
|
assert root_response.headers["cache-control"] == INDEX_CACHE_CONTROL
|
|
|
|
manifest_response = client.get("/site.webmanifest")
|
|
assert manifest_response.status_code == 200
|
|
assert manifest_response.headers["content-type"].startswith("application/manifest+json")
|
|
assert manifest_response.headers["cache-control"] == "no-store"
|
|
manifest = manifest_response.json()
|
|
assert manifest["start_url"] == "http://testserver/"
|
|
assert manifest["scope"] == "http://testserver/"
|
|
assert manifest["id"] == "http://testserver/"
|
|
assert manifest["display"] == "standalone"
|
|
assert manifest["icons"][0]["src"] == "http://testserver/web-app-manifest-192x192.png"
|
|
|
|
file_response = client.get("/robots.txt")
|
|
assert file_response.status_code == 200
|
|
assert file_response.text == "User-agent: *"
|
|
assert file_response.headers["cache-control"] == STATIC_FILE_CACHE_CONTROL
|
|
|
|
explicit_index_response = client.get("/index.html")
|
|
assert explicit_index_response.status_code == 200
|
|
assert "index page" in explicit_index_response.text
|
|
assert explicit_index_response.headers["cache-control"] == INDEX_CACHE_CONTROL
|
|
|
|
missing_response = client.get("/channel/some-route")
|
|
assert missing_response.status_code == 200
|
|
assert "index page" in missing_response.text
|
|
assert missing_response.headers["cache-control"] == INDEX_CACHE_CONTROL
|
|
|
|
asset_response = client.get("/assets/app.js")
|
|
assert asset_response.status_code == 200
|
|
assert "console.log('ok');" in asset_response.text
|
|
assert asset_response.headers["cache-control"] == ASSET_CACHE_CONTROL
|
|
|
|
|
|
def test_webmanifest_uses_forwarded_origin_headers(tmp_path):
|
|
app = FastAPI()
|
|
dist_dir = tmp_path / "frontend" / "dist"
|
|
dist_dir.mkdir(parents=True)
|
|
(dist_dir / "index.html").write_text("<html><body>index page</body></html>")
|
|
|
|
registered = register_frontend_static_routes(app, dist_dir)
|
|
assert registered is True
|
|
|
|
with TestClient(app) as client:
|
|
response = client.get(
|
|
"/site.webmanifest",
|
|
headers={
|
|
"x-forwarded-proto": "https",
|
|
"x-forwarded-host": "mesh.example.com:8443",
|
|
},
|
|
)
|
|
|
|
assert response.status_code == 200
|
|
data = response.json()
|
|
assert data["start_url"] == "https://mesh.example.com:8443/"
|
|
assert data["scope"] == "https://mesh.example.com:8443/"
|
|
assert data["id"] == "https://mesh.example.com:8443/"
|
|
|
|
|
|
def test_first_available_prefers_dist_over_prebuilt(tmp_path):
|
|
app = FastAPI()
|
|
frontend_dir = tmp_path / "frontend"
|
|
dist_dir = frontend_dir / "dist"
|
|
prebuilt_dir = frontend_dir / "prebuilt"
|
|
dist_dir.mkdir(parents=True)
|
|
prebuilt_dir.mkdir(parents=True)
|
|
(dist_dir / "index.html").write_text("<html><body>dist</body></html>")
|
|
(prebuilt_dir / "index.html").write_text("<html><body>prebuilt</body></html>")
|
|
|
|
selected = register_first_available_frontend_static_routes(app, [dist_dir, prebuilt_dir])
|
|
|
|
assert selected == dist_dir.resolve()
|
|
|
|
with TestClient(app) as client:
|
|
response = client.get("/")
|
|
assert response.status_code == 200
|
|
assert "dist" in response.text
|
|
|
|
|
|
def test_first_available_uses_prebuilt_when_dist_missing(tmp_path):
|
|
app = FastAPI()
|
|
frontend_dir = tmp_path / "frontend"
|
|
dist_dir = frontend_dir / "dist"
|
|
prebuilt_dir = frontend_dir / "prebuilt"
|
|
prebuilt_dir.mkdir(parents=True)
|
|
(prebuilt_dir / "index.html").write_text("<html><body>prebuilt</body></html>")
|
|
|
|
selected = register_first_available_frontend_static_routes(app, [dist_dir, prebuilt_dir])
|
|
|
|
assert selected == prebuilt_dir.resolve()
|
|
|
|
with TestClient(app) as client:
|
|
response = client.get("/")
|
|
assert response.status_code == 200
|
|
assert "prebuilt" in response.text
|