func oauthConfigSender() autorest.Sender {
sender := mocks.NewSender()
resp := mocks.NewResponseWithStatus("", http.StatusUnauthorized)
mocks.SetResponseHeaderValues(resp, "WWW-Authenticate", []string{
`authorization_uri="https://testing.invalid/` + fakeTenantId + `"`,
})
sender.AppendResponse(resp)
return sender
}
func TestExtractHeaderValueRetrievesFirstValue(t *testing.T) {
r := mocks.NewResponse()
v := []string{"v1", "v2", "v3"}
mocks.SetResponseHeaderValues(r, mocks.TestHeader, v)
if ExtractHeaderValue(mocks.TestHeader, r) != v[0] {
t.Errorf("autorest: ExtractHeader failed to retrieve the expected header -- expected [%s]%v, received [%s]%v",
mocks.TestHeader, v[0], mocks.TestHeader, ExtractHeaderValue(mocks.TestHeader, r))
}
}
func TestExtractHeader(t *testing.T) {
r := mocks.NewResponse()
v := []string{"v1", "v2", "v3"}
mocks.SetResponseHeaderValues(r, mocks.TestHeader, v)
if !reflect.DeepEqual(ExtractHeader(mocks.TestHeader, r), v) {
t.Errorf("autorest: ExtractHeader failed to retrieve the expected header -- expected [%s]%v, received [%s]%v",
mocks.TestHeader, v, mocks.TestHeader, ExtractHeader(mocks.TestHeader, r))
}
}
func (*DiscoverySuite) TestDiscoverAuthorizationURIHeaderMismatch(c *gc.C) {
sender := mocks.NewSender()
resp := mocks.NewResponseWithStatus("", http.StatusUnauthorized)
mocks.SetResponseHeaderValues(resp, "WWW-Authenticate", []string{`foo bar baz`})
sender.AppendResponse(resp)
client := subscriptions.NewClient()
client.Sender = sender
_, err := azureauth.DiscoverAuthorizationURI(client, "subscription_id")
c.Assert(err, gc.ErrorMatches, `authorization_uri not found in WWW-Authenticate header \("foo bar baz"\)`)
}
func discoverAuthSender() *azuretesting.MockSender {
const fakeTenantId = "11111111-1111-1111-1111-111111111111"
sender := mocks.NewSender()
resp := mocks.NewResponseWithStatus("", http.StatusUnauthorized)
mocks.SetResponseHeaderValues(resp, "WWW-Authenticate", []string{
fmt.Sprintf(
`authorization_uri="https://testing.invalid/%s"`,
fakeTenantId,
),
})
sender.AppendResponse(resp)
return &azuretesting.MockSender{
Sender: sender,
PathPattern: ".*/subscriptions/" + fakeSubscriptionId,
}
}
func (*DiscoverySuite) TestDiscoverAuthorizationURI(c *gc.C) {
sender := mocks.NewSender()
resp := mocks.NewResponseWithStatus("", http.StatusUnauthorized)
mocks.SetResponseHeaderValues(resp, "WWW-Authenticate", []string{
`foo bar authorization_uri="https://testing.invalid/meep" baz`,
})
sender.AppendResponse(resp)
client := subscriptions.NewClient()
client.Sender = sender
authURI, err := azureauth.DiscoverAuthorizationURI(client, "subscription_id")
c.Assert(err, jc.ErrorIsNil)
c.Assert(authURI, jc.DeepEquals, &url.URL{
Scheme: "https",
Host: "testing.invalid",
Path: "/meep",
})
}