Function TestSessionIDHeader_GatewayBoundOnly

pkg/httpclient/client_test.go:124–170

(t *testing.T)

}

func TestSessionIDHeader_GatewayBoundOnly(t *testing.T) {
	t.Parallel()

	tests := []struct {
		name           string
		ctxSessionID   string
		opts           []Opt
		wantHeaderSent bool
	}{
		{
			name:           "session ID present, gateway-bound (X-Cagent-Forward set) → header sent",
			ctxSessionID:   "sess-abc",
			opts:           []Opt{WithProxiedBaseURL("https://gateway.example/v1")},
			wantHeaderSent: true,
		},
		{
			name:           "session ID present, no X-Cagent-Forward → header skipped",
			ctxSessionID:   "sess-abc",
			opts:           nil,
			wantHeaderSent: false,
		},
		{
			name:           "no session ID on context, gateway-bound → header skipped",
			ctxSessionID:   "",
			opts:           []Opt{WithProxiedBaseURL("https://gateway.example/v1")},
			wantHeaderSent: false,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			t.Parallel()

			ctx := t.Context()
			if tt.ctxSessionID != "" {
				ctx = ContextWithSessionID(ctx, tt.ctxSessionID)
			}
			headers := doRequestWithCtx(t, ctx, tt.opts...)

			if tt.wantHeaderSent {
				assert.Equal(t, tt.ctxSessionID, headers.Get("X-Cagent-Session-Id"))
			} else {
				assert.Empty(t, headers.Get("X-Cagent-Session-Id"))
			}
		})
	}
}

func TestContextWithSessionID_RoundTrip(t *testing.T) {
	t.Parallel()