如何模仿通过Class.newInstance(className)创建的对象？

Question

我正在尝试将单元测试添加到一些遗留代码中，该代码具有传递给它的String类名，并使用Class.newInstance（String className）创建一个实现特定处理程序接口的对象。我可以控制我正在传递的类名，我可以获取指向新处理程序对象的指针（通过getHandler（）调用），并且我想使用Mockito观察对它的调用。

我目前的解决方案是：

1. 创建一个实现接口的新测试类TestHandler。
2. 让该测试类包含一个也实现接口的Mockito mock对象。
3. 手动将所有接口方法传递给模拟对象。
4. 使模拟对象可通过getMock（）方法访问。
5. 通过调用objectUnderTest.getHandler（）. getMock（）来观察对象。

这是可行的，但是感觉有点不优雅，尤其是必须手工编写所有的传递方法。

有没有更好的解决方案？

Answer 1

创建一个公共方法，您将在其中放置逻辑以获取类的 newInstance

ClassA objectClassA=createNewInstance(className);

同样的，还有

public ClassA createInstance(String className){
 return (ClassA) (Class.forName(className)).newInstance();
}

现在假设我们在ClassB中创建了一个classA的实例，然后在B的TestClass中，我们可以简单地模拟这个createInstance方法。

doReturn(mockClassA).when(mockClassB).createInstance(className);

Answer 2

从根本上说，您遇到了与尝试使用new测试新创建的实例相同的问题；Class.newInstance（可能正确地Class.forName（foo）. newInstance（））不会伤害您，但也不会帮助您。

顺便说一句，你的TestHandler听起来像一个通用的委托实现，这听起来非常有用（特别是如果你需要编写一个Handler包装器）。如果是，您可能希望将其提升为与生产代码树中的处理程序相邻。

虽然我认识到您提到了遗留代码，但如果允许您重构以包含测试接缝，这将变得非常容易。（为了便于解释，此处忽略反射异常。）

public ReturnType yourMethodUnderTest(String className) {
  return yourMethodUnderTest(Class.newInstance(className));
}

/** Package private for testing. */
public ReturnType yourMethodUnderTest(Handler handler) {
  return yourMethodUnderTest(Class.newInstance(className));
}

您还可以提取对象创建并在测试中替换它：

/** Instance field, package-private to replace in tests. */
Function<String, Handler> instanceCreator =
    ( x -> (Handler) Class.forName(x).newInstance());

public ReturnType yourMethodUnderTest(String className) {
  Handler handler = instanceCreator.apply(className);
  // ...
}

您甚至可以将其提取到一个方法中，并在测试中替换它：

public ReturnType yourMethodUnderTest(String className) {
  Handler handler = createHandler(className);
  // ...
}

/** Package private for testing. */
Handler createHandler(String className) {
  return Class.forName(className).newInstance();
}

@Test public void yourTest() {
  // Manually replace createHandler. You could also use a Mockito spy here.
  ObjectUnderTest objectUnderTest = new ObjectUnderTest() {
    @Override Handler createHandler(String className) {
      return mock(Handler.class);
    }
  }
  // ...
}

附注:尽管Mockito创建了一个命名的动态类型，但几乎可以肯定的是，您不能侵入它并允许您的代码通过名称来创建它。这是因为对< code>mock的调用在Mockito的内部状态中注册了实例。

// BAD: Unlikely to work
@Test public void yourTest() {
  objectUnderTest.methodUnderTest(
      mock(Handler.class).getClass().getName());
  // ...
}