Mockito - 监视真实对象称为原始方法

时间:2012-02-02 09:47:08

标签: java mockito

想象一下以下代码:

List list = .....
List spy = spy(list);
doThrow(new NullpointerException()).when(spy).get(0);

doThrow(....)执行list.get(0) - 这完全没有意义。我想定义模拟行为而不是在这里调用方法.....我错过了什么吗?

编辑: 列表由CGLIB装饰。当我删除CGLIB代理时,Mockito按预期工作。任何想法在使用CGLIB代理时如何解决这样的问题?

2 个答案:

答案 0 :(得分:8)

import static org.mockito.Mockito.doThrow;
import static org.mockito.Mockito.spy;

import java.lang.reflect.Method;

import org.junit.Test;

import net.sf.cglib.proxy.Enhancer;
import net.sf.cglib.proxy.MethodInterceptor;
import net.sf.cglib.proxy.MethodProxy;

public class MockitoSpyTest {

    @Test
    public void execTest() {

        System.out.println("*** TEST 1 ***");
        System.out.println("Test on unmodified object");
        MySet ms = new MySetImpl();
        ms.set("test value");
        System.out.println("Set contains: " + ms.get());

        // decorate ms1 with easymock
        System.out.println("\n*** TEST 2 ***");
        MySet spyMs = spy(ms);
        doThrow(new NullPointerException("my test nullpointer")).when(spyMs).get();
        System.out.println("Test decorated object with SPY");
        spyMs.set("test value");
        try {
            System.out.println("Set contains: " + spyMs.get());
        } catch (NullPointerException e) {
            System.out.println("NullPointerException - as expected");
        }

        // Enhance call with CGLIB
        System.out.println("\n*** TEST 3 ***");
        System.out.println("Test on CGLIB decorated object");
        Enhancer enc = new Enhancer();
        enc.setSuperclass(MySetImpl.class);
        enc.setInterfaces(new Class[] { MySet.class });
        enc.setCallback(new MethodInterceptor() {

            @Override
            public Object intercept(Object obj, Method method, Object[] args, MethodProxy proxy) throws Throwable {
                if ("get".equals(method.getName())) {
                    System.out.println("CGLIB decorated GET call");
                }
                return proxy.invokeSuper(obj, args);
            }
        });
        MySet ms1 = (MySet) enc.create();
        ms1.set("test value");
        System.out.println("Set contains: " + ms1.get());

        // decorate ms1 with easymock
        System.out.println("\n*** TEST 4 ***");
        System.out.println("Test on CGLIB decorated object with SPY");
        MySet spyMs1 = spy(ms1);
        doThrow(new NullPointerException("my test nullpointer")).when(spyMs1).get();
        spyMs1.set("test value");
        System.out.println("Set contains: " + spyMs1.get());
    }

    public interface MySet {
        void set(String val);

        String get();
    }

    public static class MySetImpl implements MySet {
        String val;

        public void set(String val) {
            this.val = val;
            System.out.println("Original SET call:" + val);
        }

        public String get() {

            System.out.println("Original GET call:" + val);
            return val;
        }

    }
}

上面的示例产生输出:

*** TEST 1 ***
Test on unmodified object
Original SET call:test value
Original GET call:test value
Set contains: test value

*** TEST 2 ***
Test decorated object with SPY
Original SET call:test value
NullPointerException - as expected

*** TEST 3 ***
Test on CGLIB decorated object
Original SET call:test value
CGLIB decorated GET call
Original GET call:test value
Set contains: test value

*** TEST 4 ***
Test on CGLIB decorated object with SPY
CGLIB decorated GET call
Original GET call:test value
Original SET call:test value
CGLIB decorated GET call
Original GET call:test value
Set contains: test value

现在TEST 2TEST 4应该在NullPointerException来电get - 基于模拟间谍:doThrow(new NullPointerException("my test nullpointer")).when(spyMs1).get();

“TEST 4”不会抛出预期的异常,因为它已经用CGLIB修饰 - 我们还可以在控制台上看到正在执行CGLIb调用:GLIB decorated GET call而不是调用spy对象。使用带有CGLIB代理的Spring AOP时,可以获得相同的效果。

答案 1 :(得分:0)

Mockito.doThrow(new NullpointerException()).when(spy).get(0);

我认为这里的问题是你正在尝试进行局部模拟,所以你必须在测试类上有注释:

@PrepareForTest(List.class)

这可能有效,也可能无效。看看我测试异常处理的代码,我总是在完全模拟的对象上完成它,而不是部分模拟的对象。此外,我在部分模拟时广泛使用了PowerMockito,因此库可能会满足您的需求。