Add cbor tests from smithy + add new cborEquals body assertion

Author: alextwoods

test/protocol-tests-core/pom.xml

@@ -137,6 +137,10 @@
             <groupId>com.fasterxml.jackson.core</groupId>
             <artifactId>jackson-annotations</artifactId>
         </dependency>
+        <dependency>
+            <groupId>com.fasterxml.jackson.dataformat</groupId>
+            <artifactId>jackson-dataformat-cbor</artifactId>
+        </dependency>
     </dependencies>
 
     <!-- Disable spotbugs for this test module to speed up the build. -->

test/protocol-tests-core/src/main/java/software/amazon/awssdk/protocol/asserts/marshalling/CborBodyAssertion.java

@@ -0,0 +1,66 @@
+/*
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License").
+ * You may not use this file except in compliance with the License.
+ * A copy of the License is located at
+ *
+ *  http://aws.amazon.com/apache2.0
+ *
+ * or in the "license" file accompanying this file. This file is distributed
+ * on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
+ * express or implied. See the License for the specific language governing
+ * permissions and limitations under the License.
+ */
+
+package software.amazon.awssdk.protocol.asserts.marshalling;
+
+import static org.junit.Assert.assertEquals;
+
+import com.fasterxml.jackson.databind.JsonNode;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.databind.node.ArrayNode;
+import com.fasterxml.jackson.databind.node.DoubleNode;
+import com.fasterxml.jackson.databind.node.ObjectNode;
+import com.fasterxml.jackson.dataformat.cbor.CBORFactory;
+import com.github.tomakehurst.wiremock.verification.LoggedRequest;
+import java.util.Base64;
+
+/**
+ * Asserts on the body (expected to be CBOR) of the marshalled request.
+ */
+public class CborBodyAssertion extends MarshallingAssertion {
+    private static final ObjectMapper MAPPER = new ObjectMapper(new CBORFactory());
+
+    private final String cborEquals;
+
+    public CborBodyAssertion(String cborEquals) {
+        this.cborEquals = cborEquals;
+    }
+
+    @Override
+    protected void doAssert(LoggedRequest actual) throws Exception {
+        JsonNode expected = normalizeToDoubles(MAPPER.readTree(Base64.getDecoder().decode(cborEquals)));
+        JsonNode actualJson = normalizeToDoubles(MAPPER.readTree(actual.getBody()));
+        assertEquals(expected, actualJson);
+    }
+
+    private JsonNode normalizeToDoubles(JsonNode node) {
+        if (node.isFloat() || node.isDouble()) {
+            return DoubleNode.valueOf(node.doubleValue());
+        } else if (node.isInt() || node.isLong() || node.isShort() || node.isBigInteger() || node.isBigDecimal()) {
+            return DoubleNode.valueOf(node.doubleValue());
+        } else if (node.isArray()) {
+            ArrayNode array = MAPPER.createArrayNode();
+            for (JsonNode item : node) {
+                array.add(normalizeToDoubles(item));
+            }
+            return array;
+        } else if (node.isObject()) {
+            ObjectNode obj = MAPPER.createObjectNode();
+            node.fields().forEachRemaining(entry -> obj.set(entry.getKey(), normalizeToDoubles(entry.getValue())));
+            return obj;
+        }
+        return node;
+    }
+}

test/protocol-tests-core/src/main/java/software/amazon/awssdk/protocol/asserts/marshalling/RequestBodyAssertion.java

@@ -37,6 +37,10 @@ public void setQueryEquals(String queryEquals) {
         addAssertion(new QueryBodyAssertion(queryEquals));
     }
 
+    public void setCborEquals(String cborEquals) {
+        addAssertion(new CborBodyAssertion(cborEquals));
+    }
+
     public void setEquals(String equals) {
         addAssertion(new RawBodyAssertion(equals));
     }