春靴

Question

我想在我的Junit5单元测试中将值注入到带有@Value注释的私有字段。

@Value("$(invoice.maxRetry)")
private maxRetry;

我引用了这个，并使用了ReflectionTestUtils。setField通过注入值解决了我的问题，但在验证方法被调用的次数时失败。

MyClass（我的类别）：

    public class MessageProcessor {
         
    @Value("$(invoice.maxRetry)")
        private maxRetry;
    
    protected void handleMessage() {
            if(retry > maxRetry) {
                kafkaTemplate.sendMessage(msg);
       }
}

测试类：

@ExtendWith(MockitoExtension.class)
public class MessageProcessorTest {

@Mock
private kafkaTemplate kafkaTemplate;

@Mock
private MessageProcessor messageProcessor

@Test
test() {
     ReflectionTestUtils.setField(messageProcessor, "maxRetry", "5");
     doNothing().when(kafkaTemplate).sendMessage(any());
     messageProcessor.handleMessage();
     verify(kafkaTemplate).sendMessage(any());
  }
}

运行上述测试时出错

org.mockito.exceptions.verification.TooManyActualInvocations: 
kafkaTemplate.sendMessage(<any>);
Wanted 1 time:
But was 2

我想kafkaTemplate.sendMessage（）；被调用一次，但被调用两次后添加反射TestUtils。

需要关于如何修复此问题的建议。

Answer 1

通过稍微重构类并支持构造注入，可以避免使用ReflectionTestUtils：

public class MessageProcessor {
     
    private String maxRetry;
    private KafkaTemplate template;

    // ... any further fields

    public class MessageProcessor(@Value("$(invoice.maxRetry)") String maxRetry, KafkaTemplate kafkaTemplate) {
       this.maxRetry = maxRetry;
       this.kafkaTemplate = kafkaTemplate;
    }

 }

然后，在测试中，您可以通过手动创建被测试类的实例（MessageProcessor）来控制maxRetry的值。

@ExtendWith(MockitoExtension.class)
public MessageProcessorTest {

  @Mock 
  private KafkaTemplate kafkaTemplate;

  private MessageProcessor messageProcessor;

  @BeforeEach
  void setUp() {
    this.messageProcessor = new MessageProcessor("42", kafkaTemplate);

  }

}

... 然后只依赖JUnit和Mocktio，这将帮助您添加验证。