package cn.iocoder.yudao.framework.web.core.handler;
|
|
import jakarta.servlet.http.HttpServletResponse;
|
import org.junit.jupiter.api.Test;
|
import org.springframework.web.bind.annotation.GetMapping;
|
import org.springframework.web.bind.annotation.RestController;
|
import org.springframework.web.context.request.async.AsyncRequestNotUsableException;
|
import org.springframework.test.web.servlet.setup.MockMvcBuilders;
|
|
import java.io.IOException;
|
|
import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.get;
|
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.content;
|
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status;
|
|
class GlobalExceptionHandlerClientAbortTest {
|
|
@Test
|
void doesNotSerializeJsonAfterZipResponseBecomesUnusable() throws Exception {
|
MockMvcBuilders.standaloneSetup(new ZipController())
|
.setControllerAdvice(new GlobalExceptionHandler("test", null))
|
.build()
|
.perform(get("/zip"))
|
.andExpect(status().isOk())
|
.andExpect(content().bytes(new byte[0]));
|
}
|
|
@RestController
|
static class ZipController {
|
|
@GetMapping("/zip")
|
public void download(HttpServletResponse response) throws IOException {
|
response.setContentType("application/zip");
|
response.flushBuffer();
|
throw new AsyncRequestNotUsableException("client disconnected");
|
}
|
|
}
|
|
}
|
