@@ -7,7 +7,6 @@ import (
77 "testing"
88 "time"
99
10− "github.com/pusher/oauth2_proxy/pkg/apis/sessions"
1110 "github.com/stretchr/testify/assert"
1211)
1312
@@ -118,7 +117,7 @@ func testAzureBackend(payload string) *httptest.Server {
118117 } else if r.Method == "POST" && r.Body != nil {
119118 w.WriteHeader(200)
120119 w.Write([]byte(payload))
121− } else if r.Header.Get("Authorization") != "Bearer imaginary_access_token" {
120+ } else if !IsAuthorizedInHeader(r.Header) {
122121 w.WriteHeader(403)
123122 } else {
124123 w.WriteHeader(200)
@@ -134,7 +133,7 @@ func TestAzureProviderGetEmailAddress(t *testing.T) {
134133 bURL, _ := url.Parse(b.URL)
135134 p := testAzureProvider(bURL.Host)
136135
137− session := &sessions.SessionState{AccessToken: "imaginary_access_token"}
136+ session := CreateAuthorizedSession()
138137 email, err := p.GetEmailAddress(session)
139138 assert.Equal(t, nil, err)
140139 assert.Equal(t, "user@windows.net", email)
@@ -147,7 +146,7 @@ func TestAzureProviderGetEmailAddressMailNull(t *testing.T) {
147146 bURL, _ := url.Parse(b.URL)
148147 p := testAzureProvider(bURL.Host)
149148
150− session := &sessions.SessionState{AccessToken: "imaginary_access_token"}
149+ session := CreateAuthorizedSession()
151150 email, err := p.GetEmailAddress(session)
152151 assert.Equal(t, nil, err)
153152 assert.Equal(t, "user@windows.net", email)
@@ -160,7 +159,7 @@ func TestAzureProviderGetEmailAddressGetUserPrincipalName(t *testing.T) {
160159 bURL, _ := url.Parse(b.URL)
161160 p := testAzureProvider(bURL.Host)
162161
163− session := &sessions.SessionState{AccessToken: "imaginary_access_token"}
162+ session := CreateAuthorizedSession()
164163 email, err := p.GetEmailAddress(session)
165164 assert.Equal(t, nil, err)
166165 assert.Equal(t, "user@windows.net", email)
@@ -173,7 +172,7 @@ func TestAzureProviderGetEmailAddressFailToGetEmailAddress(t *testing.T) {
173172 bURL, _ := url.Parse(b.URL)
174173 p := testAzureProvider(bURL.Host)
175174
176− session := &sessions.SessionState{AccessToken: "imaginary_access_token"}
175+ session := CreateAuthorizedSession()
177176 email, err := p.GetEmailAddress(session)
178177 assert.Equal(t, "type assertion to string failed", err.Error())
179178 assert.Equal(t, "", email)
@@ -186,7 +185,7 @@ func TestAzureProviderGetEmailAddressEmptyUserPrincipalName(t *testing.T) {
186185 bURL, _ := url.Parse(b.URL)
187186 p := testAzureProvider(bURL.Host)
188187
189− session := &sessions.SessionState{AccessToken: "imaginary_access_token"}
188+ session := CreateAuthorizedSession()
190189 email, err := p.GetEmailAddress(session)
191190 assert.Equal(t, nil, err)
192191 assert.Equal(t, "", email)
@@ -199,7 +198,7 @@ func TestAzureProviderGetEmailAddressIncorrectOtherMails(t *testing.T) {
199198 bURL, _ := url.Parse(b.URL)
200199 p := testAzureProvider(bURL.Host)
201200
202− session := &sessions.SessionState{AccessToken: "imaginary_access_token"}
201+ session := CreateAuthorizedSession()
203202 email, err := p.GetEmailAddress(session)
204203 assert.Equal(t, "type assertion to string failed", err.Error())
205204 assert.Equal(t, "", email)