Test callsFilterForMapWithUnionWithSensitiveData

Author: trivikr

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

@@ -292,19 +292,8 @@ public void filtersSensitiveMemberPointingToList() {
     }
 
     @Test
-    public void callsFilterForMapWithSensitiveData() {
-        testStructureCodegen("test-map-with-sensitive-data.smithy",
-                                "  export const filterSensitiveLog = (obj: User): any => ({\n"
-                                + "    ...obj,\n"
-                                + "    ...(obj.password && { password:\n"
-                                + "      SENSITIVE_STRING\n"
-                                + "    }),\n"
-                                + "  })\n");
-    }
-
-    @Test
-    public void callsFilterInMapWithSensitiveData() {
-        testStructureCodegen("test-map-with-sensitive-data.smithy",
+    public void callsFilterForMapWithStructureWithSensitiveData() {
+        testStructureCodegen("test-map-with-structure-with-sensitive-data.smithy",
                             "  export const filterSensitiveLog = (obj: GetFooInput): any => ({\n"
                             + "    ...obj,\n"
                             + "    ...(obj.foo && { foo:\n"
@@ -318,6 +307,33 @@ public void callsFilterInMapWithSensitiveData() {
                             + "  })\n");
     }
 
+    @Test
+    public void callsFilterInMapWithStructureWithSensitiveData() {
+        testStructureCodegen("test-map-with-structure-with-sensitive-data.smithy",
+                                "  export const filterSensitiveLog = (obj: User): any => ({\n"
+                                + "    ...obj,\n"
+                                + "    ...(obj.password && { password:\n"
+                                + "      SENSITIVE_STRING\n"
+                                + "    }),\n"
+                                + "  })\n");
+    }
+
+    @Test
+    public void callsFilterForMapWithUnionWithSensitiveData() {
+        testStructureCodegen("test-map-with-union-with-sensitive-data.smithy",
+                        "  export const filterSensitiveLog = (obj: GetFooInput): any => ({\n"
+                        + "    ...obj,\n"
+                        + "    ...(obj.foo && { foo:\n"
+                        + "      Object.entries(obj.foo).reduce((acc: any, [key, value]: [string, TestUnion]) => ({\n"
+                        + "        ...acc,\n"
+                        + "        [key]:\n"
+                        + "          TestUnion.filterSensitiveLog(value)\n"
+                        + "        ,\n"
+                        + "      }), {})\n"
+                        + "    }),\n"
+                        + "  })\n");
+    }
+
     @Test
     public void callsFilterForMapWithSensitiveMember() {
         testStructureCodegen("test-map-with-sensitive-member.smithy",

smithy-typescript-codegen/src/test/resources/software/amazon/smithy/typescript/codegen/test-map-with-sensitive-data.smithy renamed to smithy-typescript-codegen/src/test/resources/software/amazon/smithy/typescript/codegen/test-map-with-structure-with-sensitive-data.smithy

@@ -0,0 +1,26 @@
+namespace smithy.example
+
+service Example {
+    version: "1.0.0",
+    operations: [GetFoo]
+}
+
+operation GetFoo {
+    input: GetFooInput
+}
+
+structure GetFooInput {
+    foo: UserMap
+}
+
+map UserMap {
+    key: String,
+    value: TestUnion
+}
+
+union TestUnion {
+    bar: String,
+
+    @sensitive
+    sensitiveBar: String
+}