How do I mock controller context in my unit test s

2019-03-18 07:59发布

问题:

I am attempting to create a unit test for my controller, but the action I am testing uses a partial view to string function which doesn't want to work in my tests.

private string RenderPartialViewToString(string viewName, object model = null)
{
   if (string.IsNullOrEmpty(viewName))
            viewName = ControllerContext.RouteData.GetRequiredString("action");

   ViewData.Model = model;

   using (System.IO.StringWriter sw = new System.IO.StringWriter())
   {
      ViewEngineResult viewResult = ViewEngines.Engines.FindPartialView(ControllerContext, viewName);
      ViewContext viewContext = new ViewContext(ControllerContext, viewResult.View, ViewData, TempData, sw);
      viewResult.View.Render(viewContext, sw);

      return sw.GetStringBuilder().ToString();
   }
}

This gives me an error of "Object reference not set to an instance of an object" on the line ViewEngineResult viewResult = ViewEngines.Engines.FindPartialView(ControllerContext, viewName);

My setup in the unit test for the controller is (with a few bits removed to simplify it):

var mock = new Mock<ControllerContext>();
mock.SetupGet(p => p.HttpContext.User.Identity.Name).Returns(userName);
if (userName != null)
{
   mock.SetupGet(p => p.HttpContext.Request.IsAuthenticated).Returns(true);
   mock.SetupGet(p => p.HttpContext.User.Identity.IsAuthenticated).Returns(true);
}
else
{
   mock.SetupGet(p => p.HttpContext.Request.IsAuthenticated).Returns(false);
}
var controller = new BlogController();
controller.ControllerContext = mock.Object;

I've not had any luck trying to find a solution or work around. Any help appreciated. Thanks.


As suggested I have tried setting up route data but still getting the error. This is what I have added:

var routeData = new RouteData();
routeData.Values.Add("controller", "BlogController");
mock.SetupGet(m => m.RouteData).Returns(routeData);

回答1:

Final solution thanks to help in the comments.

var mock = new Mock<ControllerContext>();
mock.SetupGet(p => p.HttpContext.User.Identity.Name).Returns(userName);
if (userName != null)
{
   mock.SetupGet(p => p.HttpContext.Request.IsAuthenticated).Returns(true);
   mock.SetupGet(p => p.HttpContext.User.Identity.IsAuthenticated).Returns(true);
}
else
{
   mock.SetupGet(p => p.HttpContext.Request.IsAuthenticated).Returns(false);
}

var routeData = new RouteData();
routeData.Values.Add("controller", "BlogController");
mock.SetupGet(m => m.RouteData).Returns(routeData);

var view = new Mock<IView>();
var engine = new Mock<IViewEngine>();
var viewEngineResult = new ViewEngineResult(view.Object, engine.Object);
engine.Setup(e => e.FindPartialView(It.IsAny<ControllerContext>(), It.IsAny<string>(), It.IsAny<bool>())).Returns(viewEngineResult);
ViewEngines.Engines.Clear();
ViewEngines.Engines.Add(engine.Object);

var controller = new BlogController();
controller.ControllerContext = mock.Object;


回答2:

Here's a version using AutoMoq that renders a string of your choice.

Subject.ControllerContext = new ControllerContext(
    Mocked<HttpContextBase>().Object,
    new RouteData(),
    Subject);

ViewEngines.Engines.Clear();
ViewEngines.Engines.Add(Mocked<IViewEngine>().Object);
Mocked<IViewEngine>()
    .Setup(x => x.FindPartialView(Subject.ControllerContext,
                                  It.IsAny<string>(), It.IsAny<bool>()))
    .Returns(new ViewEngineResult(Mocked<IView>().Object,
                                  Mocked<IViewEngine>().Object));
Mocked<IView>()
    .Setup(x => x.Render(It.IsAny<ViewContext>(), It.IsAny<TextWriter>()))
    .Callback((ViewContext c, TextWriter w) => w.WriteLine("RENDERED"));