e610e253ef
The write handler was building a hand-rolled map that dropped the type and domain fields from writeArgs. Pass the struct directly so all fields reach the ingestion server. Strengthen the test to assert the request body contains the type field. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
66 lines
2.1 KiB
Go
66 lines
2.1 KiB
Go
// internal/skills/brain/handlers_test.go
|
|
package brain_test
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/mathiasbq/supervisor/internal/skills/brain"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestHandle_BrainQuery_CallsIngestServer(t *testing.T) {
|
|
called := false
|
|
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
assert.Equal(t, "/query", r.URL.Path)
|
|
called = true
|
|
json.NewEncoder(w).Encode(map[string]any{
|
|
"results": []map[string]any{
|
|
{"path": "wiki/concepts/tdd.md", "title": "TDD", "excerpt": "Test-driven development.", "score": 3},
|
|
},
|
|
})
|
|
}))
|
|
defer srv.Close()
|
|
|
|
s := brain.New(brain.Config{IngestBaseURL: srv.URL})
|
|
args, _ := json.Marshal(map[string]string{"query": "test driven development"})
|
|
out, err := s.Handle(context.Background(), "brain_query", args)
|
|
require.NoError(t, err)
|
|
assert.True(t, called)
|
|
|
|
var result map[string]any
|
|
require.NoError(t, json.Unmarshal(out, &result))
|
|
results := result["results"].([]any)
|
|
assert.Len(t, results, 1)
|
|
}
|
|
|
|
func TestHandle_BrainWrite_CallsIngestServer(t *testing.T) {
|
|
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
assert.Equal(t, "/write", r.URL.Path)
|
|
var body map[string]string
|
|
require.NoError(t, json.NewDecoder(r.Body).Decode(&body))
|
|
assert.Equal(t, "concept", body["type"])
|
|
assert.Equal(t, "# Test\n\nSome learning.", body["content"])
|
|
json.NewEncoder(w).Encode(map[string]string{"path": "raw/test.md"})
|
|
}))
|
|
defer srv.Close()
|
|
|
|
s := brain.New(brain.Config{IngestBaseURL: srv.URL})
|
|
args, _ := json.Marshal(map[string]string{"content": "# Test\n\nSome learning.", "type": "concept"})
|
|
out, err := s.Handle(context.Background(), "brain_write", args)
|
|
require.NoError(t, err)
|
|
var result map[string]string
|
|
require.NoError(t, json.Unmarshal(out, &result))
|
|
assert.Equal(t, "raw/test.md", result["path"])
|
|
}
|
|
|
|
func TestHandle_UnknownTool_ReturnsError(t *testing.T) {
|
|
s := brain.New(brain.Config{IngestBaseURL: "http://localhost:3300"})
|
|
_, err := s.Handle(context.Background(), "brain_unknown", nil)
|
|
assert.Error(t, err)
|
|
}
|