work on test

prestodb · Dec 3, 2024 · a329169 · a329169
1 parent b1e2b2f
commit a329169
Showing 1 changed file with 9 additions and 0 deletions.
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
@@ -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()
     {