22

如果没有在您正在测试的类中创建/初始化模拟对象，它不是静态的（单例模式），或者您没有某种类型的测试构造函数可以挂钩，如何在运行时获取模拟对象？

在我正在为其编写一些单元测试的课程中，我遇到了一个我还没有遇到/解决的场景。我有一个 JMS 资源（QueueConnectionFactory供参考，但没关系），这是我正在测试的类的私有变量。由于它具有javax.annotation.Resource注释，因此在运行时假定它是可用的。在测试期间，它不是，这就需要模拟这个对象。

它不是静态类，也不是以静态方式使用的，如果是的话，我可以使用我遇到的各种静态模拟方法轻松模拟。由于资源永远不会在本地创建（在构造函数中甚至在测试构造函数中），我无法传入 Mock 对象，因此在测试运行时，使用模拟而不是实际对象。我如何模拟这个资源，以便在测试执行时，它将用于代替@Resource我正在测试的类中的私有对象？

作为参考，代码正在调用createConnection()由于QueueConnectionFactory工厂尚未初始化/模拟而引发空指针异常的代码。

@Stateless
public class Example{
  @Resource(name = "jms/exampleQCF")
  private QueueConnectionFactory queueFactory;

  ...

  public void testMe(){
    Connection connection = queueFactory.createConnection();
    ...
  }
}
4

2 回答 2

41

在进行了更多的搜索并查看了 Mockito/Powermock 必须提供的所有选项之后，我找到了解决方案（如果其他人遇到同样的问题，我将分享该解决方案）。

当您有从未初始化的私有成员变量（并且只是假设在其他地方创建）时，您可以使用@InjectMocks注释将您想要的模拟“注入”到您正在测试的类中。

  1. 在你的测试类中为你正在测试的类添加一个变量，并给它注解@InjectMocks（org.Mockito.InjectMocks）。
  2. 使用@Mock注释来设置要注入的模拟。使用@Mock (name = "privateVariableNameHere")name 属性将 Mock 对象映射到您正在测试的类中的私有变量。
  3. 在设置函数中或在调用类之前，初始化模拟。我发现最简单的方法是使用带有@Before注释的“设置”方法。然后在那里调用MockitoAnnotations.initMocks(this);以使用@Mock注释快速初始化任何内容。
  4. 在您的测试方法中定义您的 Mock 功能（在调用您正在测试的方法之前）。
  5. 使用该@InjectMock对象，调用您正在测试的方法......模拟应该被挂钩并按照前面步骤中的定义工作。

因此，对于我上面使用的示例类，测试/模拟的代码将Connection作为模拟返回，你可以用它做任何事情。根据上面我的问题中的示例，代码如下所示：

@RunWith(PowerMockRunner.class)
@PrepareForTest({/* Static Classes I am Mocking */})
public class ExampleTest {
  @Mock (name = "queueFactory") //same name as private var.
  QueueConnectionFactory queueFactoryMock;
  @Mock
  Connection connectionMock; //the object we want returned
  @InjectMocks
  Example exampleTester; //the class to test

  @Before
  public void setup(){
    MockitoAnnotations.initMocks(this); // initialize all the @Mock objects
    // Setup other Static Mocks
  }

  @Test
  public void testTestMe(){
    //Mock your objects like other "normally" mocked objects
    PowerMockito.when(queueFactoryMock.createConnection()).thenReturn(connectionMock);
    //...Mock ConnectionMock functionality...
    exampleTester.testMe();
  }
}
于 2013-10-22T19:49:39.650 回答
3

这里有几种方法：

  1. ReflectionTestUtilsSpring 测试框架：ReflectionTestUtils.setField(objectToTest, "privateFieldName", mockObjectToInject);. 有了这个，你就不会引入另一个依赖项。
  2. org.mockito.internal.util.reflection.FieldSetter.
  3. PowerMock.Whitebox.setInternalState()模拟一个私人领域。

如果您需要模拟内部局部变量的创建，请使用PowerMockito.whenNew(Foo.class).withNoArguments().thenReturn(foo);. 非常非常好用。找不到其他方法来做同样的事情。

只有 Mockito 不能模拟局部变量的创建，因为when(any(Foo.class)它不起作用；将返回空值。它编译但不起作用。

参考： Mockito：模拟私有字段初始化

于 2019-05-16T08:43:38.040 回答