Add more constraints tests wrt nesting for CBOR, Smile

Author: cowtowncoder

cbor/src/test/java/com/fasterxml/jackson/dataformat/cbor/gen/constraints/DeeplyNestedCBORReadWriteTest.java

@@ -0,0 +1,70 @@
+package com.fasterxml.jackson.dataformat.cbor.gen.constraints;
+
+import com.fasterxml.jackson.core.JsonParser;
+import com.fasterxml.jackson.core.StreamReadConstraints;
+import com.fasterxml.jackson.core.StreamWriteConstraints;
+import com.fasterxml.jackson.core.exc.StreamConstraintsException;
+
+import com.fasterxml.jackson.databind.JsonNode;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.databind.node.ObjectNode;
+
+import com.fasterxml.jackson.dataformat.cbor.CBORFactory;
+import com.fasterxml.jackson.dataformat.cbor.CBORTestBase;
+
+public class DeeplyNestedCBORReadWriteTest extends CBORTestBase
+{
+    private final ObjectMapper MAPPER_VANILLA = cborMapper();
+
+    private final ObjectMapper MAPPER_CONSTRAINED = cborMapper(
+            CBORFactory.builder()
+            // Use higher limit for writing to simplify testing setup
+                .streamReadConstraints(StreamReadConstraints.builder()
+                    .maxNestingDepth(10).build())
+                .streamWriteConstraints(StreamWriteConstraints.builder()
+                        .maxNestingDepth(12).build())
+            .build()
+            );
+
+    public void testDeepNestingRead() throws Exception
+    {
+        final byte[] DOC = MAPPER_CONSTRAINED.writeValueAsBytes(createDeepNestedDoc(11));
+        try (JsonParser p = MAPPER_CONSTRAINED.createParser(DOC)) {
+            _testDeepNestingRead(p);
+        }
+    }
+
+    private void _testDeepNestingRead(JsonParser p) throws Exception
+    {
+        try {
+            while (p.nextToken() != null) { }
+            fail("expected StreamConstraintsException");
+        } catch (StreamConstraintsException e) {
+            assertEquals("Document nesting depth (11) exceeds the maximum allowed (10, from `StreamReadConstraints.getMaxNestingDepth()`)",
+                    e.getMessage());
+        }
+    }
+
+    public void testDeepNestingWrite() throws Exception
+    {
+        final JsonNode docRoot = createDeepNestedDoc(13);
+        try {
+            MAPPER_CONSTRAINED.writeValueAsBytes(docRoot);
+            fail("Should not pass");
+        } catch (StreamConstraintsException e) {
+            assertEquals("Document nesting depth (13) exceeds the maximum allowed (12, from `StreamWriteConstraints.getMaxNestingDepth()`)",
+                    e.getMessage());
+        }
+    }
+
+    private JsonNode createDeepNestedDoc(final int depth) throws Exception
+    {
+        final ObjectNode root = MAPPER_VANILLA.createObjectNode();
+        ObjectNode curr = root;
+        for (int i = 0; i < depth; ++i) {
+            curr = curr.putObject("nested"+i);
+        }
+        curr.put("value", 42);
+        return root;
+    }
+}

smile/src/test/java/com/fasterxml/jackson/dataformat/smile/constraints/DeeplyNestedSmileReadWriteTest.java

@@ -0,0 +1,71 @@
+package com.fasterxml.jackson.dataformat.smile.constraints;
+
+import com.fasterxml.jackson.core.JsonParser;
+import com.fasterxml.jackson.core.StreamReadConstraints;
+import com.fasterxml.jackson.core.StreamWriteConstraints;
+import com.fasterxml.jackson.core.exc.StreamConstraintsException;
+
+import com.fasterxml.jackson.databind.JsonNode;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.databind.node.ObjectNode;
+
+import com.fasterxml.jackson.dataformat.smile.SmileFactory;
+import com.fasterxml.jackson.dataformat.smile.databind.SmileMapper;
+import com.fasterxml.jackson.dataformat.smile.BaseTestForSmile;
+
+public class DeeplyNestedSmileReadWriteTest extends BaseTestForSmile
+{
+    private final ObjectMapper MAPPER_VANILLA = smileMapper();
+
+    private final ObjectMapper MAPPER_CONSTRAINED = new SmileMapper(
+            SmileFactory.builder()
+            // Use higher limit for writing to simplify testing setup
+                .streamReadConstraints(StreamReadConstraints.builder()
+                    .maxNestingDepth(10).build())
+                .streamWriteConstraints(StreamWriteConstraints.builder()
+                        .maxNestingDepth(12).build())
+            .build()
+            );
+
+    public void testDeepNestingRead() throws Exception
+    {
+        final byte[] DOC = MAPPER_CONSTRAINED.writeValueAsBytes(createDeepNestedDoc(11));
+        try (JsonParser p = MAPPER_CONSTRAINED.createParser(DOC)) {
+            _testDeepNestingRead(p);
+        }
+    }
+
+    private void _testDeepNestingRead(JsonParser p) throws Exception
+    {
+        try {
+            while (p.nextToken() != null) { }
+            fail("expected StreamConstraintsException");
+        } catch (StreamConstraintsException e) {
+            assertEquals("Document nesting depth (11) exceeds the maximum allowed (10, from `StreamReadConstraints.getMaxNestingDepth()`)",
+                    e.getMessage());
+        }
+    }
+
+    public void testDeepNestingWrite() throws Exception
+    {
+        final JsonNode docRoot = createDeepNestedDoc(13);
+        try {
+            MAPPER_CONSTRAINED.writeValueAsBytes(docRoot);
+            fail("Should not pass");
+        } catch (StreamConstraintsException e) {
+            assertEquals("Document nesting depth (13) exceeds the maximum allowed (12, from `StreamWriteConstraints.getMaxNestingDepth()`)",
+                    e.getMessage());
+        }
+    }
+
+    private JsonNode createDeepNestedDoc(final int depth) throws Exception
+    {
+        final ObjectNode root = MAPPER_VANILLA.createObjectNode();
+        ObjectNode curr = root;
+        for (int i = 0; i < depth; ++i) {
+            curr = curr.putObject("nested"+i);
+        }
+        curr.put("value", 42);
+        return root;
+    }
+}