add map of strings and ints to unit test

Author: jmontgomery-castlighthealth

src/test/java/at/grahsl/kafka/connect/mongodb/converter/RecordConverterTest.java

@@ -34,6 +34,7 @@
 import java.util.Arrays;
 import java.util.HashMap;
 import java.util.LinkedHashMap;
+import java.util.List;
 import java.util.Map;
 
 import static org.junit.jupiter.api.Assertions.*;
@@ -45,7 +46,8 @@ public class RecordConverterTest {
     public static Schema OBJ_SCHEMA_1;
     public static Struct OBJ_STRUCT_1;
     public static Map OBJ_MAP_1;
-    public static BsonDocument EXPECTED_BSON_DOC_BYTES_1;
+    public static BsonDocument EXPECTED_BSON_DOC_OBJ_STRUCT_1;
+    public static BsonDocument EXPECTED_BSON_DOC_OBJ_MAP_1;
     public static BsonDocument EXPECTED_BSON_DOC_RAW_1;
 
     @BeforeAll
@@ -82,6 +84,8 @@ public static void initializeTestData() {
                                     .build())
                 )
                 .field("mySubDoc2", SchemaBuilder.map(Schema.STRING_SCHEMA, Schema.INT32_SCHEMA).build())
+                .field( "myMapOfStrings", SchemaBuilder.map(Schema.STRING_SCHEMA, SchemaBuilder.array(Schema.STRING_SCHEMA).build()).build())
+                .field( "myMapOfInts", SchemaBuilder.map(Schema.STRING_SCHEMA, SchemaBuilder.array(Schema.INT32_SCHEMA).build()).build())
                 .field("myBytes", Schema.BYTES_SCHEMA)
                 .field("myDate", Date.SCHEMA)
                 .field("myTimestamp", Timestamp.SCHEMA)
@@ -108,6 +112,16 @@ public static void initializeTestData() {
                         )
                 )
                 .put("mySubDoc2",new HashMap<String,Integer>(){{ put("k1",9); put("k2",8); put("k3",7);}})
+                .put("myMapOfStrings", new HashMap<String, List<String>>(){{
+                   put("k1", Arrays.asList("v1-a", "v1-b"));
+                   put("k2", Arrays.asList("v2-a"));
+                   put("k3", Arrays.asList("v3-a", "v3-b", "v3-c"));
+                }})
+                .put("myMapOfInts", new HashMap<String, List<Integer>>(){{
+                   put("k1", Arrays.asList(11, 12));
+                   put("k2", Arrays.asList(21));
+                   put("k3", Arrays.asList(31, 32, 33));
+                }})
                 .put("myBytes", new byte[]{75, 97, 102, 107, 97, 32, 114, 111, 99, 107, 115, 33})
                 .put("myDate", java.util.Date.from(ZonedDateTime.of(
                                 LocalDate.of(2017,3,17), LocalTime.MIDNIGHT, ZoneOffset.systemDefault()
@@ -154,7 +168,7 @@ public static void initializeTestData() {
         // thus I'm cheating a little by using a Decimal128 here...
         OBJ_MAP_1.put("myDecimal", Decimal128.parse("12345.6789"));
 
-        EXPECTED_BSON_DOC_BYTES_1 = new BsonDocument()
+        BsonDocument commonMapAndStructFields = new BsonDocument()
                 .append("_id", new BsonString("1234567890"))
                 .append("myString", new BsonString("some foo bla text"))
                 .append("myInt", new BsonInt32(42))
@@ -192,7 +206,32 @@ public static void initializeTestData() {
                 ))
                 .append("myDecimal", new BsonDecimal128(new Decimal128(new BigDecimal("12345.6789"))));
 
-        EXPECTED_BSON_DOC_RAW_1 = EXPECTED_BSON_DOC_BYTES_1.clone();
+        EXPECTED_BSON_DOC_OBJ_STRUCT_1 = commonMapAndStructFields.clone()
+           .append("myMapOfStrings", new BsonDocument("k1", new BsonInt32(9))
+              .append("k1", new BsonArray(Arrays.asList(
+                 new BsonString("v1-a"),
+                 new BsonString("v1-b"))))
+              .append("k2", new BsonArray(Arrays.asList(
+                 new BsonString("v2-a"))))
+              .append("k3", new BsonArray(Arrays.asList(
+                 new BsonString("v3-a"),
+                 new BsonString("v3-b"),
+                 new BsonString("v3-c"))))
+           ).append("myMapOfInts", new BsonDocument("k1", new BsonInt32(9))
+              .append("k1", new BsonArray(Arrays.asList(
+                 new BsonInt32(11),
+                 new BsonInt32(12))))
+              .append("k2", new BsonArray(Arrays.asList(
+                 new BsonInt32(21))))
+              .append("k3", new BsonArray(Arrays.asList(
+                 new BsonInt32(31),
+                 new BsonInt32(32),
+                 new BsonInt32(33))))
+           );
+
+        EXPECTED_BSON_DOC_OBJ_MAP_1 = commonMapAndStructFields.clone();
+
+        EXPECTED_BSON_DOC_RAW_1 = commonMapAndStructFields.clone();
         EXPECTED_BSON_DOC_RAW_1.replace("myBytes",new BsonString("S2Fma2Egcm9ja3Mh"));
         EXPECTED_BSON_DOC_RAW_1.replace("myDate",new BsonInt64(1489705200000L));
         EXPECTED_BSON_DOC_RAW_1.replace("myTimestamp",new BsonInt64(1489705200000L));
@@ -216,7 +255,7 @@ public void testJsonRawStringConversion() {
     public void testAvroOrJsonWithSchemaConversion() {
         RecordConverter converter = new AvroJsonSchemafulRecordConverter();
         assertAll("",
-                () -> assertEquals(EXPECTED_BSON_DOC_BYTES_1, converter.convert(OBJ_SCHEMA_1, OBJ_STRUCT_1)),
+                () -> assertEquals(EXPECTED_BSON_DOC_OBJ_STRUCT_1, converter.convert(OBJ_SCHEMA_1, OBJ_STRUCT_1)),
                 () -> assertThrows(DataException.class, () -> converter.convert(OBJ_SCHEMA_1,null)),
                 () -> assertThrows(DataException.class, () -> converter.convert(null, OBJ_STRUCT_1)),
                 () -> assertThrows(DataException.class, () -> converter.convert(null,null))
@@ -228,7 +267,7 @@ public void testAvroOrJsonWithSchemaConversion() {
     public void testJsonObjectConversion() {
         RecordConverter converter = new JsonSchemalessRecordConverter();
         assertAll("",
-                () -> assertEquals(EXPECTED_BSON_DOC_BYTES_1, converter.convert(null, OBJ_MAP_1)),
+                () -> assertEquals(EXPECTED_BSON_DOC_OBJ_MAP_1, converter.convert(null, OBJ_MAP_1)),
                 () -> assertThrows(DataException.class, () -> converter.convert(null,null))
         );
     }