Test callsFilterInMapWithUnionWithSensitiveData

Author: trivikr

smithy-typescript-codegen/src/test/java/software/amazon/smithy/typescript/codegen/StructureGeneratorTest.java

@@ -334,6 +334,17 @@ public void callsFilterForMapWithUnionWithSensitiveData() {
                         + "  })\n");
     }
 
+    @Test
+    public void callsFilterInMapWithUnionWithSensitiveData() {
+        testStructureCodegen("test-map-with-union-with-sensitive-data.smithy",
+                            "  export const filterSensitiveLog = (obj: TestUnion): any => {\n"
+                            + "    if (obj.bar !== undefined) return {bar: obj.bar};\n"
+                            + "    if (obj.sensitiveBar !== undefined) return {[sensitiveBar]: SENSITIVE_STRING};\n"
+                            + "    if (obj.$unknown !== undefined) return {[obj.$unknown[0]]: obj.$unknown[1]};\n"
+                            + "  }\n");
+    }
+
+
     @Test
     public void callsFilterForMapWithSensitiveMember() {
         testStructureCodegen("test-map-with-sensitive-member.smithy",