non blocking test

FasterXML · Aug 23, 2023 · 6f3a9dc · 6f3a9dc
1 parent 954a2d0
commit 6f3a9dc
Showing 1 changed file with 30 additions and 11 deletions.
diff --git a/src/test/java/com/fasterxml/jackson/core/read/LargeNameReadTest.java b/src/test/java/com/fasterxml/jackson/core/read/LargeNameReadTest.java
@@ -6,6 +6,10 @@
 import com.fasterxml.jackson.core.JsonToken;
 import com.fasterxml.jackson.core.StreamReadConstraints;
 import com.fasterxml.jackson.core.exc.StreamConstraintsException;
+import com.fasterxml.jackson.core.json.async.NonBlockingJsonParser;
+
+import java.io.IOException;
+import java.nio.charset.StandardCharsets;
 
 public class LargeNameReadTest extends BaseTest {
 
@@ -14,10 +18,7 @@ public void testLargeName() throws Exception
         final String doc = generateJSON(1000);
         final JsonFactory jsonFactory = JsonFactory.builder().build();
         try (JsonParser jp = createParserUsingStream(jsonFactory, doc, "UTF-8")) {
-            JsonToken jsonToken;
-            while ((jsonToken = jp.nextToken()) != null) {
-
-            }
+            consumeTokens(jp);
         }
     }
 
@@ -28,10 +29,24 @@ public void testLargeNameWithSmallLimit() throws Exception
                 .streamReadConstraints(StreamReadConstraints.builder().maxNameLength(100).build())
                 .build();
         try (JsonParser jp = createParserUsingStream(jsonFactory, doc, "UTF-8")) {
-            JsonToken jsonToken;
-            while ((jsonToken = jp.nextToken()) != null) {
+            consumeTokens(jp);
+            fail("expected StreamConstraintsException");
+        } catch (StreamConstraintsException e) {
+            assertTrue("Unexpected exception message: " + e.getMessage(),
+                    e.getMessage().contains("Name value length"));
+        }
+    }
 
-            }
+    public void testAsyncLargeNameWithSmallLimit() throws Exception
+    {
+        final byte[] doc = generateJSON(1000).getBytes(StandardCharsets.UTF_8);
+        final JsonFactory jsonFactory = JsonFactory.builder()
+                .streamReadConstraints(StreamReadConstraints.builder().maxNameLength(100).build())
+                .build();
+
+        try (NonBlockingJsonParser jp = (NonBlockingJsonParser) jsonFactory.createNonBlockingByteArrayParser()) {
+            jp.feedInput(doc, 0, doc.length);
+            consumeTokens(jp);
             fail("expected StreamConstraintsException");
         } catch (StreamConstraintsException e) {
             assertTrue("Unexpected exception message: " + e.getMessage(),
@@ -46,17 +61,21 @@ public void testReaderLargeNameWithSmallLimit() throws Exception
                 .streamReadConstraints(StreamReadConstraints.builder().maxNameLength(100).build())
                 .build();
         try (JsonParser jp = createParserUsingReader(jsonFactory, doc)) {
-            JsonToken jsonToken;
-            while ((jsonToken = jp.nextToken()) != null) {
-                System.out.println(jsonToken);
-            }
+            consumeTokens(jp);
             fail("expected StreamConstraintsException");
         } catch (StreamConstraintsException e) {
             assertTrue("Unexpected exception message: " + e.getMessage(),
                     e.getMessage().contains("Name value length"));
         }
     }
 
+    private void consumeTokens(JsonParser jp) throws IOException {
+        JsonToken jsonToken;
+        while ((jsonToken = jp.nextToken()) != null) {
+
+        }
+    }
+
     private String generateJSON(final int nameLen) {
         final StringBuilder sb = new StringBuilder();
         sb.append("{\"");