Add tests for complex shapes inside Union

Author: trivikr

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

@@ -179,6 +179,39 @@ public void callsFilterInUnionWithoutSensitiveData() {
                                 + "  }\n");
     }
 
+    @Test
+    public void callsFilterInUnionWithStructure() {
+        testStructureCodegen("test-union-with-structure.smithy",
+                                "  export const filterSensitiveLog = (obj: TestUnion): any => {\n"
+                                + "    if (obj.fooUser !== undefined) return {fooUser:\n"
+                                + "    User.filterSensitiveLog(obj.fooUser)\n"
+                                + "    };\n"
+                                + "    if (obj.$unknown !== undefined) return {[obj.$unknown[0]]: 'UNKNOWN'};\n"
+                                + "  }\n");
+    }
+
+    @Test
+    public void callsFilterInUnionWithList() {
+        testStructureCodegen("test-union-with-list.smithy",
+                                "  export const filterSensitiveLog = (obj: TestUnion): any => {\n"
+                                + "    if (obj.list !== undefined) return {list:\n"
+                                + "    obj.list\n"
+                                + "    };\n"
+                                + "    if (obj.$unknown !== undefined) return {[obj.$unknown[0]]: 'UNKNOWN'};\n"
+                                + "  }\n");
+    }
+
+    @Test
+    public void callsFilterInUnionWithMap() {
+        testStructureCodegen("test-union-with-map.smithy",
+                                "  export const filterSensitiveLog = (obj: TestUnion): any => {\n"
+                                + "    if (obj.map !== undefined) return {map:\n"
+                                + "    obj.map\n"
+                                + "    };\n"
+                                + "    if (obj.$unknown !== undefined) return {[obj.$unknown[0]]: 'UNKNOWN'};\n"
+                                + "  }\n");
+    }
+
     @Test
     public void filtersStreamingUnion() {
         testStructureCodegen("test-streaming-union.smithy",

smithy-typescript-codegen/src/test/resources/software/amazon/smithy/typescript/codegen/test-union-with-list.smithy

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

smithy-typescript-codegen/src/test/resources/software/amazon/smithy/typescript/codegen/test-union-with-map.smithy

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

smithy-typescript-codegen/src/test/resources/software/amazon/smithy/typescript/codegen/test-union-with-structure.smithy

@@ -0,0 +1,23 @@
+namespace smithy.example
+
+service Example {
+    version: "1.0.0",
+    operations: [GetFoo]
+}
+
+operation GetFoo {
+    input: GetFooInput
+}
+
+structure GetFooInput {
+    foo: TestUnion
+}
+
+union TestUnion {
+    fooUser: User
+}
+
+structure User {
+    firstname: String,
+    lastname: String
+}