1 files changed, 44 insertions, 0 deletions
diff --git a/internal/web/search_test.go b/internal/web/search_test.go
new file mode 100644
index 0000000..6a9b0a5
--- /dev/null
+++ b/internal/web/search_test.go
@@ -0,0 +1,44 @@
+package web
+
+import (
+	"regexp"
+	"strings"
+	"testing"
+)
+
+func TestSearchPattern(t *testing.T) {
+	for q, want := range map[string]string{
+		"go tem":       "go.*tem",
+		"  a.b   (c) ": `a\.b.*\(c\)`,
+		"one":          "one",
+		"   ":          "",
+		"":             "",
+	} {
+		if got := searchPattern(q); got != want {
+			t.Errorf("searchPattern(%q) = %q, want %q", q, got, want)
+		}
+	}
+	// The pattern is what Postgres gets, and also what Go runs for the snippet.
+	re := regexp.MustCompile("(?is)" + searchPattern("go tem"))
+	if !re.MatchString("Go\ntemplates") || re.MatchString("templates for go") {
+		t.Error("words match in order, across lines, any case")
+	}
+}
+
+func TestSearchSnippet(t *testing.T) {
+	re := regexp.MustCompile("(?is)" + searchPattern("tem"))
+	got := string(searchSnippet("Go <b>Templates</b>\n\nare fine.", re))
+	if got != "Go &lt;b&gt;<mark>Tem</mark>plates&lt;/b&gt; are fine." {
+		t.Errorf("snippet: %q", got)
+	}
+	// Long bodies are trimmed around the match, rune by rune.
+	body := strings.Repeat("α", 200) + " tem " + strings.Repeat("ω", 200)
+	got = string(searchSnippet(body, re))
+	if !strings.HasPrefix(got, "…"+strings.Repeat("α", 79)+" <mark>tem</mark> ") || !strings.HasSuffix(got, strings.Repeat("ω", 79)+"…") {
+		t.Errorf("trimmed snippet: %q", got)
+	}
+	// Only the title matched: the body's start, no mark.
+	if got := string(searchSnippet("Nothing here.", re)); got != "Nothing here." {
+		t.Errorf("title-only snippet: %q", got)
+	}
+}