Mockito and Javax annotation @PostConstruct

Question

我正在使用JUnit Jupiter为一个具有javax.annotation.PostConstruct注释的方法的类编写一个Java单元测试。

我想测试类中的一个方法，但遇到了一个问题，即创建对象后总是调用@PostConstruct方法(这并不奇怪(。我创建的用于@PostConstruct方法的Mockito mock对象没有提供我需要的结果。

以下是创建对象时调用的方法：

@Service
@Slf4j
public class MyService {
@Inject
private MyRepository myRepository;
private Integer myId;
@PostConstruct
public void postInit() {
MyObject myObj = myRepository.findByName("FOO");
myId = myObj.getId();
}
public void myMethod() {
... code to be tested here...
}
}

我的JUnit测试类包含以下内容：

@ExtendWith(SpringExtension.class)
@DirtiesContext(classMode = DirtiesContext.ClassMode.AFTER_EACH_TEST_METHOD)
public class MyServiceTest {
private static final Integer ID = 0;
@Test
public void myTest() {
// Given.
MyObject myObj = new MyObject();
myObj.setId(ID);
MyRepository myRepository = Mockito.mock(MyRepository.class);
Mockito.when(myRepository.findByName(anyString())).thenReturn(myObj);
// Test runs here.
myService.myMethod()
}
}

当我运行单元测试时，我可以看到mock已经通过我的IDE实例化了。但是，当调用findByName("FOO")方法时，它总是返回NULL对象，即使它应该返回MyObject的正确实例。

因此，单元测试失败并出现NullPointerException，因为myObj为null。

我在这里做的事情有什么明显不正确的地方吗>？

提前感谢您的帮助。

Answer 1

我稍微修改了您的测试，通过Mockito创建了MyService，并使myMethod返回id:

@ExtendWith(SpringExtension.class)
class MyServiceTest {
private static final Integer ID = 42;
@InjectMocks
MyService myService;
@Test
public void myTest() {
// Given.
MyObject myObj = new MyObject();
myObj.setId(ID);
MyRepository myRepository = Mockito.mock(MyRepository.class);
Mockito.when(myRepository.findByName(anyString())).thenReturn(myObj);
// Test runs here.
assertThat(myService.myMethod()).isEqualTo(ID);
}
}

当我运行这个时，我得到：

java.lang.NullPointerException: Cannot invoke "java.lang.Integer.intValue()" because "this.myId" is null
at MyService.myMethod(MyService.java:24)
at MyServiceTest.myTest(MyServiceTest.java:28)
...

很明显，我们对findByName的嘲讽并没有奏效。

这是因为myRepository在我们的测试中只是一个局部变量——没有什么可以将它注入到MyService中。

解决这个问题的一种方法是不使用spring测试基础设施，只使用普通的Mockito。我们使用标准的@Mock/@InjectMocks注释来创建我们的Mock和我们正在测试的对象。这里的缺点是我们必须记住自己调用postInit：

@ExtendWith(MockitoExtension.class)
class MyServiceTest {
private static final Integer ID = 42;
@Mock
MyRepository myRepository;
@InjectMocks
MyService myService;
@Test
public void myTest() {
// Given.
MyObject myObj = new MyObject();
myObj.setId(ID);
Mockito.when(myRepository.findByName(anyString())).thenReturn(myObj);
myService.postInit();
// Test runs here.
assertThat(myService.myMethod()).isEqualTo(ID);
}
}

要使用Spring调用postInit，可以这样做：(我不确定是否有更惯用的方法来创建mock MyRepository实例(

@ExtendWith(SpringExtension.class)
@SpringBootTest(classes = {MyService.class})
@ContextConfiguration(classes = {MyServiceTest.Configuration.class})
class MyServiceTest {
@TestConfiguration
static class Configuration {
@Bean
public MyRepository myRepository() {
MyRepository myRepository = mock(MyRepository.class);
MyObject myObj = new MyObject();
myObj.setId(ID);
Mockito.when(myRepository.findByName(anyString())).thenReturn(myObj);
return myRepository;
}
}
private static final Integer ID = 42;

@Autowired
MyService myService;

@Test
public void myTest() {
assertThat(myService.myMethod()).isEqualTo(ID);
}
}

避免使用@SpringBootTest的一种方法是在我们的测试配置中创建MyService实例(我也将其转换为构造函数注入(：

@ExtendWith(SpringExtension.class)
@ContextConfiguration(classes = {MyServiceTest.Configuration.class})
class MyServiceTest {
@TestConfiguration
static class Configuration {
@Bean
public MyRepository myRepository() {
MyRepository myRepository = mock(MyRepository.class);
MyObject myObj = new MyObject();
myObj.setId(ID);
Mockito.when(myRepository.findByName(anyString())).thenReturn(myObj);
return myRepository;
}
@Bean
public MyService myService(MyRepository myRepository) {
return new MyService(myRepository);
}
}
private static final Integer ID = 42;
@Autowired
MyService myService;
@Test
public void myTest() {
assertThat(myService.myMethod()).isEqualTo(ID);
}
}