Update OpenIdConnectChallengeTests

1. Set header and status code in OnRedirectToIdentityProviderEventHandlesResponse
2. Move field to the top
This commit is contained in:
Troy Dai 2016-08-23 21:27:00 -07:00
parent c5c11e81b3
commit d887d74819
1 changed files with 7 additions and 3 deletions

View File

@ -14,6 +14,8 @@ namespace Microsoft.AspNetCore.Authentication.Tests.OpenIdConnect
{ {
public class OpenIdConnectChallengeTests public class OpenIdConnectChallengeTests
{ {
private static readonly string ChallengeEndpoint = TestServerBuilder.TestHost + TestServerBuilder.Challenge;
[Fact] [Fact]
public async Task ChallengeIsIssuedCorrectly() public async Task ChallengeIsIssuedCorrectly()
{ {
@ -259,7 +261,10 @@ namespace Microsoft.AspNetCore.Authentication.Tests.OpenIdConnect
{ {
OnRedirectToIdentityProvider = context => OnRedirectToIdentityProvider = context =>
{ {
context.Response.StatusCode = 410;
context.Response.Headers.Add("tea", "Oolong");
context.HandleResponse(); context.HandleResponse();
return Task.FromResult(0); return Task.FromResult(0);
} }
}; };
@ -270,7 +275,8 @@ namespace Microsoft.AspNetCore.Authentication.Tests.OpenIdConnect
var transaction = await server.SendAsync(ChallengeEndpoint); var transaction = await server.SendAsync(ChallengeEndpoint);
var res = transaction.Response; var res = transaction.Response;
Assert.Equal(HttpStatusCode.OK, res.StatusCode); Assert.Equal(HttpStatusCode.Gone, res.StatusCode);
Assert.Equal("Oolong", res.Headers.GetValues("tea").Single());
Assert.Null(res.Headers.Location); Assert.Null(res.Headers.Location);
} }
@ -316,7 +322,5 @@ namespace Microsoft.AspNetCore.Authentication.Tests.OpenIdConnect
Assert.StartsWith(".AspNetCore.Correlation.OpenIdConnect.", secondCookie); Assert.StartsWith(".AspNetCore.Correlation.OpenIdConnect.", secondCookie);
Assert.Contains("expires", secondCookie); Assert.Contains("expires", secondCookie);
} }
private static string ChallengeEndpoint => TestServerBuilder.TestHost + TestServerBuilder.Challenge;
} }
} }