如何模拟 JPA 存储库以使用 Junit 进行测试？

Question

修改后无法保存对象。

我正在将对象的属性设置为新值，当我要保存时，我收到了 NullPointerException。

/* The method to be tested */
public Rem afterInitRem(Rem rem) {
        
    /** Initialize regles with status MISSING **/
    List<Regle> regles = rem.getRegles();
    Regle regle = new Regle();
    regle.setCode("REGLE1");
    regle.setStatus(RegleStatus.MISSING);
    regles.add(regle);
    return remRepository.save(rem);
}
        
/*The test*/
@Mock
private RemRepository remRepository;

@BeforeEach
void beforeEachTest() {
    rem = new Rem();
}

@AfterEach
void reInitVar() {
    beforeEachTest();
}

@Test
public void afterInitRemTest() {
    target.afterInitrem(rem);
    when(remRepository.save(any(Rem.class))).thenReturn(rem);
    ArgumentCaptor<Rem> argument = ArgumentCaptor.forClass(Rem.class);
    verify(regleRepository).save(argument.capture());

    assertEquals("REGLE1", argument.getValue().getRegles().get(0).getCode());
    assertEquals(RegleStatus.MISSING, argument.getValue().getRegles().get(0).getStatus());
    assertEquals(1, argument.getValue().getRegles().size());
}

我在调试，看到设置了属性，问题出在return remRepository.save(rem);

Answer 1

当您使用 Junit5 时，请确保您的代码中有任何一个来初始化模拟：

@ExtendWith(MockitoExtension.class) on the test class

或

@BeforeEach
void beforeEachTest() {
    MockitoAnnotations.initMocks(this);
    rem = new Rem();
}

此外，您需要确保已手动或使用 @InjectMocks 注释将模拟注入 SUT。

最后，确保在调用实际 SUT 方法之前完成所有模拟设置。