使用 PowerMock 模拟静态和动态方法

Question

假设我们有

public class Foo {
   public static Foo getInstance() {...}

   public Bar bar(Baz baz) {...}
}

我想做的就是在我的单元测试中模拟它。我需要模拟 Foo 类的静态和动态方法。模拟 getInstance() 就像简单的

import static org.powermock.api.easymock.PowerMock.replace;
import static org.powermock.api.easymock.PowerMock.method;

public class MyTest {

   @Test
   public void myTest() {
      replace(method(Foo.class, "getInstance"))
         .with(method(MyTest.class, "getMockInstance"));
   }

   public static Foo getMockInstance() {
      Foo foo = EasyMock.createMock(Foo.class);
      EasyMock.replay(foo);
      return foo;
   }
}

问题是，如何模拟 bar 方法？

之前使用 replace(method(...)).with(method(...)) 的技巧不起作用，因为它不是为动态方法设计的。

尝试在已经模拟的类之上进行模拟也不起作用：

...
@Test
public void myTest() {
      replace(method(Foo.class, "getInstance"))
         .with(method(MyTest.class, "getMockInstance"));

      Foo foo = Foo.getInstance();  // works well
      Baz baz1 = new Baz();
      Baz baz2 = new Baz();
      EasyMock.expect(foo.bar(baz1)).andReturn(baz2);  // exception thrown
      EasyMock.replay(foo);
}
...

上面的代码会抛出AssertionError：意外的方法调用栏。

那么我该如何做到这两点呢？我不想将 .bar(...) 的模拟放入 getMockInstance 中，因为在现实世界中，我需要一些静态<代码>中无法获得的数据。代码>getMockInstance方法。

Answer 1

我认为问题在于您在 foo 模拟上调用重播两次，一次在模拟静态方法 getMockInstance() 中，一次在您告诉模拟 foo< /code> 期待 foo.bar(bar1) 调用。尝试将 getMockInstance() 更改为

   public static Foo getMockInstance() {
      Foo foo = EasyMock.createMock(Foo.class);
      return foo;
   }

，然后告诉 EasyMock 在等待 bar 方法调用后重播 foo 。所以 MyTest.java 看起来像这样：

@Test
public void myTest() {
      replace(method(Foo.class, "getInstance"))
         .with(method(MyTest.class, "getMockInstance"));

      Foo foo = Foo.getInstance();  // works well
      Baz baz1 = new Baz();
      Baz baz2 = new Baz();
      EasyMock.expect(foo.bar(baz1)).andReturn(baz2);  // exception thrown
      EasyMock.replay(foo);
}

  public static Foo getMockInstance() {
      Foo foo = EasyMock.createMock(Foo.class);
      return foo;
   }