22 lines
595 B
Go
22 lines
595 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)
|
|
}
|
|
}
|