可以使用注释将原语变量注入模拟类吗？

Question

例如，我有handler：

@Component
public class MyHandler {

  @AutoWired
  private MyDependency myDependency;

  @Value("${some.count}")
  private int someCount;

  public int someMethod(){
    if (someCount > 2) {
    ...
  }
}

为了测试它，我编写了以下测试：

@RunWith(MockitoJUnitRunner.class}
class MyHandlerTest {

  @InjectMocks
  MyHandler myHandler;

  @Mock
  MyDependency myDependency;

  @Test
  public void testSomeMethod(){
    ReflectionTestUtils.setField(myHandler, "someCount", 4);
    myHandler.someMethod();
  }
}

我可以使用reflectiontestutils模拟变量someCount。我可以用Mockito注释来模拟它吗？

Answer 1

没有一种内置的方法可以做到这一点，而且要注意@InjectMocks也有它的缺点：Mockito的@InjectMocks更多的是一种礼貌，而不是一个完全安全的特性，如果测试中的系统添加了任何字段，它就会无声无息地失败。

相反，可以考虑为测试创建一个构造函数或工厂方法：尽管测试代码应该存在于测试中，而不是生产类中，但测试是类的消费者，您可以为它们显式地设计一个构造函数。

@Component
public class MyHandler {

  @AutoWired
  private MyDependency myDependency;

  @Value("${some.count}")
  private int someCount;

  /** Visible for testing. */
  MyHandler(MyDependency myDependency, int someCount) {
    this.myDependency = myDependency;
    this.someCount = someCount;
  }

  // ....
}