模拟方法以返回某些参数并为所有其他参数抛出异常

Question

我Moq用来模拟一个方法：

public class DbService : IDbService
{
    public Record GetRecord(Guid Id)
    {
        // Returns the record if exists.
        // Throws exception if record not found.
    }
}

我想模拟一下，如果我传递这三个 guid，它会返回记录，并且所有其他 Guid 都会抛出异常。

Guid id = Guid.NewGuid();
Guid id2 = Guid.NewGuid();
Guid id3 = Guid.NwGuid();

var mock = new Mock<IDbService>();
mock.Setup(t => t.GetRecord(id)).Returns(record1);
mock.Setup(t => t.GetRecord(id2)).Returns(record2);
mock.Setup(t => t.GetRecord(id3)).Returns(record3);
// Throw exception for all other guids...

我怎样才能做到这一点？

Answer 1

其实这很简单：

mock.Setup(t => t.GetRecord(It.IsAny<Guid>())).Throws(new MyException());
mock.Setup(t => t.GetRecord(id)).Returns(record1);
mock.Setup(t => t.GetRecord(id2)).Returns(record2);
mock.Setup(t => t.GetRecord(id3)).Returns(record3);

Answer 2

您可以MockBehavior.Strict为此使用：

Guid id = Guid.NewGuid();
Guid id2 = Guid.NewGuid();
Guid id3 = Guid.NwGuid();

var mock = new Mock<IDbService>(MockBehavior.Strict);

mock.Setup(t => t.GetRecord(id)).Returns(record1);
mock.Setup(t => t.GetRecord(id2)).Returns(record2);
mock.Setup(t => t.GetRecord(id3)).Returns(record3);

// do stuff

// will throw if GetRecord was called with arguments
// that don't match to any configured setup.
mock.VerifyAll();