Moq：设置一个模拟方法在第一次调用时失败，在第二次调用时成功

Posted 2023-02-26

【中文标题】Moq：设置一个模拟方法在第一次调用时失败，在第二次调用时成功

【英文标题】：Moq: Setup a mocked method to fail on the first call, succeed on the second

【发布时间】：2011-10-26 02:34:59

【问题描述】：

使用 Moq 模拟一个在第一次调用时会抛出异常，然后在第二次调用时成功的方法的最简洁的方法是什么？

【问题讨论】：

Moq 4.2 及更高版本现在通过SetupSequence() 支持此功能。

【参考方案1】：

到目前为止，我想出的最好的是：

interface IFoo

    void Bar();


[Test]
public void TestBarExceptionThenSuccess()

    var repository = new MockRepository(MockBehavior.Default);
    var mock = repository.Create<IFoo>();

    mock.Setup(m => m.Bar()).
        Callback(() => mock.Setup(m => m.Bar())). // Setup() replaces the initial one
        Throws<Exception>();                      // throw an exception the first time

    ...

【讨论】：

+1 这是一个很酷的方法。偶然发现代码的人可能不清楚那里发生了什么。

不错，没想到这个！

请注意，替换模拟会清除历史记录，因此您无法执行mock.Verify(m => m.Bar(), Times.Exactly(2))。

【参考方案2】：

我会使用Callback 并增加一个计数器来确定是否从Callback 抛出异常。

[Test]
public void TestMe()

    var count = 0;
    var mock = new Mock<IMyClass>();
    mock.Setup(a => a.MyMethod()).Callback(() =>
        
            count++;
            if(count == 1)
                throw new ApplicationException();
        );
    Assert.Throws(typeof(ApplicationException), () => mock.Object.MyMethod());
    Assert.DoesNotThrow(() => mock.Object.MyMethod());


public interface IMyClass

    void MyMethod();

【讨论】：

这是一个很好的答案，一旦您希望发生 2 件以上的事情，这种方法就会变得更可取。

这种方法很简单，很好，但请查看@Mathias 回答中 Phil Haack 的帖子。它很光滑。使用扩展方法你可以做类似reader.Setup(r => r.Read()).ReturnsInOrder(true, true, false);

ReturnsInOrder() 如果你想返回一个值序列，那么它很好，但是如何使用它来抛出异常作为序列的一部分并不明显。

这个答案没有错，但使用SetupSequence 肯定更容易、更优雅。检查@aghidini 的答案。

【参考方案3】：

Phil Haack 有一个有趣的blog post 来设置一个返回特定结果序列的方法。看起来这将是一个很好的起点，涉及一些工作，因为您现在需要的不是特定类型的值序列，而是可能是 T 类型或异常的结果序列。

【参考方案4】：

从 Moq 4.2 开始，您可以只使用内置方法 SetupSequence()（如 @RichardBarnett 评论所述）。

例子：

var mock = new Mock<IMyClass>();
mock.SetupSequence(x => x.MyMethod("param1"))
    .Throws<MyException>()
    .Returns("test return");

【讨论】：

这仅在MyMethod 具有返回类型时才有效。如果方法是void，那么你不能使用这种方法。如果有办法用void 方法做到这一点会很好