首页 > 解决方案 > 如何使用 RestTemplate 和 JUnit 测试 restclient?

问题描述

我是 JUNIT 的新手,并且RestTemplate用来调用我的服务,我得到了 200 个响应。但是,我无法使用 JUnit 测试该类。尝试了不同的方法并获得 400 和 404。我想发布请求正文(json)并测试状态。如果有任何问题,请告诉我。

/**
* Rest client implementation
**/  
public class CreateEmailDelegate implements CDM {
         
    @Autowired
    private RestTemplate restTemplate;
    private  String url = "http://example.com/communications/emails";
        
    public ResponseEntity<CDResponse> createEmail(CDMEmailRequest cDRequest) throws UnavailableServiceException, InvalidInputException {
        HttpHeaders httpHeaders = new HttpHeaders();
        httpHeaders.set("SR_API_Key", SR_API_KEY);
        httpHeaders.set("consumerIdentification", CONSUMER_IDENTIFICATION);
        httpHeaders.setContentType(MediaType.APPLICATION_JSON);         
        HttpEntity< CDMEmailRequest > cDRequestEntity = new HttpEntity<>( cDRequest, httpHeaders);  
        ResponseEntity< CDResponse > cDResponse = null;
                
        try {
            cDResponse = restTemplate.postForEntity(url, cDRequestEntity, CDResponse.class);
        } catch (Exception e) {
            LOGGER.error(e.getMessage());
            throw  e;       
        }
                
        return cDResponse;
    }
}  

我的测试类返回 404 状态而不是 200

    @RunWith(SpringJUnit4ClassRunner.class)
    public class CreateEmailCommunicationDelegateTest {
    
        @Before
        public void setup() {
            httpHeaders = new HttpHeaders();
            httpHeaders.set("SR_API_Key", SR_API_KEY);
            httpHeaders.set("consumerIdentification", CONSUMER_IDENTIFICATION);
            httpHeaders.setContentType(MediaType.APPLICATION_JSON);
            DefaultMockMvcBuilder builder = MockMvcBuilders.webAppContextSetup(this.wac);
            this.mockMvc = builder.build();
        }
    
    
        public void testResponse() throws Exception, HttpClientErrorException, JsonProcessingException {
            String url = "http://example.com/CommunicationDeliveryManagement-Service-1.0.0/communications/emails";
    
            CDMEmailRequest anObject = new CDMEmailRequest();
            ResultMatcher ok = MockMvcResultMatchers.status().isOk();
            ObjectMapper mapper = new ObjectMapper();
            mapper.configure(SerializationFeature.WRAP_ROOT_VALUE, false);
            ObjectWriter ow = mapper.writer().withDefaultPrettyPrinter();
            String requestJson = ow.writeValueAsString(anObject);
    
            System.out.println(requestJson);
            MockHttpServletRequestBuilder builder = MockMvcRequestBuilders.post(url).contentType(MediaType.APPLICATION_JSON_UTF8).content(requestJson);
            this.mockMvc.perform(builder).andExpect(ok).andDo(MockMvcResultHandlers.print());
        }
    } 

 

我的测试类使用TestRestTemplate而不是MockMvc返回 400

@RunWith(SpringJUnit4ClassRunner.class)
public class CreateEmailCommunicationDelegateTest {

    @Before
    public void setup() {
        httpHeaders = new HttpHeaders();
        // rest headers as above
    }

    @Test
    public void testResponse() throws Exception, HttpClientErrorException, JsonProcessingException {

        String url = "http://example.com/CommunicationDeliveryManagement-Service-1.0.0/communications/emails";

        String username = "";
        String password = "";
        HttpEntity<CDMEmailRequest>
                cDEntity = new HttpEntity<>(httpHeaders);

        restTemplate = new TestRestTemplate(username, password);

        responseEntity =
                restTemplate.exchange(url, HttpMethod.POST, cDEntity,
                        CDResponse.class);

        assertNotNull(responseEntity);
        assertEquals(HttpStatus.OK,
                responseEntity.getStatusCode());
    }
} 

标签: javarestspring-bootjunitresttemplate

解决方案


我认为您正在尝试实现集成测试而不是单元测试,这是有很大区别的。MockMvc应该用于实现单元测试和TestRestTemplate集成测试。您既不能使用它来测试客户端实现。

请参阅Spring Boot 中的单元和集成测试

如果您正在使用 Spring Boot,您可以使用另一种方法来实现您的目标,请参阅这个问题Spring boot testing of a rest client using @RestClientTest


推荐阅读