[英]Mocking user IP address in MVC unit test with NSubstitute
I have the following method in my HomeController.我的 HomeController 中有以下方法。 The purpose is to split users based on IP address to test different versions of the home page:
目的是根据IP地址拆分用户,测试不同版本的首页:
[HttpGet]
public ActionResult Index()
{
var userIpAddress = GetUserIpAddress();
if (IsIpAddressOddOrEven(userIpAddress))
{
return RedirectToAction(HomePage);
}
return RedirectToAction(HomePageAlternative);
}
The GetUserIpAddress
method: GetUserIpAddress
方法:
private string GetUserIpAddress()
{
HttpContext context = System.Web.HttpContext.Current;
var ipAddress = context.Request.ServerVariables["HTTP_X_FORWARDED_FOR"];
if (!string.IsNullOrEmpty(ipAddress))
{
string[] addresses = ipAddress.Split(',');
if (addresses.Length != 0)
{
return addresses[0];
}
}
return context.Request.ServerVariables["REMOTE_ADDR"];
}
I want to write a unit test to ensure that this works properly.我想编写一个单元测试以确保它正常工作。 However, every time the unit test runs it is just taking the IP address I currently have.
但是,每次单元测试运行时,它只是使用我目前拥有的 IP 地址。 I am struggling to work out how to mock the result of the 'GetUserIpAddress' method to return an odd or even string.
我正在努力研究如何模拟“GetUserIpAddress”方法的结果以返回奇数或偶数字符串。 My attempt so far:
到目前为止我的尝试:
[Test]
public void Test()
{
var controller = CreateMvcController<HomeController>();
var result = controller.Index();
controller.HttpContext.Request.ServerVariables["HTTP_X_FORWARDED_FOR"].Returns("1");
Assert.IsInstanceOf<RedirectToRouteResult>(result);
var redirectToRouteResult = result as RedirectToRouteResult;
Assert.AreEqual(HomeController.HomePage, redirectToRouteResult.RouteValues["action"]);
}
I got the error that the result of controller.HttpContext.Request.ServerVariables["HTTP_X_FORWARDED_FOR"].Returns("1");
我得到的错误是
controller.HttpContext.Request.ServerVariables["HTTP_X_FORWARDED_FOR"].Returns("1");
is not a string but a HttpResponseBase, and in any case I am not convinced I am going about this the right way.不是一个字符串,而是一个 HttpResponseBase,无论如何我都不相信我会以正确的方式去做。 Please can someone point me in the right direction to do this?
请有人能指出我正确的方向吗? Thank you
谢谢
I suggest you to move GetUserIpAddress
method to a helper class which you can inject into HomeController
.我建议您将
GetUserIpAddress
方法移至可以注入HomeController
的助手 class 。 Therefore you can mock it while doing the unit tests.因此,您可以在进行单元测试时对其进行模拟。
Your HomeController
will be like this您的
HomeController
将是这样的
public HomeController(IUserIpAddressHelper userIpAddressHelper)
{
_userIpAddressHelper = userIpAddressHelper;
}
[HttpGet]
public ActionResult Index()
{
var userIpAddress = _userIpAddressHelper.GetUserIpAddress(System.Web.HttpContext.Current);
if (_userIpAddressHelper.IsIpAddressOddOrEven(userIpAddress))
{
return RedirectToAction(HomePage);
}
return RedirectToAction(HomePageAlternative);
}
So, you'll be able to mock UserIpAddressHelper
and inject it when writing the test.因此,您将能够模拟
UserIpAddressHelper
并在编写测试时注入它。
public void Test()
{
var userIpAddressHelper = Substitute.For<IUserIpAddressHelper>();
userIpAddressHelper.GetUserIpAddress(Arg.Any<HttpContext>()).Returns("0.0.0.0");
var controller = new HomeController(userIpAddressHelper);
var result = controller.Index();
Assert.IsInstanceOf<RedirectToRouteResult>(result);
var redirectToRouteResult = result as RedirectToRouteResult;
Assert.AreEqual(HomeController.HomePage, redirectToRouteResult.RouteValues["action"]);
}
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.