java - 使用 Mockito 和 PowerMockito 模拟类对象

Question

是否可以使用 Mockito 和/或 PowerMockito 模拟类对象？

就像是：

Class<Runnable> mockRunnableClass = mock(Class<Runnable>.class);

Answer 1

如果您不能重构代码，为什么不使用代理没有很多选项，正如@jherics 提到的那样，java 系统类由引导类加载器加载，而 powermock 无法重新定义它们的字节码。

然而，Powermock 现在带有一个代理，它将允许系统类模拟。在这里查看完整的解释。

主要思想是修改您的 java 命令并添加：

-javaagent: path/to/powermock-module-javaagent-1.4.12.jar

该代理所做的基本工作是取消类的最终确定，以允许将来在特定测试中进行模拟，这就是为什么您需要使用特定类型与代理进行通信的原因，例如使用 JUnit ：

@Rule PowerMockRule rule = new PowerMockRule(); // found in the junit4 rule agent jar

也支持 TestNG。只需查看wiki 页面以获取更多信息。

希望有帮助。

Answer 2

模拟类的替代方法可能是使用工厂。我知道您担心重构，但这可以在不更改类的公共 API 的情况下完成。您没有提供太多代码来理解您尝试测试的类，但这里有一个在不更改 API 的情况下进行重构的示例。这是一个微不足道的课程，但它可能会给你一个想法。

public class Instantiator {

  public Runnable getNewInstance(Class<Runnable> runnableClass) throws Exception {
    return runnableClass.newInstance();
  }
}

当然，测试这个微不足道的类最简单的方法是使用真正的 Runnable 类，但是如果你试图模拟这个类，你会遇到你遇到的问题。所以，你可以重构它：

public class PassThruFactory {
  public Object newInstance(Class<?> clazz) throws Exception {
    return clazz.newInstance();
  }
}

public class Instantiator {
  private PassThruFactory factory = new PassThruFactory();

  public Runnable getNewInstance(Class<Runnable> runnableClass) throws Exception {
    return (Runnable)factory.newInstance(runnableClass);
  }
}

现在 Instantiator 使用相同的公共 API 完成了它之前所做的（非常简单的）事情，并且不需要该类的任何客户端自己进行任何特殊的注入。但是，如果您想模拟工厂类并注入它，那很容易做到。

Answer 3

首先，如评论中所述，您需要执行以下操作：

Class<Runnable> mockRunnableaClass = (Class<Runnable>)mock(Class.class);

但由于 PowerMock 的限制，这不会以通常的方式工作。您不能简单地从 java.lang、java.net、java.io 或其他系统类中模拟类，因为它们是由 Java 的引导类加载器加载的，并且不能由 PowerMock 的类加载器操作字节码。（请参阅PowerMock 常见问题解答#4。）从 PowerMock 1.2.5 开始，您可以解决此问题。如果您要测试的课程是这样的：

public class ClassToTest {
  private Class<Runnable> runnableClass;

  public void setRunnableClass(Class<Runnable> runnableClass) {
    this.runnableClass = runnableClass;
  }

  public Runnable foo() {
    return runnableClass.newInstance();
  }
}

然后你会这样做：

@RunWith(PowerMockRunner.class)
@PrepareForTest({ ClassToTest.class }) // Prepare the calling class for test
public class SystemClassUserTest {

  @Test
  public void testFoo() throws Exception {
    Class<Runnable> mockClass = (Class<Runnable>) mock(Class.class);
    Runnable mockRunnable = mock(Runnable.class);

    ClassToTest objectUT = new ClassToTest();
    objectUT.setRunnableClass(mockClass);

    when(mockClass.newInstance()).thenReturn(mockRunnable);
    assertThat(objectUT.foo(), is(sameInstance(mockRunnable);
  }
}

Answer 4

这个怎么样。创建一个在类 PCService 中有一个对象（MS）的 get 方法，然后模拟它。

public class PCService implements PCServiceIf {
    public MSIf getMS() {
        return ms;
    }

    private MSIf ms = new MS();
    public boolean isMovieAccessibleToMyLevel(String myLevel, String movieId)  {
        return  getMS().getPCL(movieId);
    }
}


@Test
public void testIsMovieAccessibleToMyLevelMock()  {
    msMock = mock(MS.class);
    spy  = spy(new PCService());

    doReturn(msMock).when(spy).getMS();
    when(msMock.getPCL(movieId)).thenReturn(value);
    when(spy.getMS().getPCL(movieId)).thenReturn(value);
    assertTrue(spy.isMovieAccessibleToMyLevel("PG", movieId) == true);
}