feat: escalate brain search to web when facts cannot confirm

internal/websearch/lookup_test.go

@@ -0,0 +1,97 @@
+package websearch
+
+import (
+	"context"
+	"encoding/json"
+	"net/http"
+	"net/http/httptest"
+	"os"
+	"path/filepath"
+	"testing"
+	"time"
+)
+
+func TestLookupRefusesPIIWithoutFetch(t *testing.T) {
+	hits := 0
+	srv := httptest.NewServer(http.HandlerFunc(func(http.ResponseWriter, *http.Request) {
+		hits++
+	}))
+	defer srv.Close()
+	out := Lookup(context.Background(), "Personalnummer 12", LookupOpt{
+		EnvPath: writeEnv(t, srv.URL),
+		CachePath: filepath.Join(t.TempDir(), "c.sqlite"),
+		Client:    srv.Client(),
+		Sleep:     func(context.Context, time.Duration) error { return nil },
+	})
+	if out.Status != StatusRefused {
+		t.Fatalf("status = %s", out.Status)
+	}
+	if hits != 0 {
+		t.Fatal("PII query left the host")
+	}
+}
+
+func TestLookupSkipsWhenNoConfig(t *testing.T) {
+	out := Lookup(context.Background(), "LadybugDB", LookupOpt{
+		EnvPath:   filepath.Join(t.TempDir(), "missing.env"),
+		CachePath: filepath.Join(t.TempDir(), "c.sqlite"),
+		Sleep:     func(context.Context, time.Duration) error { return nil },
+	})
+	if out.Status != StatusSkipped {
+		t.Fatalf("status = %s", out.Status)
+	}
+}
+
+func TestLookupFetchesOnceAndCaches(t *testing.T) {
+	hits := 0
+	payload := Payload{Query: "x", Results: []RawHit{{Title: "t", URL: "http://example.com", Content: "c", Engine: "bing"}}}
+	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		hits++
+		json.NewEncoder(w).Encode(payload)
+	}))
+	defer srv.Close()
+	opt := LookupOpt{
+		EnvPath:   writeEnv(t, srv.URL),
+		CachePath: filepath.Join(t.TempDir(), "c.sqlite"),
+		Client:    srv.Client(),
+		Now:       func() float64 { return 1_000 },
+		Sleep:     func(context.Context, time.Duration) error { return nil },
+	}
+	a := Lookup(context.Background(), "LadybugDB", opt)
+	b := Lookup(context.Background(), "LadybugDB", opt)
+	if a.Status != StatusOK || b.Status != StatusOK {
+		t.Fatalf("a=%s b=%s", a.Status, b.Status)
+	}
+	if hits != 1 {
+		t.Fatalf("hits = %d, want 1 (second from cache)", hits)
+	}
+	if !b.Cached {
+		t.Fatal("second lookup not cached")
+	}
+}
+
+func TestLookupEmptyIsThrottled(t *testing.T) {
+	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		w.Write([]byte(`{"query":"x","results":[]}`))
+	}))
+	defer srv.Close()
+	out := Lookup(context.Background(), "LadybugDB", LookupOpt{
+		EnvPath:   writeEnv(t, srv.URL),
+		CachePath: filepath.Join(t.TempDir(), "c.sqlite"),
+		Client:    srv.Client(),
+		Now:       func() float64 { return 1_000 },
+		Sleep:     func(context.Context, time.Duration) error { return nil },
+	})
+	if out.Status != StatusThrottled {
+		t.Fatalf("status = %s", out.Status)
+	}
+}
+
+func writeEnv(t *testing.T, url string) string {
+	t.Helper()
+	p := filepath.Join(t.TempDir(), "search.env")
+	if err := os.WriteFile(p, []byte("BRAIN_SEARCH_URL="+url+"\n"), 0o600); err != nil {
+		t.Fatal(err)
+	}
+	return p
+}