编程语言
首页 > 编程语言> > java – 是否可以在现有对象中模拟单个方法?

java – 是否可以在现有对象中模拟单个方法?

作者:互联网

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

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

有没有办法模拟deleteItem方法,以便在现有的DBClient对象中保留凭据,端点等等?

编辑:mockito在这种情况下不可用

解决方法:

您可以使用动态代理拦截所需的任何方法调用,因此您可以决定调用实际方法还是执行任何您想要的方法.

这是一个如何拦截方法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;
    }
}

标签:easymock,white-box,java,powermock
来源: https://codeday.me/bug/20190725/1534758.html