Test callsFilterForListWithUnionWithSensitiveData

Author: trivikr

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

@@ -202,8 +202,8 @@ public void filtersSensitiveMemberPointingToUnion() {
     }
 
     @Test
-    public void callsFilterForListWithSensitiveData() {
-        testStructureCodegen("test-list-with-sensitive-data.smithy",
+    public void callsFilterForListWithStructureWithSensitiveData() {
+        testStructureCodegen("test-list-with-structure-with-sensitive-data.smithy",
                                 "  export const filterSensitiveLog = (obj: GetFooInput): any => ({\n"
                                 + "    ...obj,\n"
                                 + "    ...(obj.foo && { foo:\n"
@@ -216,8 +216,8 @@ public void callsFilterForListWithSensitiveData() {
     }
 
     @Test
-    public void callsFilterInListWithSensitiveData() {
-        testStructureCodegen("test-list-with-sensitive-data.smithy",
+    public void callsFilterInListWithStructureWithSensitiveData() {
+        testStructureCodegen("test-list-with-structure-with-sensitive-data.smithy",
                                 "  export const filterSensitiveLog = (obj: User): any => ({\n"
                                 + "    ...obj,\n"
                                 + "    ...(obj.password && { password:\n"
@@ -226,6 +226,20 @@ public void callsFilterInListWithSensitiveData() {
                                 + "  })\n");
     }
 
+    @Test
+    public void callsFilterForListWithUnionWithSensitiveData() {
+        testStructureCodegen("test-list-with-union-with-sensitive-data.smithy",
+                                "  export const filterSensitiveLog = (obj: GetFooInput): any => ({\n"
+                                + "    ...obj,\n"
+                                + "    ...(obj.foo && { foo:\n"
+                                + "      obj.foo.map(\n"
+                                + "        item =>\n"
+                                + "        TestUnion.filterSensitiveLog(item)\n"
+                                + "      )\n"
+                                + "    }),\n"
+                                + "  })\n");
+    }
+
     @Test
     public void callsFilterForListWithSensitiveMember() {
         testStructureCodegen("test-list-with-sensitive-member.smithy",

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

@@ -0,0 +1,25 @@
+namespace smithy.example
+
+service Example {
+    version: "1.0.0",
+    operations: [GetFoo]
+}
+
+operation GetFoo {
+    input: GetFooInput
+}
+
+structure GetFooInput {
+    foo: UserList
+}
+
+list UserList {
+    member: TestUnion
+}
+
+union TestUnion {
+    bar: String,
+
+    @sensitive
+    sensitiveBar: String
+}