Function TestApplyHeadersHTTP

cmd/tempo-cli/shared_test.go:13–76

(t *testing.T)

)

func TestApplyHeadersHTTP(t *testing.T) {
	tests := []struct {
		name            string
		headers         []string
		expectedHeaders map[string]string
	}{
		{
			name:    "single header",
			headers: []string{"X-TOKEN=my-secret"},
			expectedHeaders: map[string]string{
				"X-Token": "my-secret",
			},
		},
		{
			name:    "multiple headers",
			headers: []string{"X-TOKEN=my-secret", "X-Custom=value"},
			expectedHeaders: map[string]string{
				"X-Token":  "my-secret",
				"X-Custom": "value",
			},
		},
		{
			name:            "malformed header without equals is ignored",
			headers:         []string{"no-equals-sign"},
			expectedHeaders: map[string]string{},
		},
		{
			name:    "value containing equals",
			headers: []string{"Authorization=Bearer token=abc"},
			expectedHeaders: map[string]string{
				"Authorization": "Bearer token=abc",
			},
		},
		{
			name:            "empty input",
			headers:         []string{},
			expectedHeaders: map[string]string{},
		},
		{
			name:            "empty key is ignored",
			headers:         []string{"=value"},
			expectedHeaders: map[string]string{},
		},
		{
			name:    "whitespace around key is trimmed",
			headers: []string{" X-TOKEN =my-secret"},
			expectedHeaders: map[string]string{
				"X-Token": "my-secret",
			},
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			req, _ := http.NewRequest("GET", "http://localhost:3200/api/search", nil)
			applyHeadersHTTP(req, tt.headers)

			require.Len(t, req.Header, len(tt.expectedHeaders))