如何在单元测试中Mock并注入ViewData以验证代码逻辑？

场景说明 #

你的代码中存在两处核心逻辑：

1. 向ViewData写入数据：

ViewData.Add("MyData", myDataObject);

2. 读取ViewData并根据值触发服务调用：

var myDataObject = ViewData["MyData"] as MyDataObject;
if(myDataObject.MyProp == "somevalue"){ myService.GetSomething() }

需要通过单元测试验证：当ViewData中MyData的MyProp为"somevalue"时，myService.GetSomething("somevalue")被调用一次。

实现步骤与代码示例 #

1. 定义基础类型 #

先明确测试涉及的核心类与接口：

// 自定义数据对象
public class MyDataObject
{
    public string MyProp { get; set; }
}

// 服务接口
public interface IMyService
{
    void GetSomething(string value);
}

// 待测试的控制器
public class MyController : Controller
{
    private readonly IMyService _myService;

    public MyController(IMyService myService) => _myService = myService;

    // 向ViewData写入数据的方法
    public IActionResult SetViewData()
    {
        var data = new MyDataObject { MyProp = "somevalue" };
        ViewData.Add("MyData", data);
        return View();
    }

    // 读取ViewData并执行逻辑的方法
    public IActionResult ProcessViewData()
    {
        var data = ViewData["MyData"] as MyDataObject;
        if(data?.MyProp == "somevalue")
        {
            _myService.GetSomething("somevalue");
        }
        return View();
    }
}

2. 编写单元测试 #

以xUnit + Moq为例，直接操作Controller的ViewData完成测试：

using Xunit;
using Moq;

public class MyControllerTests
{
    [Fact]
    public void ProcessViewData_WithMatchingMyProp_CallsGetSomethingOnce()
    {
        // 模拟服务实例
        var mockService = new Mock<IMyService>();
        // 创建控制器，注入模拟服务
        var controller = new MyController(mockService.Object);
        
        // 直接向控制器的ViewData添加测试数据
        controller.ViewData.Add("MyData", new MyDataObject { MyProp = "somevalue" });
        
        // 执行待测试的方法
        controller.ProcessViewData();
        
        // 验证服务方法是否按预期被调用
        mockService.Verify(m => m.GetSomething("somevalue"), Times.Once);
    }

    [Fact]
    public void ProcessViewData_WithNonMatchingMyProp_DoesNotCallGetSomething()
    {
        var mockService = new Mock<IMyService>();
        var controller = new MyController(mockService.Object);
        
        controller.ViewData.Add("MyData", new MyDataObject { MyProp = "othervalue" });
        controller.ProcessViewData();
        
        mockService.Verify(m => m.GetSomething(It.IsAny<string>()), Times.Never);
    }
}

关键要点 #

• Controller的ViewData本质是ViewDataDictionary类型，单元测试中可直接访问、修改，无需额外注入工具
• 如果测试的是独立于Controller的逻辑类，可将ViewDataDictionary作为依赖注入或方法参数传入，测试时构造好目标数据的ViewDataDictionary即可
• 使用Moq的Verify方法可精准验证服务方法的调用次数、参数是否符合预期

内容的提问来源于stack exchange，提问作者Robert Corvus