mirror of
https://github.com/pomerium/pomerium.git
synced 2025-04-29 18:36:30 +02:00
96424dac0f
* add google cloud serverless support * force ipv4 for google cloud serverless * disable long line linting * fix destination hostname * add test * add support for service accounts * fix utc time in test
40 lines
997 B
Go
40 lines
997 B
Go
package authorize
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestGCPIdentityTokenSource(t *testing.T) {
|
|
originalGCPIdentityDocURL := gcpIdentityDocURL
|
|
defer func() {
|
|
gcpIdentityDocURL = originalGCPIdentityDocURL
|
|
gcpIdentityNow = time.Now
|
|
}()
|
|
|
|
now := time.Date(2020, 1, 1, 1, 0, 0, 0, time.UTC)
|
|
gcpIdentityNow = func() time.Time {
|
|
return now
|
|
}
|
|
|
|
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
assert.Equal(t, "Google", r.Header.Get("Metadata-Flavor"))
|
|
assert.Equal(t, "full", r.URL.Query().Get("format"))
|
|
assert.Equal(t, "example", r.URL.Query().Get("audience"))
|
|
_, _ = w.Write([]byte(now.Format(time.RFC3339)))
|
|
}))
|
|
defer srv.Close()
|
|
|
|
gcpIdentityDocURL = srv.URL
|
|
|
|
src, err := getGoogleCloudServerlessTokenSource("", "example")
|
|
assert.NoError(t, err)
|
|
|
|
token, err := src.Token()
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, "2020-01-01T01:00:00Z", token.AccessToken)
|
|
}
|