Spring Boot 测试 : exception in REST controller

Question

我有一个 Spring Boot 应用程序，想通过集成测试来覆盖我的 REST Controller 。
这是我的 Controller :

@RestController
@RequestMapping("/tools/port-scan")
public class PortScanController {
    private final PortScanService service;

    public PortScanController(final PortScanService portScanService) {
        service = portScanService;
    }

    @GetMapping("")
    public final PortScanInfo getInfo(
            @RequestParam("address") final String address,
            @RequestParam(name = "port") final int port)
            throws InetAddressException, IOException {
        return service.scanPort(address, port);
    }
}

在其中一个测试用例中，我想测试该端点在某些情况下会引发异常。这是我的测试课:

@RunWith(SpringRunner.class)
@WebMvcTest(PortScanController.class)
public class PortScanControllerIT {
    @Autowired
    private MockMvc mvc;

    private static final String PORT_SCAN_URL = "/tools/port-scan";

    @Test
    public void testLocalAddress() throws Exception {
        mvc.perform(get(PORT_SCAN_URL).param("address", "192.168.1.100").param("port", "53")).andExpect(status().isInternalServerError());
    }
}

最好的方法是什么？当前实现不处理从 PortScanController.getInfo() 抛出的 InetAddressException ，当我开始测试时，我收到并出错:

org.springframework.web.util.NestedServletException: Request processing failed; nested exception is com.handytools.webapi.exceptions.InetAddressException: Site local IP is not supported

不可能在 @Test 注释中指定预期的异常，因为原始的 InetAddressException 是用 NestedServletException 包装的。

Answer 1

Spring Boot 测试包自带 AssertJ它有非常方便的方法来验证抛出的异常。

要验证原因:

@Test
public void shouldThrowException() {
    assertThatThrownBy(() -> methodThrowingException()).hasCause(InetAddressException .class);
}

还有一些你可能感兴趣的方法。我建议看看 docs .