在具体类Mockito中模拟链接调用

Question

public abstract class MyAbstractClass{

    @Autowired
    private MyFinalClass myFinalClass;

    //concrete method
    protected Object myConcreteMethod(String name){
        try{
            Object b = myFinalClass.getObject(name);
            return b;
        } catch(MyException e){
          LOGGER.log("ERROR THROWN" + e);
        }
    }

}

public final class MyFinalClass {

    public Object getObject(String name) throws MyException{
        **** HAS NOT YET BEEN IMPLEMENTED ****
        return null;
    }

}

public MyException extends Exception{
    ....
}

@ContextConfiguration(locations = "classpath:applicationContext-test.xml")
@RunWith(PowerMockRunner.class)
public class MyAbstractClassTest  {

    public static class ExampleConcreteClass extends MyAbstractClass{

    }

    @InjectMocks
    ExampleConcreteClass exampleConcreteClass;

    @Mock
    MyFinalClass myFinalClass;

    @Before
    public void setUp(){
        MockitoAnnotations.initMocks(this);
    }

    @Test
    public void testMyConcreteMethod(){
        try{
           Mockito.when(myFinalClass).getObject(name). 
           thenThrow(new MyException()); 

           exampleConcreteClass.myConcreteMethod();
           Assert.fail();
        }catch(MyException e){
           Assert.assertTrue(Boolean.TRUE);
        }
    }

}

因为MyFinalClass中的getObject（字符串名）方法的实现尚未完成，所以我在调用getObject方法时抛出新的MyException（）。但是，该异常不会作为mockito.when（...）调用未被调用，导致断言失败。

请帮帮忙。谢谢你。

Answer 1

Mockito无法模拟最终类。有几种方法可以绕过这个问题。

1）在MyAbstractClass中，考虑不引用最终类，而是引用MyFinalClass实现（或扩展）的接口或抽象类。因此假设MyFinalClass实现了一个名为MyGetTableAPI的接口。那么MyAbstractClass将如下所示。

 public abstract class MyAbstractClass{

    @Autowired
    private MyGettableApi gettableService;

    //concrete method
    protected Object myConcreteMethod(String name){
        try{
            Object b = gettableService.getObject(name);
            return b;
        } catch(MyException e){
          LOGGER.log("ERROR THROWN" + e);
        }
    }

}  


public final class MyFinalClass implements MyGettableApi {

    @Override
    public Object getObject(String name) throws MyException{
        **** HAS NOT YET BEEN IMPLEMENTED ****
        return null;
    }

}


public interface MyGettableApi {
    public Object getObject(String name);
}

现在，在您的测试代码中，您可以模拟MyGetTableAPI，因为它不是最终的，没有任何问题。

    @Mock
    MyGettableApi myFinalClass;

2）另一种方法是将调用包装到包装在另一个方法中的MyFinalClass.GetObject（name）；。

public abstract class MyAbstractClass{

    @Autowired
    private MyFinalClass myFinalClass;

    //concrete method
    protected Object myConcreteMethod(String name){
        try{
            Object b = myGetObjectWrapperMethod(name);
            return b;
        } catch(MyException e){
          LOGGER.log("ERROR THROWN" + e);
        }
    }

    Object myGetObjectWrapperMethod(String name) {
        return myFinalClass.getObject(name);
    }

}

现在，在您的测试代码中，您将@spy测试的MyAbstractClass实例。

@InjectMocks
@Spy
ExampleConcreteClass exampleConcreteClass;

然后您可以模拟我们的新包装方法。

doThrow(new MyException()).when(exampleConcreteClass).myGetObjectWrapperMethod(name);