[ASP.NET Identity] 使用 Microsoft.Owin.Testing 測試 OAuth Server 和 Web API

2016-08-03

3350
0
ASP.NET Identity
2020-02-01

續上篇，https://dotblogs.com.tw/yc421206/2016/08/02/identity_oauth_owin_setup

隨著功能的演進，原本用 Fiddler 編寫的測試腳本，越來越不容易管理，Web API 的品質也越來越不穩定
有了 TestServer 之後，我的問題就一掃而空，接下來就來分享我的作法，

開發環境

Windows 10 Enterprise x64 CHT
VS2015 Update3 ENG

[TestInitialize]
public void Setup()
{
    Server = TestServer.Create(app =>
                               {
                                   var startup = new Startup();
                                   startup.Configuration(app);
                                   app.UseErrorPage(); // See Microsoft.Owin.Diagnostics
                                   app.UseWelcomePage("/Welcome"); // See Microsoft.Owin.Diagnostics
                                   var config = new HttpConfiguration();

                                   WebApiConfig.Register(config);

                                   app.UseWebApi(config);
                               });
}

然後使用 TestServer.CreateRequest 發送命令

protected virtual async Task<HttpResponseMessage> PostAsync<TModel>(string uri, TModel model)
{
    return await Server.CreateRequest(uri)
                       .And(
                           request =>
                           request.Content =
                           new ObjectContent(typeof(TModel), model, new JsonMediaTypeFormatter()))
                       .PostAsync();
}

protected virtual async Task<HttpResponseMessage> GetAsync(string uri)
{
    return await Server.CreateRequest(uri)
                       .GetAsync();
}

最後將這些方法集結成抽象類別

public abstract class BaseServerTest
{
    protected TestServer Server;


    protected abstract string Password { get; set; }

    protected abstract string Username { get; set; }

    [TestInitialize]
    public void Setup()
    {
       
        Server = TestServer.Create(app =>
                                   {
                                       var startup = new Startup();
                                       startup.Configuration(app);
                                       app.UseErrorPage(); // See Microsoft.Owin.Diagnostics
                                       app.UseWelcomePage("/Welcome"); // See Microsoft.Owin.Diagnostics
                                       var config = new HttpConfiguration();

                                       WebApiConfig.Register(config);

                                       app.UseWebApi(config);
                                   });
    }

    [TestCleanup]
    public void Teardown()
    {
        if (Server != null)
        {
            Server.Dispose();
        }
    }

    protected virtual async Task<HttpResponseMessage> GetAsync(string uri)
    {
        return await Server.CreateRequest(uri)
                           .GetAsync();
    }

    protected virtual async Task<HttpResponseMessage> PostAsync<TModel>(string uri, TModel model)
    {
        return await Server.CreateRequest(uri)
                           .And(
                               request =>
                               request.Content =
                               new ObjectContent(typeof(TModel), model, new JsonMediaTypeFormatter()))
                           .PostAsync();
    }

    protected virtual async Task<HttpResponseMessage> GetAsync(string uri, string accessToken)
    {
        return await Server.CreateRequest(uri)
                           .AddHeader("Authorization", "Bearer " + accessToken)
                           .GetAsync();
    }

    protected virtual async Task<HttpResponseMessage> PostAsync<TModel>(string uri, TModel model, string accessToken)
    {
        return await Server.CreateRequest(uri)
                           .AddHeader("Authorization", "Bearer " + accessToken)
                           .And(
                               request =>
                               request.Content =
                               new ObjectContent(typeof(TModel), model, new JsonMediaTypeFormatter()))
                           .PostAsync();
    }
}

Step3.加入測試方法

建立一個實作 BaseServerTest 的 AccountApiControllerTest 類，測試方法調用 Web API

測試註冊帳號

[TestMethod]
public async Task Register_Test()
{
    var model = new RegisterBindingModel
    {
        Email = Username,
        Password = Password,
        ConfirmPassword = Password
    };

    var response = await PostAsync(s_baseUri + "/Register", model);
    Assert.AreEqual(HttpStatusCode.OK, response.StatusCode);
}

測試OAuth登入

[TestMethod]
public async Task Login_Bearer_Test()
{
    var model = new RegisterBindingModel
    {
        Email = Username,
        Password = Password,
        ConfirmPassword = Password
    };

    await PostAsync(s_baseUri + "/Register", model);

    var form = new Dictionary<string, string>
    {
        {"grant_type", "password"},
        {"username", Username},
        {"password", Password}
    };


    var content = new FormUrlEncodedContent(form);
    var response =
        Server.HttpClient.PostAsync("/oauth/token", content).Result;

    var token = response.Content.ReadAsAsync<Token>(new[] {new JsonMediaTypeFormatter()}).Result;

    response.StatusCode.Should().Be(HttpStatusCode.OK);
    token.AccessToken.Should().NotBeNullOrEmpty();
}