git@cosmocr.at/gitweb.git @ main · summary refs log
main / internal / server / mermaid_test.go /
internal/server/mermaid_test.go · 1.8 KiB · raw
view: SourceMarkdown ☑ force

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")
}

}