package server
import ( "io" "net/http" "net/http/httptest" "os" "path/filepath" "strings" "testing"
"gitweb/internal/store"
)
// TestMermaidPageAndAssets checks that a markdown page with a mermaid block
// renders the diagram container, that the page wires the init script, and that
// the mermaid assets are served same-origin.
func TestMermaidPageAndAssets(t testing.T) {
ns := t.TempDir()
dir := filepath.Join(ns, "u@d.com", "docs")
if err := os.MkdirAll(dir, 0o755); err != nil {
t.Fatal(err)
}
os.WriteFile(filepath.Join(ns, "u@d.com", "Access"), []byte("read,list: all\n: u@d.com\n"), 0o644)
os.WriteFile(filepath.Join(dir, "diagram.md"),
[]byte("# Diagram\n\nmermaid\ngraph TD\n A-->B\n\n"), 0o644)
web := os.DirFS("../../web")
srv, err := New(Config{Store: store.NewLocalStore(ns), Web: http.FS(web), CacheDir: t.TempDir()})
if err != nil {
t.Fatal(err)
}
ts := httptest.NewServer(srv.Handler())
defer ts.Close()
get := func(p string) (int, string) {
resp, err := http.Get(ts.URL + p)
if err != nil {
t.Fatal(err)
}
defer resp.Body.Close()
b, _ := io.ReadAll(resp.Body)
return resp.StatusCode, string(b)
}
code, body := get("/u@d.com/docs/diagram.md")
if code != 200 {
t.Fatalf("page status %d", code)
}
if !strings.Contains(body, `<pre class="mermaid">`) {
t.Errorf("rendered page missing mermaid container:\n%s", body)
}
if !strings.Contains(body, "/_assets/mermaid-init.js") {
t.Errorf("page did not include the mermaid init script")
}
if code, _ := get("/_assets/mermaid-init.js"); code != 200 {
t.Errorf("mermaid-init.js status %d, want 200", code)
}
code, bundle := get("/_assets/mermaid.min.js")
if code != 200 {
t.Fatalf("mermaid.min.js status %d, want 200", code)
}
if !strings.Contains(bundle, "mermaid") {
t.Errorf("served bundle does not look like mermaid")
}
}