Làm thế nào để làm cho một giả ném một ngoại lệ lần đầu tiên và trả về một giá trị thứ hai

Question

Tôi đang sử dụng Moq như khuôn khổ mocking của tôi và tôi cần phải kiểm tra một lớp học khi một loại ngoại lệ cụ thể được chạy nó sẽ tiếp tục cố gắng cho đến khi tình hình được giải quyết một khi điều đó xảy ra việc thực hiện kết thúc.

Vì vậy, những gì tôi cần là một cái gì đó tương tự như:

myMock = Mock<IFoo>(); 

myMock.Setup(m => m.Excecute()).Throws<SpecificException>(); 
myMock.Setup(m => m.Execute()); 

var classUnderTest = MyClass(myMock); 
classUnderTest.DoSomething(); 

Assert.AreEqual(expected, classUnderTest.Result); 

Nhờ sự giúp đỡ bạn có thể cho.

Answer 1

Đây là một cách, dựa trên ví dụ Moq QuickStart trả về các giá trị khác nhau cho mỗi lệnh gọi.

var mock = new Mock<IFoo>(); 
var calls = 0; 
mock.Setup(foo => foo.GetCountThing()) 
    .Returns(() => calls) 
    .Callback(() => 
    { 
     calls++; 
     if (calls == 1) 
     { 
      throw new InvalidOperationException("foo"); 
     } 
    }); 

try 
{ 
    Console.WriteLine(mock.Object.GetCountThing()); 
} 
catch (InvalidOperationException) 
{ 
    Console.WriteLine("Got exception"); 
} 

Console.WriteLine(mock.Object.GetCountThing()); 

Nếu phương pháp này trả về void, sử dụng:

var myMock = new Mock<IFoo>(); 
bool firstTimeExecuteCalled = true; 

myMock.Setup(m => m.Execute()) 
     .Callback(() => 
     { 
      if (firstTimeExecuteCalled) 
      { 
       firstTimeExecuteCalled = false; 
       throw new SpecificException(); 
      } 
     }); 

try 
{ 
    myMock.Object.Execute(); 
} 
catch (SpecificException) 
{ 
    // Would really want to call Assert.Throws instead of try..catch. 
    Console.WriteLine("Got exception"); 
} 

myMock.Object.Execute(); 
Console.WriteLine("OK!"); 

Answer 2

Tại sao không thực sự viết đối tượng thử nghiệm của riêng bạn mà thực hiện điều này? Nếu nó chỉ sẽ được sử dụng để thử nghiệm tức là một cái gì đó như:

public class Mock : IFoo 
{ 
    private int _calls; 

    public Mock() 
    { 
     _calls = 0; 
    } 

    public int Execute() 
    { 
     _calls++; 

     if (_calls == 1) 
      throw new Exception(); 

     return value; 

    } 

}