Function TestBearerAuthTransport

pkg/http/transport/bearer_test.go:16–75

(t *testing.T)

)

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

	tests := []struct {
		name          string
		token         string
		tokenProvider func() string
		wantAuth      string
	}{
		{
			name:     "static token",
			token:    "static-token",
			wantAuth: "Bearer static-token",
		},
		{
			name:          "token provider takes precedence over static token",
			token:         "static-token",
			tokenProvider: func() string { return "provided-token" },
			wantAuth:      "Bearer provided-token",
		},
		{
			name:          "token provider with empty static token",
			tokenProvider: func() string { return "provided-token" },
			wantAuth:      "Bearer provided-token",
		},
		{
			name:          "token provider may return empty before authorization",
			tokenProvider: func() string { return "" },
			wantAuth:      "",
		},
	}

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

			var gotAuth string
			server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
				gotAuth = r.Header.Get(headers.AuthorizationHeader)
				w.WriteHeader(http.StatusOK)
			}))
			defer server.Close()

			rt := &BearerAuthTransport{
				Transport:     http.DefaultTransport,
				Token:         tc.token,
				TokenProvider: tc.tokenProvider,
			}

			req, err := http.NewRequestWithContext(context.Background(), http.MethodGet, server.URL, nil)
			require.NoError(t, err)

			resp, err := rt.RoundTrip(req)
			require.NoError(t, err)
			defer resp.Body.Close()

			assert.Equal(t, tc.wantAuth, gotAuth)
		})