dela
0335d572de
ci / go (push) Waiting to run
ci / go-db (agent) (push) Waiting to run
ci / go-db (config) (push) Waiting to run
ci / go-db (db) (push) Waiting to run
ci / go-db (evidence) (push) Waiting to run
ci / go-db (llmrec) (push) Waiting to run
ci / go-db (server) (push) Waiting to run
detections / detections (push) Waiting to run
web / web (push) Waiting to run
docs / links (push) Canceled after 0s
67 lines
2.1 KiB
Go
67 lines
2.1 KiB
Go
package agent
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
"net/http"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/Autumn-27/norma/transcript"
|
|
)
|
|
|
|
// fakeRT records the request it saw and returns a minimal 200 response.
|
|
type fakeRT struct{ seen *http.Request }
|
|
|
|
func (f *fakeRT) RoundTrip(req *http.Request) (*http.Response, error) {
|
|
f.seen = req
|
|
return &http.Response{
|
|
StatusCode: 200,
|
|
Header: make(http.Header),
|
|
Body: io.NopCloser(strings.NewReader("")),
|
|
}, nil
|
|
}
|
|
|
|
func newReq(ctx context.Context) *http.Request {
|
|
req, _ := http.NewRequestWithContext(ctx, "POST", "https://api.example.com/v1/messages", strings.NewReader("{}"))
|
|
return req
|
|
}
|
|
|
|
func TestSessionHeaderInjectedFromContext(t *testing.T) {
|
|
base := &fakeRT{}
|
|
rt := quotaAwareTransport{base: base, sessionHeaderKey: "x-session-id"}
|
|
ctx := transcript.WithSessionID(context.Background(), "conv-42")
|
|
if _, err := rt.RoundTrip(newReq(ctx)); err != nil {
|
|
t.Fatalf("RoundTrip: %v", err)
|
|
}
|
|
if got := base.seen.Header.Get("x-session-id"); got != "conv-42" {
|
|
t.Fatalf("x-session-id = %q, want conv-42", got)
|
|
}
|
|
}
|
|
|
|
func TestSessionHeaderSkippedWhenKeyEmpty(t *testing.T) {
|
|
base := &fakeRT{}
|
|
rt := quotaAwareTransport{base: base} // no key configured
|
|
ctx := transcript.WithSessionID(context.Background(), "conv-42")
|
|
if _, err := rt.RoundTrip(newReq(ctx)); err != nil {
|
|
t.Fatalf("RoundTrip: %v", err)
|
|
}
|
|
// The header name is whatever the user would have set; with no key, nothing
|
|
// session-related is added. Assert the common key stays absent.
|
|
if got := base.seen.Header.Get("x-session-id"); got != "" {
|
|
t.Fatalf("unexpected session header %q with empty key", got)
|
|
}
|
|
}
|
|
|
|
func TestSessionHeaderSkippedWhenNoSessionID(t *testing.T) {
|
|
base := &fakeRT{}
|
|
rt := quotaAwareTransport{base: base, sessionHeaderKey: "x-session-id"}
|
|
// Context carries no session id (transcript persistence off).
|
|
if _, err := rt.RoundTrip(newReq(context.Background())); err != nil {
|
|
t.Fatalf("RoundTrip: %v", err)
|
|
}
|
|
if got := base.seen.Header.Get("x-session-id"); got != "" {
|
|
t.Fatalf("x-session-id = %q, want empty when no session id on context", got)
|
|
}
|
|
}
|