是否可以在已经存在的对象中模拟单个方法

Question

对于集成测试，我需要在不破坏其中其余信息的情况下模拟java服务客户机中的特定方法。它没有自构造函数，所以像这样的解决方案是不可能的:

private DBClient mockClient = new DBClient(alreadyExistingClient){
    @Override
    void deleteItem(Item i){
        //my stuff goes here
    }
};

是否有一种方法来模拟deleteItem方法，使凭据，端点等…保存在现有的DBClient对象吗?

edit: mockito在这种情况下不可用

Answer 1

你可以使用动态代理来拦截任何你想要的方法调用，所以你可以决定是调用真正的方法还是做任何你想做的。

这是一个如何拦截方法Set.add()的例子，您可以对deleteItem()做完全相同的操作

package example.dynamicproxy;
import java.lang.reflect.InvocationHandler;
import java.lang.reflect.Method;
import java.lang.reflect.Proxy;
import java.util.Set;
public class SetProxyFactory {
    public static Set<?> getSetProxy(final Set<?> s) {
        final ClassLoader classLoader = s.getClass().getClassLoader();
        final Class<?>[] interfaces = new Class[] {Set.class};
        final InvocationHandler invocationHandler = new InvocationHandler() {
            @Override
            public Object invoke(final Object proxy, final Method method, final Object[] args) throws Throwable {
                if (method.getName().equals("add")) {
                    System.out.println("add() intercepted");
                    // do/return whatever you want
                }
                // or invoke the real method
                return method.invoke(s, args);
            }
        };
        final Object proxy = Proxy.newProxyInstance(classLoader, interfaces, invocationHandler);
        return (Set<?>) proxy;
    }
}

Answer 2

您可以使用low -fi并创建DBClient类的子类。向这个子类传递要模拟的DBClient实例。

在子类中使用组合，并将所有方法调用委托给原始DBClient，除了您想要模拟的那个。将模拟实现添加到所需的方法中。

它不像mock框架那样可重用，但应该可以工作。

DBClient mockDbClient = new DBClient() {
     private DBClient dbClientDelegate;
     public void setDelegate(DBClient dbClient) {
         dbClientDelegate = dbClient;
    }
    //override all methods. 
    //delegate to the corresponding method of the dbClientDelegate instance
    //overide the method you want to mock, add asserts for method arguments
    //return mock data as appropriate
}
mockDbClient.setDelegate(preinstantiatedDbClient);
//inject mockDbClient to test class
//call test class / method

Answer 3

在Mockito 2+中，您可以使用spy功能来实现此目的:

    PrintStream realSystemOut = System.out;
    realSystemOut.println("XXX");
    PrintStream mockedSystemOut = Mockito.spy(realSystemOut);
    Mockito.doNothing().when(mockedSystemOut).println(Mockito.anyString());
    mockedSystemOut.println("YYY");

输出:

XXX