diff --git a/presto-tests/src/test/java/com/facebook/presto/server/TestAsyncPageTransportServlet.java b/presto-tests/src/test/java/com/facebook/presto/server/TestAsyncPageTransportServlet.java index d67dd943a7479..cbf5fd91510b7 100644 --- a/presto-tests/src/test/java/com/facebook/presto/server/TestAsyncPageTransportServlet.java +++ b/presto-tests/src/test/java/com/facebook/presto/server/TestAsyncPageTransportServlet.java @@ -34,6 +34,7 @@ class TestServlet TaskId taskId; OutputBufferId bufferId; String requestURI; + HttpServletRequest request; long token; void parse(String uri) throws IOException @@ -50,6 +51,7 @@ protected void processRequest( this.taskId = taskId; this.bufferId = bufferId; this.token = token; + this.request = request; } @Override @@ -80,6 +82,13 @@ public void testParsing() assertEquals(789, servlet.token); } + @Test + public void testSanitization() + { + TestServlet servlet = parse("/v1/task/async/0.1.2.3.4/results/456/789"); + // verify this.request here + } + @Test (expectedExceptions = { IllegalArgumentException.class }) public void testParseTooFewElements() {