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

导语

在 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

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