使用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: Unexpected method call bar 。

那我该怎么做？ 我不想将.bar(...)的getMockInstance放到getMockInstance因为在现实世界中，我需要一些静态getMockInstance方法中不可用的数据。

Answer 1

我认为问题在于您在foo模拟上调用了两次重播，一次是在模拟的静态方法getMockInstance() ，一次是在告诉foo.bar(bar1) foo期待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;
   }