java - 模拟多次调用的静态方法

Question

我有一个静态方法，它用在多个地方，主要是在静态初始化 block 中。它以一个 Class 对象作为参数，并返回该类的实例。我只想在将特定类对象用作参数时模拟此静态方法。但是当从其他地方使用不同的 Class 对象调用该方法时，它返回 null。
如果参数不是模拟参数，我们如何让静态方法执行实际实现？

class ABC{
    void someMethod(){
        Node impl = ServiceFactory.getImpl(Node.class); //need to mock this call
        impl.xyz();
    }
}

class SomeOtherClass{
    static Line impl = ServiceFactory.getImpl(Line.class); //the mock code below returns null here
}


class TestABC{
    @Mocked ServiceFactory fact;
    @Test
    public void testSomeMethod(){
         new NonStrictExpectations(){
              ServiceFactory.getImpl(Node.class);
              returns(new NodeImpl());
         }
    }
}

Answer 1

你想要的是一种“部分模拟”的形式，特别是 dynamic partial mocking在 JMockit API 中:

@Test
public void testSomeMethod() {
    new NonStrictExpectations(ServiceFactory.class) {{
        ServiceFactory.getImpl(Node.class); result = new NodeImpl();
    }};

    // Call tested code...
}

只有符合记录期望的调用才会被模拟。当调用动态模拟类时，其他人将执行真正的实现。