点击上方蓝字关注“汪宇杰博客”

导语

在 ASP.NET Core 中，当你在 UrlHelperExtensions 类上使用扩展方法时，很难在单元测试中编写Mock。因为Moq框架不支持模拟扩展方法。

问题

例如，我的博客代码中使用了 Url.Page() 方法：

var callbackUrl = Url.Page("/Index", null, null, Request.Scheme);

但是单元测试中，像这样 Mock 就会爆：

var mockUrlHelper = new Mock<IUrlHelper>(MockBehavior.Strict);mockUrlHelper.Setup(x => x.Page("/Index", null, null, It.IsAny<string>())).Returns("callbackUrl").Verifiable();

爆炸现场

System.NotSupportedException : Unsupported expression: x => x.Page("/Index", null, null, It.IsAny<string>())    Extension methods (here: UrlHelperExtensions.Page) may not be used in setup / verification expressions.

解决方法

我们需要 Mock 这个拓展方法调用的底层方法。在本案例中，底层方法是

Microsoft.AspNetCore.Mvc.IUrlHelper.RouteUrl(UrlRouteContext routeContext)

我是怎么知道的呢？很简单，.NET 都已经开源多少年了，直接看一眼源代码就能知道微软如何单元测试 UrlHelperExtensions。

https://source.dot.net/ 

从微软的代码里复制两个助手方法

private Mock<IUrlHelper> CreateMockUrlHelper(ActionContext context = null)

{

    context ??= GetActionContextForPage("/Page");

    var urlHelper = _mockRepository.Create<IUrlHelper>();

    urlHelper.SetupGet(h => h.ActionContext)

        .Returns(context);

    return urlHelper;

}

private static ActionContext GetActionContextForPage(string page)

{

    return new()

    {

        ActionDescriptor = new()

        {

            RouteValues = new Dictionary<string, string>

            {

                { "page", page },

            }

        },

        RouteData = new()

        {

            Values =

            {

                [ "page" ] = page

            }

        }

    };

}

修改我们的单元测试

var mockUrlHelper = CreateMockUrlHelper();mockUrlHelper.Setup(h => h.RouteUrl(It.IsAny<UrlRouteContext>())).Returns("callbackUrl");

现在单元测试就能顺利跑过了！

完整的单元测试代码见下方供参考：

[Test]

public async Task SignOutAAD()

{

    _mockOptions.Setup(m => m.Value).Returns(new AuthenticationSettings

    {

        Provider = AuthenticationProvider.AzureAD

    });

    var mockUrlHelper = CreateMockUrlHelper();

    mockUrlHelper.Setup(h => h.RouteUrl(It.IsAny<UrlRouteContext>()))

        .Returns("callbackUrl");

    var ctx = new DefaultHttpContext();

    var ctl = CreateAuthController();

    ctl.ControllerContext = new() { HttpContext = ctx };

    ctl.Url = mockUrlHelper.Object;

    var result = await ctl.SignOut();

    Assert.IsInstanceOf(typeof(SignOutResult), result);

}

汪宇杰博客

Azure | .NET | 微软 MVP

无广告，不卖课，做纯粹的技术公众号