RestTemplate返回null时模拟

Question

我试图测试一个简单POST Rest调用，但存在NullPointerException。

我的RestController：

@RestController
@RequestMapping("/v1/mail")
@Slf4j
public class EmailForwardController {

    @Autowired
    EmailForwardConfig emailConfig;

    @Autowired
    RestTemplate restTemplate;

    @PostMapping("/forward")
    public ResponseEntity<String> setupEmailForward(@RequestParam String fromEmail, @RequestParam String toEmail, @RequestParam String referenceId) {
        log.info("Email Forward: from=" + fromEmail + " to=" + toEmail + " reference=" + referenceId);
        String uri = emailConfig.getUrl() + "forward";
        JSONObject jsonObject = new JSONObject();
        jsonObject.put("from", fromEmail);
        jsonObject.put("to", toEmail);
        jsonObject.put("clientId", referenceId);
        HttpEntity<String> httpEntity = new HttpEntity<>(jsonObject.toJSONString());
        ResponseEntity<String> response = restTemplate.postForEntity(uri, httpEntity, String.class);
        System.out.println(response); // -> this is null at UNIT Test
        if (response.getBody() == null) {
            log.error("No response for call!");
            return new ResponseEntity<>("Problem!", HttpStatus.INTERNAL_SERVER_ERROR);
        }
        return new ResponseEntity<>(response.getBody(), response.getStatusCode());

    }

}

我的单元测试：

@RunWith(MockitoJUnitRunner.class)
@SpringBootTest(classes = EmailForward.class, webEnvironment = SpringBootTest.WebEnvironment.DEFINED_PORT)
public class ServiceTest {

    @InjectMocks
    EmailForwardController emailForwardController;

    @Mock
    RestTemplate restTemplate;

    @Mock
    EmailForwardConfig emailConfig;

    @Before
    public void setup() {

    }

    @Test
    public void test_forward_ok() {
        assertNotNull(restTemplate);
        assertNotNull(emailForwardController);
//        System.out.println(Mockito.verify(restTemplate).postForEntity(any(), any(), any()));
        Mockito.when(restTemplate.postForEntity(any(), any(), any())).thenReturn(new ResponseEntity<>("done", HttpStatus.OK));
        ResponseEntity<String> responseEntity = emailForwardController.setupEmailForward("test", "test", "test");
        assertNotNull(responseEntity);
        assertNotNull(responseEntity.getStatusCode());
        assertNotNull(responseEntity.getBody());
        assertEquals(responseEntity.getBody(), "Done");
        assertEquals(responseEntity.getStatusCode(), HttpStatus.OK);
    }


}

由于以下错误，此测试失败：NullPointerException，确切地说，response ResponseEntity对象为Null。

verify输出将执行以下操作：

Wanted but not invoked:
restTemplate.postForEntity(
    <any>,
    <any>,
    <any>
);
-> at xx.xxx.ms.oauth2.ServiceTest.test_forward_ok(ServiceTest.java:42)
Actually, there were zero interactions with this mock.

知道为什么响应是空的吗？我为测试使用了正确的注释，因此应该注入模拟。也许我在这里遗漏了一些重要的东西，因为如果我删除@mock EmailForwardConfig emailconfig；这个对象在RESTController中是空的。我觉得我忘记了什么，但我想不出来。

Answer 1

您对RestTemplate.PostForEntity的嘲弄是不正确的。

将您得嘲讽更改为：

Mockito.when(restTemplate.postForEntity(any(), any(), any())).thenReturn(new ResponseEntity<>("done", HttpStatus.OK));

致：

Mockito.when(restTemplate.postForEntity(anyString(),any(),
                any(Class.class))).thenReturn(new ResponseEntity<>("done", HttpStatus.OK));