模拟混凝土类Mockito中的链接调用

时间:2017-08-26 01:47:26

标签: java unit-testing junit mockito powermockito

我正在使用Mockito服务来测试可以在MyFinalClass中抛出的异常并在MyAbstractClass中捕获,因为具体方法调用了MyFinalClass中的getObject方法。

请参阅以下代码。

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{
    ....
}

以下是我编写的测试类。不幸的是我遇到了Mockito.when(myFinalClass).getObject(name).thenReturn(Mockito.mock(Object.b))的问题,因为它似乎没有被调用。

@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(String name)方法尚未完成实现,所以我在调用getObject方法时抛出了新的MyException()。但是,由于Mockito.when(...)调用未被调用,因此异常不会被抛出,从而导致Assertion失败。

请帮忙。谢谢。

1 个答案:

答案 0 :(得分:2)

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);