0c541f6 · typeable/oauth2-proxy

git.iko.soy/typeable/oauth2-proxy

Adding additional asserts to the TestAzureProviderREdeemReturnsIdToken to ensure that the refresh token and expires on date are both being set

0c541f6leyshon7 years ago

providers/azure_test.go+6−2

@@ -5,6 +5,7 @@ import (

5 "net/http/httptest"

6 "net/url"

7 "testing"

8+ "time"

9 "github.com/pusher/oauth2_proxy/pkg/apis/sessions"

10 "github.com/pusher/oauth2_proxy/pkg/apis/sessions"

10 "github.com/stretchr/testify/assert"

11 "github.com/stretchr/testify/assert"

@@ -205,7 +206,8 @@ func TestAzureProviderGetEmailAddressIncorrectOtherMails(t *testing.T) {

205}

206}

206

207

207func TestAzureProviderRedeemReturnsIdToken(t *testing.T) {

208func TestAzureProviderRedeemReturnsIdToken(t *testing.T) {

208− b := testAzureBackend(`{ "id_token": "testtoken1234" }`)

209+ b := testAzureBackend(`{ "id_token": "testtoken1234", "expires_on": "1136239445", "refresh_token": "refresh1234" }`)

210+ timestamp, err := time.Parse(time.RFC3339, "2006-01-02T22:04:05Z")

209 defer b.Close()

211 defer b.Close()

210

212

211 bURL, _ := url.Parse(b.URL)

213 bURL, _ := url.Parse(b.URL)

@@ -214,4 +216,6 @@ func TestAzureProviderRedeemReturnsIdToken(t *testing.T) {

214 s, err := p.Redeem("https://localhost", "1234")

216 s, err := p.Redeem("https://localhost", "1234")

215 assert.Equal(t, nil, err)

217 assert.Equal(t, nil, err)

216 assert.Equal(t, "testtoken1234", s.IDToken)

218 assert.Equal(t, "testtoken1234", s.IDToken)

219+ assert.Equal(t, timestamp, s.ExpiresOn.UTC())

220+ assert.Equal(t, "refresh1234", s.RefreshToken)

217−}

221+}