JMockit 有没有办法从模拟方法调用原始方法？

Question

在我的模拟类中，我正在模拟方法 foo()。 对于某些测试用例，我希望 foo() 的模拟实现返回一个特殊值。 对于其他测试用例，我想使用 foo() 的实际实现。 我在模拟类中定义了一个布尔值，以便我可以在模拟方法中确定是否要返回特殊值，还是使用“真实”方法。 问题是，我似乎无法弄清楚如何从模拟方法中调用真正的方法。

我发现您可以在名为“it”的模拟对象中定义一个特殊成员（具有被模拟对象的类型）。 这允许您从模拟实现中引用真实的类。 所以，我的计划是，如果我需要调用 foo() 的“真实”实现，模拟方法将调用 it.foo()。 但是，这不起作用，因为调用 it.foo() 只是再次调用模拟版本，而不是真实版本，所以我最终会无限递归。

有什么方法可以让这项工作发挥作用吗？

编辑：使用代码示例可能会更清楚，这是我当前的模拟方法实现的样子：

private RealClass it;
...
public SomeClass foo() {
    if(fakeIt) {
        return new SomeClass("fakevalue");
    } else {
        // doesn't work, just keeps calling the mock foo
        // in infinite recursion
        return it.foo();
    }
}

编辑2：另外，对于我的大多数测试用例，我不想要模拟实现。 因此，我最初的尝试是仅在需要模拟对象的测试用例中调用 Mockit.redefineMethods() 。 但这不起作用 - 似乎你只能在安装/拆卸中执行此操作...当我尝试这样做时，我的模拟实现从未被调用。

解决方案的注释：

起初我认为给出的答案不起作用，但在玩了一些之后，问题似乎是我将 JMockit“核心”方法与“注释”驱动方法混合在一起。 显然，在使用注释时，您需要使用 Mockit.setupMocks，而不是 Mockit.redefineMethods()。 这就是最终起作用的：

@Before 
public void setUp() throws Exception
{
    Mockit.setUpMocks(MyMockClass.class);
}

然后，对于模拟类：

@MockClass(realClass = RealClass.class)
public static class MyMockClass {
    private static boolean fakeIt = false;
    private RealClass it;

    @Mock(reentrant = true)
    public SomeClass foo() {
        if(fakeIt) {
            return new SomeClass("fakevalue");
        } else {
            return it.foo();
        }
    }
}

Answer 1

在 JMockit 的较新版本中， 可以从中调用 Invocable.proceed() MockUp 实现。 请参阅访问调用上下文。

public class MyMockClass extends MockUp<RealClass> {

    private static boolean fakeIt = false;

    @Mock
    public SomeClass foo(Invocation inv) {
        if (fakeIt) {
            return new SomeClass("fakevalue");
        } else {
            return inv.proceed();
        }
    }
}

Answer 2

我认为你可以使用 @Mock 注释来做到这一点。 从文档中，您的模拟类上的 @Mock(reentrant=true) 应该可以做到这一点。

请参阅 http://jmockit.googlecode.com/svn/trunk/ www/javadoc/mockit/Mock.html

有关示例，请参见此处 http://jmockit.googlecode .com/svn/trunk/www/tutorial/StateBasedTesting.html#reentrant

不过我还没有测试过这个..

Answer 3

您还可以对要测试的对象进行子类化并重写应返回特殊值的方法，而不是抛出模拟对象。

例如：

RealClass toTest = new RealClass(){
      public String foo(){
          return "special value";
      }
}

//use toTest in test

通过在测试中保留此定义，其他人也可以清楚哪些方法正在被“模拟”。

Answer 4

谁仍然感兴趣， Jmockit docs 中都有对此的支持

无论 以本题为例，可以这样实现：

SomeClass someClass = new SomeClass("fakevalue");

new Expectations(someClass){{
    someClass.foo();
    result = <mock>;
}};