SetupSequence in Moq

Question

I want a mock that returns 0 the first time, then returns 1 anytime the method is called thereafter. The problem is that if the method is called 4 times, I have to write:

mock.SetupSequence(x => x.GetNumber())     .Returns(0)     .Returns(1)     .Returns(1)     .Returns(1); 

Otherwise, the method returns null.

Is there any way to write that, after the initial call, the method returns 1?

Answer 1

The cleanest way is to create a Queue and pass .Dequeue method to Returns

.Returns(new Queue<int>(new[] { 0, 1, 1, 1 }).Dequeue);

Answer 2

That's not particulary fancy, but I think it would work:

    var firstTime = true;      mock.Setup(x => x.GetNumber())         .Returns(()=>                         {                             if(!firstTime)                                 return 1;                              firstTime = false;                             return 0;                         });