1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67

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\n```mermaid\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")
	}
}