使用不同的值两次调用方法 使用 MOQ 进行单元测试

Question

我想测试一个构造，它在其中调用一个方法两次以获得两个不同的值

public class  stubhandler
{

public stubhandler()
{

string codetext = model.getValueByCode(int a,string b); // 1,"High"    result Canada
string datatext = model.getValueByCode(int a,string b); // 10, "Slow"   result Motion

}

}

为了测试上面的内容，我使用了一个单元测试类

[TestMethod]
public void StubHandlerConstructor_Test()
{
Mock<Model> objMock = new Mock<>(Model);
objMock.Setup(m => m.getValueByCode(It.IsAny<int>,It.IsAny<string>)).Returns("Canada");

objMock.Setup(m => m.getValueByCode(It.IsAny<int>,It.IsAny<string>)).Returns("Motion");

stubhandler  classstubhandler = new stubhandler();

}

上述方法通过但codetext和datatext包含相同的值Motion 我希望他们设置为

codetext = Canada
datatext = Motion

我怎样才能做到这一点？

我试过objMock.VerifyAll() 没通过测试？？

Answer 1

如果使用 MOQ 4，可以使用 SetupSequence，否则可以使用 lambda 完成

使用 SetupSequence 很容易解释。

使用 lambdas 并不太麻烦。重要的一点是返回值是在声明设置时设置的。如果刚刚使用过

mockFoo.Setup(mk => mk.Bar()).Returns(pieces[pieceIdx++]);

设置总是返回pieces[0]。通过使用 lambda，评估被推迟到 Bar() 被调用。

public interface IFoo {
    string Bar();
}

public class Snafu {

    private IFoo _foo;
    public Snafu(IFoo foo) {
        _foo = foo;
    }

    public string GetGreeting() {
        return string.Format("{0} {1}",
                             _foo.Bar(),
                             _foo.Bar());
    }

}

[TestMethod]
public void UsingSequences() {

    var mockFoo = new Mock<IFoo>();
    mockFoo.SetupSequence(mk => mk.Bar()).Returns("Hello").Returns("World");

    var snafu = new Snafu(mockFoo.Object);

    Assert.AreEqual("Hello World", snafu.GetGreeting());

}

[TestMethod]
public void NotUsingSequences() {

    var pieces = new[] {
            "Hello",
            "World"
    };
    var pieceIdx = 0;

    var mockFoo = new Mock<IFoo>();
    mockFoo.Setup(mk => mk.Bar()).Returns(()=>pieces[pieceIdx++]);

    var snafu = new Snafu(mockFoo.Object);

    Assert.AreEqual("Hello World", snafu.GetGreeting());

}

Answer 2

Moq documentation 说你可以模拟类似使用Callback 方法连续返回的东西：

var values = new [] { "Canada", "Motion" };
int callNumber = 0;

mock.Setup(m => m.getValueByCode(It.IsAny<int>(), It.IsAny<string>()))
    .Returns((i,s) => values[callNumber])
    .Callback(() => callNumber++);

这可以解决问题，但它不是最优雅的解决方案。 Matt Hamilton proposes 在他的博文中更好，巧妙地使用了队列：

var values = new Queue<string> { "Canada", "Motion" };

mock.Setup(m => m.getValueByCode(It.IsAny<int>(), It.IsAny<string>()))
    .Returns(() => values.Dequeue());

调用mock.Object.getValueByCode 两次，将分别产生"Canada" 和"Motion" 字符串。