压缩代码:SetupSequence在连续调用中返回不同的值

时间:2016-08-31 21:57:46

标签: c# unit-testing moq

我正在测试一个调用其他方法的方法。我有一个使用此方法生成模拟连接对象的工作测试:

    private Mock<IDatabaseConnection> MockOutGetControlDocInfoData()
    {
        Mock<IDatabaseConnection> mockConn = new Mock<IDatabaseConnection>();
        List<Mock<IDbCommand>> mockCmds = new List<Mock<IDbCommand>>();
        List<long> vals = new List<long>() { 2, 2, 2, 2, 10, 2, 2, 2, 2, 2 };
        foreach (long val in vals)
        {
            mockCmds.Add(CreateMockCmdObjectWithReturnValue(val));
        }
        mockConn.SetupAllProperties();
        mockConn.Setup(c => c.Conn.ConnectionString).Returns("What the heck.");
        mockConn.SetupSequence(c => c.CreateCommand(It.IsAny<string>()))
            .Returns(mockCmds[0].Object)
            .Returns(mockCmds[1].Object)
            .Returns(mockCmds[2].Object)
            .Returns(mockCmds[3].Object)
            .Returns(mockCmds[4].Object)
            .Returns(mockCmds[5].Object)
            .Returns(mockCmds[6].Object)
            .Returns(mockCmds[7].Object)
            .Returns(mockCmds[8].Object)
            .Returns(mockCmds[9].Object);
        return mockConn;
    }

我对SetupSequence返回感到不满意,看起来它应该是循环的一部分,但我不知道如何将多个返回值放入SetupSequence中。有什么改进的想法吗?

至少现在,我可以手动创建测试值列表。

它不应该相关,但如果需要,我可以提供CreateMockCmdObjectWithReturnValue代码。

1 个答案:

答案 0 :(得分:3)

var sequence = mockConn.SetupSequence(c => c.CreateCommand(It.IsAny<string>()));
foreach (var cmd in mockCmds)
{
   sequence = sequence.Returns(cmd.Object);
}