如何使用power mock对spring boot rest控制器和异常处理程序进行单元测试

Question

我有一个简单的Spring boot应用程序，其中包含Employee controller，如果经过的年份大于2014，则返回员工姓名，如果不小于2014，则抛出自定义异常并在异常处理程序中处理
我想使用powermock对异常流进行单元测试，但我不确定如何进行。我已经浏览了一些链接，但无法理解
目前我正在获取java.lang.IllegalArgumentException：需要WebApplicationContext。

EmployeeController.java

@RestController
public class EmployeeController{

    @GetMapping(value = "/employee/{joiningYear}",produces = MediaType.APPLICATION_JSON_VALUE)
    public List<String> getEmployeeById(@PathVariable int joiningYear) throws YearViolationException {

        if(joiningYear < 2014){
            throw new YearViolationException("year should not be less than 2014");
        }else{

            // send all employee's names joined in that year 
        }
        return null;
    }
}   

异常处理程序

@RestControllerAdvice
public class GlobalControllerExceptionHandler {


    @ExceptionHandler(value = { YearViolationException.class })
    @ResponseStatus(HttpStatus.BAD_REQUEST)
    public ApiErrorResponse yearConstraintViolationExceptio(YearViolationException ex) {

        return new ApiErrorResponse(400, 5001, ex.getMessage());
    }
}

自定义异常

public class YearViolationException extends Exception {

    /**
     * 
     */
    private static final long serialVersionUID = 1L;

    public YearViolationException(String message) {

        super(message);
    }

}

Junit到单元测试异常处理程序

@RunWith(PowerMockRunner.class)
@WebAppConfiguration
@SpringBootTest
public class ExceptionControllerTest {

    @Autowired
    private WebApplicationContext applicationContext;

    private MockMvc mockMVC;

    @Before
    public void setUp() {

        mockMVC = MockMvcBuilders.webAppContextSetup(applicationContext).build();
    }

    @Test
    public void testhandleBanNotNumericException() throws Exception {

        mockMVC.perform(get("/employee/2010").accept(MediaType.APPLICATION_JSON)).andDo(print())
                .andExpect(status().isBadRequest())
        .andExpect(content().contentTypeCompatibleWith(MediaType.APPLICATION_JSON));

    }
}

Answer 1

从您的设置看，您根本不需要使用Mocks。看起来您想加载完整的应用程序上下文并使用mockMVC向您的rest控制器发送请求。这实际上是集成测试！

现在，不幸的是，我们在这里使用了Spring Boot 1.3，所以我不确定＜code＞@RunWith（PowerMockRunner.class）

@RunWith(SpringJUnit4ClassRunner.class)
@SpringApplicationConfiguration(classes = Application.class)
@WebAppConfiguration

另一方面，如果您想进行简单的单元测试，则无需加载应用程序上下文。相反，您可以只使用常规的 Mockito 与@Mock，并与@RunWith（MockitoJUnitRunner.class）结合使用@InjectMocks，并像调用任何其他被测试方法一样直接调用要测试的方法。

希望这有帮助。

Answer 2

正如其他人所说，您根本不需要mockMVC。如果您想测试RESTendpoint，您需要的是TestRestTemplate。与SpringRunner一起跑步。类与WebEnvironment设置一样重要。

@RunWith(SpringRunner.class)
@SpringBootTest(webEnvironment=WebEnvironment.RANDOM_PORT)
public class RestServiceApplicationTests {

    private String baseUrl = "http://localhost:8090";

    private String endpointToThrowException = "/employee/2010";

    @Autowired
    private TestRestTemplate testRestTemplate;

    @Test(expected = YearViolationException.class)
    public void testhandleBanNotNumericException() {
        testRestTemplate.getForObject(baseUrl + endpointToThrowException, String.class);
}