33 lines
984 B
Go
33 lines
984 B
Go
|
package service
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestWithHTTPUpstreamProfile_DefaultKeepsContext(t *testing.T) {
|
||
|
|
ctx := context.Background()
|
||
|
|
got := WithHTTPUpstreamProfile(ctx, HTTPUpstreamProfileDefault)
|
||
|
|
if got != ctx {
|
||
|
|
t.Fatal("default profile should not wrap context")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestWithHTTPUpstreamProfile_OpenAI(t *testing.T) {
|
||
|
|
ctx := WithHTTPUpstreamProfile(context.TODO(), HTTPUpstreamProfileOpenAI)
|
||
|
|
if profile := HTTPUpstreamProfileFromContext(ctx); profile != HTTPUpstreamProfileOpenAI {
|
||
|
|
t.Fatalf("expected profile %q, got %q", HTTPUpstreamProfileOpenAI, profile)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestWithHTTPUpstreamRedirectsDisabled(t *testing.T) {
|
||
|
|
//nolint:staticcheck // Exercises the defensive nil-context fallback.
|
||
|
|
ctx := WithHTTPUpstreamRedirectsDisabled(nil)
|
||
|
|
if !HTTPUpstreamRedirectsDisabled(ctx) {
|
||
|
|
t.Fatal("expected redirects to be disabled")
|
||
|
|
}
|
||
|
|
if HTTPUpstreamRedirectsDisabled(context.Background()) {
|
||
|
|
t.Fatal("redirects should remain enabled by default")
|
||
|
|
}
|
||
|
|
}
|