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

Question

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

Answer 1

我会使用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();
}

Answer 2

从Moq 4.2开始,您可以使用内置方法SetupSequence()(如@RichardBarnett评论所述).

例:

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

Answer 3

到目前为止,我提出的最好的是:

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

    ...
}

Answer 4

Phil Haack有一篇关于设置返回特定结果序列的方法的有趣博客文章.它似乎是一个很好的起点,涉及一些工作,因为你需要现在有一系列结果可以是T类型或异常,而不是某种类型的值序列.