模拟多个静态方法

Question

我有一个类，它使两个不同的静态函数调用一个是一些Util函数，而另一个是局部的类，像这样:

public class A {
    .
    .
    public static myfunc1() {
        obj1 = myfunc(param);
        .
        .
        Obj obj = Util.getObj(param);
        .
        .
    }
    static obj1 myfunc(param) {
        ..
    }
}

我想为这个类编写单元测试，它看起来像这样:

public class Atest {
    .
    .
    public void testMyfunc1() {
        .
        .
        A a = new A();
        A spyA = spy(a);
        PowerMockito.doReturn(mockObj).when(spyA).myfunc(mockParam);
        .
        .
    }
    .
    .
}

但是它给了我unfinhedstubbingexception。

我也试过这样做:

PowerMockito.when(spyA.myfunc(mockParam)).thenReturn(mockObj);

但是它不覆盖函数。

其次，我需要知道如何重写Util.getObj()函数

Answer 1

经过一些研究，我知道PowerMock提供了一个名为easymock的模块，其中有一个名为mockStaticPartial的方法，用于模拟类中的特定静态函数。对于Utils.getObj()方法，也可以做类似的事情。所以，我的测试类现在看起来像这样:

@RunWith(PowerMockRunner.class)
@PrepareForTest({A.class, Utils.class})
public class Atest {
    .
    .
    @Before
    public void setup() {
        .
        .
        mockStaticPartial(A.class, "myfunc");
        mockStatic(Utils.class);
        .
        .
    }
    public void testMyfunc1() {
        .
        .
        PowerMockito.when(A.mufync()).thenReturn(mockObj1);
        PowerMockito.when(Utils.getObj(mockParam)).thenReturn(mockObj);
        .
        .
        //More testing logic goes here
    }
    .
    .
}

Answer 2

间谍是用来模拟类实例的，而静态方法是类级别的。您应该使用PowerMockito.mockStatic或PowerMockito.spy。

然而，最好的方法——重构代码并去掉静态。