statuses update

workers/transcribe/internal/prompts/prompts_test.go

@@ -0,0 +1,116 @@
+package prompts
+
+import (
+	"context"
+	"encoding/json"
+	"net/http"
+	"net/http/httptest"
+	"os"
+	"path/filepath"
+	"testing"
+)
+
+func samplePrompts() []byte {
+	data, err := json.Marshal([]map[string]any{
+		{"id": 1, "id_section": 1, "name": "behavioral", "prompt": "p1", "dt_create": "2026-01-01"},
+		{"id": 2, "id_section": 1, "name": "client_data", "prompt": "p2", "dt_create": "2026-01-01"},
+		{"id": 3, "id_section": 2, "name": "other", "prompt": "p3", "dt_create": "2026-01-01"},
+	})
+	if err != nil {
+		panic(err)
+	}
+	return data
+}
+
+func TestLoadStatic(t *testing.T) {
+	dir := t.TempDir()
+	path := filepath.Join(dir, "prompts.json")
+	if err := os.WriteFile(path, samplePrompts(), 0o644); err != nil {
+		t.Fatal(err)
+	}
+
+	l := New("static", path, "", "", 1)
+	got, err := l.Load(context.Background())
+	if err != nil {
+		t.Fatal(err)
+	}
+	if len(got) != 2 {
+		t.Fatalf("want 2 prompts for section 1, got %d", len(got))
+	}
+	if got[0].Name != "behavioral" || got[1].Name != "client_data" {
+		t.Fatalf("unexpected names: %v, %v", got[0].Name, got[1].Name)
+	}
+}
+
+func TestLoadStaticAllSections(t *testing.T) {
+	dir := t.TempDir()
+	path := filepath.Join(dir, "prompts.json")
+	if err := os.WriteFile(path, samplePrompts(), 0o644); err != nil {
+		t.Fatal(err)
+	}
+
+	l := New("static", path, "", "", 0)
+	got, err := l.Load(context.Background())
+	if err != nil {
+		t.Fatal(err)
+	}
+	if len(got) != 3 {
+		t.Fatalf("want 3 prompts without filter, got %d", len(got))
+	}
+}
+
+func TestLoadStaticMissingFile(t *testing.T) {
+	l := New("static", filepath.Join(t.TempDir(), "missing.json"), "", "", 1)
+	_, err := l.Load(context.Background())
+	if err == nil {
+		t.Fatal("expected error for missing file")
+	}
+}
+
+func TestLoadHTTP(t *testing.T) {
+	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		if r.URL.Path != "/metrics/" {
+			t.Fatalf("path: got %s", r.URL.Path)
+		}
+		if r.URL.Query().Get("id_section") != "1" {
+			t.Fatalf("id_section: got %s", r.URL.Query().Get("id_section"))
+		}
+		if auth := r.Header.Get("Authorization"); auth != "Bearer test-key" {
+			t.Fatalf("auth: got %q", auth)
+		}
+		w.Header().Set("Content-Type", "application/json")
+		_, _ = w.Write(samplePrompts())
+	}))
+	defer srv.Close()
+
+	l := New("http", "", srv.URL, "test-key", 1)
+	got, err := l.Load(context.Background())
+	if err != nil {
+		t.Fatal(err)
+	}
+	if len(got) != 2 {
+		t.Fatalf("want 2 prompts, got %d", len(got))
+	}
+}
+
+func TestLoadHTTPErrorStatus(t *testing.T) {
+	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, _ *http.Request) {
+		w.WriteHeader(http.StatusInternalServerError)
+		_, _ = w.Write([]byte("boom"))
+	}))
+	defer srv.Close()
+
+	l := New("http", "", srv.URL, "", 1)
+	_, err := l.Load(context.Background())
+	if err == nil {
+		t.Fatal("expected error on 500")
+	}
+}
+
+func TestLoadHTTPMissingBaseURL(t *testing.T) {
+	l := New("http", "", "", "", 1)
+	_, err := l.Load(context.Background())
+	if err == nil {
+		t.Fatal("expected error without base URL")
+	}
+}