test: add protocol test for union (#4645)

aws · Jun 6, 2024 · 74e3e02 · 74e3e02
1 parent 2d3f3d1
commit 74e3e02
Show file tree

Hide file tree

Showing 3 changed files with 157 additions and 0 deletions.
diff --git a/.changes/next-release/feature-protocol-2da87ea9.json b/.changes/next-release/feature-protocol-2da87ea9.json
@@ -0,0 +1,5 @@
+{
+  "type": "feature",
+  "category": "protocol",
+  "description": "test for union"
+}
diff --git a/test/fixtures/protocol/output/json.json b/test/fixtures/protocol/output/json.json
@@ -432,6 +432,82 @@
       }
     ]
   },
+  {
+    "description": "Union output, ignore __type before",
+    "metadata": {
+      "protocol": "json"
+    },
+    "shapes": {
+      "OutputShape": {
+        "type": "structure",
+        "members": {
+          "Str": {
+            "shape": "StringType"
+          }
+        },
+        "union": true
+      },
+      "StringType": {
+        "type": "string"
+      }
+    },
+    "cases": [
+      {
+        "given": {
+          "output": {
+            "shape": "OutputShape"
+          },
+          "name": "OperationName"
+        },
+        "result": {
+          "Str": "foo"
+        },
+        "response": {
+          "status_code": 200,
+          "headers": {},
+          "body": "{\"__type\": \"com.foo#OutputShape\", \"Str\": \"foo\"}"
+        }
+      }
+    ]
+  },
+  {
+    "description": "Union output, ignore __type after",
+    "metadata": {
+      "protocol": "json"
+    },
+    "shapes": {
+      "OutputShape": {
+        "type": "structure",
+        "members": {
+          "Str": {
+            "shape": "StringType"
+          }
+        },
+        "union": true
+      },
+      "StringType": {
+        "type": "string"
+      }
+    },
+    "cases": [
+      {
+        "given": {
+          "output": {
+            "shape": "OutputShape"
+          },
+          "name": "OperationName"
+        },
+        "result": {
+          "Str": "foo"
+        },
+        "response": {
+          "status_code": 200,
+          "headers": {},
+          "body": "{\"Str\": \"foo\", \"__type\": \"com.foo#OutputShape\"}"
+        }
+      }
+    ]
+  },
   {
     "description": "Modeled exceptions",
     "metadata": {

diff --git a/test/fixtures/protocol/output/rest-json.json b/test/fixtures/protocol/output/rest-json.json
@@ -844,6 +844,82 @@
       }
     ]
   },
+  {
+    "description": "Union output, ignore __type before",
+    "metadata": {
+      "protocol": "json"
+    },
+    "shapes": {
+      "OutputShape": {
+        "type": "structure",
+        "members": {
+          "Str": {
+            "shape": "StringType"
+          }
+        },
+        "union": true
+      },
+      "StringType": {
+        "type": "string"
+      }
+    },
+    "cases": [
+      {
+        "given": {
+          "output": {
+            "shape": "OutputShape"
+          },
+          "name": "OperationName"
+        },
+        "result": {
+          "Str": "foo"
+        },
+        "response": {
+          "status_code": 200,
+          "headers": {},
+          "body": "{\"__type\": \"com.foo#OutputShape\", \"Str\": \"foo\"}"
+        }
+      }
+    ]
+  },
+  {
+    "description": "Union output, ignore __type after",
+    "metadata": {
+      "protocol": "json"
+    },
+    "shapes": {
+      "OutputShape": {
+        "type": "structure",
+        "members": {
+          "Str": {
+            "shape": "StringType"
+          }
+        },
+        "union": true
+      },
+      "StringType": {
+        "type": "string"
+      }
+    },
+    "cases": [
+      {
+        "given": {
+          "output": {
+            "shape": "OutputShape"
+          },
+          "name": "OperationName"
+        },
+        "result": {
+          "Str": "foo"
+        },
+        "response": {
+          "status_code": 200,
+          "headers": {},
+          "body": "{\"Str\": \"foo\", \"__type\": \"com.foo#OutputShape\"}"
+        }
+      }
+    ]
+  },
   {
     "description": "Modeled exceptions",
     "metadata": {