如何在rest模板抛出的异常下为if else条件编写Junit

Question

我是 Junit 的新手。我阅读了 Mockito 的基础知识。我无法弄清楚如何为测试代码覆盖率编写 Junit。 我正在使用休息模板调用来接收已发送请求的响应对象。 我在这里抛出异常（HttpStatusCodeException）下执行一些逻辑。 以下是我的代码：

public void processData(Request data, HttpHeaders headers, String priority) throws JsonProcessingException {

    try {

        HttpEntity<Request> reqEntity = new HttpEntity<>(data, headers);
        ResponseEntity<Response> helloResponse = restTemplate.postForEntity(someendpoint, reqEntity, Response.class);
        if (priority.equals("low")) {
            func1();
        } else {
            func2();
        }

    } catch (HttpStatusCodeException ex) {

        if (ex.getStatusCode().equals(HttpStatus.SERVICE_UNAVAILABLE) && priority.equals("low")) {

            func3();

        } else if (ex.getStatusCode().equals(HttpStatus.SERVICE_UNAVAILABLE) && priority.equals("high")) {

            func4();

        }

        else if (ex.getStatusCode().equals(HttpStatus.FORBIDDEN) && priority.equals("low")) {

            func5();

        }

        else if (ex.getStatusCode().equals(HttpStatus.FORBIDDEN) && priority.equals("high")) {

            func6();

        }

        else {

            ObjectMapper mapper = new ObjectMapper();
            Response response = mapper.readValue(ex.getResponseBodyAsString(), Response.class);
            if (priority.equals("low")) {
                func7();
            } else {
                func8();
            }
            
        }

    }
}

}

得到一些帮助真的很有帮助。

Answer 1

如下操作。

@Test
public void test1(){

Mockito.when(restTemplate.postForEntity(ArgumentMatchers.any(), ArgumentMatchers.any, ArgumentMatchers.<Class<Response>>any())).thenReturn(new ResponseEntity<>(new Response(), HttpStatus.ok));
  //do your call and assertion
  //processData( data,  headers,  priority)
}

测试异常覆盖

  @Test
    public void testException1(){
    
    doThrow(HttpClientErrorException(HttpStatus.SERVICE_UNAVAILABLE).when(restTemplate).postForEntity(ArgumentMatchers.any(), ArgumentMatchers.any, ArgumentMatchers.<Class<Response>>any());
    //do your call and assertion
    //processData( data,  headers,  "low")
    }
    
    
    
    @Test
    public void testException2(){
         doThrow(HttpClientErrorException(HttpStatus.FORBIDDEN).when(restTemplate).postForEntity(ArgumentMatchers.any(), ArgumentMatchers.any, ArgumentMatchers.<Class<Response>>any());
    //do your call and assertion
    //processData( data,  headers,  "low")
    }

  

   

 @Test
 public void testException3(){ 
         doThrow(HttpClientErrorException(HttpStatus.SERVICE_UNAVAILABLE).when(restTemplate).postForEntity(ArgumentMatchers.any(), ArgumentMatchers.any, ArgumentMatchers.<Class<Response>>any());
            //do your call and assertion
            //processData( data,  headers,  "high")
            }
            
            
            
 @Test
 public void testException4(){
            
            doThrow(HttpClientErrorException(HttpStatus.FORBIDDEN).when(restTemplate).postForEntity(ArgumentMatchers.any(), ArgumentMatchers.any, ArgumentMatchers.<Class<Response>>any());
            //do your call and assertion
            //processData( data,  headers,  "high")
            }