Add test cases for required+default+httpPayload

disneystreaming · Jun 25, 2024 · 4ce6c2e · 4ce6c2e
1 parent 8eb295c
commit 4ce6c2e
Show file tree

Hide file tree

Showing 2 changed files with 64 additions and 2 deletions.
diff --git a/modules/protocol-tests/resources/META-INF/smithy/HttpPayloadWithDefault.smithy b/modules/protocol-tests/resources/META-INF/smithy/HttpPayloadWithDefault.smithy
@@ -54,3 +54,52 @@ apply HttpPayloadWithDefault @httpResponseTests([{
         body: "default value"
     }
 }])
+
+apply HttpPayloadRequiredWithDefault @httpRequestTests([{
+    id: "SimpleRestJsonSomeRequiredHttpPayloadWithDefault"
+    documentation: "Pass JSON string value as is if payload provided"
+    protocol: simpleRestJson
+    method: "PUT"
+    uri: "/httpPayloadRequiredWithDefault"
+    body: "\"custom value\""
+    headers: {
+        "Content-Type": "application/json"
+    }
+    requireHeaders: ["Content-Length"]
+    params: {
+        body: "custom value"
+    }
+    bodyMediaType: "application/json"
+}, {
+    id: "SimpleRestJsonNoneRequiredHttpPayloadWithDefault"
+    documentation: "Use default value when there is no payload"
+    protocol: simpleRestJson
+    method: "PUT"
+    uri: "/httpPayloadRequiredWithDefault"
+    params: {
+        body: "default value"
+    }
+}])
+
+apply HttpPayloadRequiredWithDefault @httpResponseTests([{
+    id: "SimpleRestJsonSomeRequiredHttpPayloadWithDefault"
+    documentation: "Pass JSON string value as is if payload provided"
+    protocol: simpleRestJson
+    code: 200
+    body: "\"custom value\""
+    bodyMediaType: "application/json"
+    headers: {
+        "Content-Type": "application/json"
+    }
+    params: {
+        body: "custom value"
+    }
+}, {
+    id: "SimpleRestJsonNoneRequiredHttpPayloadWithDefault"
+    documentation: "Use default value when there is no payload"
+    protocol: simpleRestJson
+    code: 200
+    params: {
+        body: "default value"
+    }
+}])
diff --git a/modules/protocol-tests/resources/META-INF/smithy/Pizza.smithy b/modules/protocol-tests/resources/META-INF/smithy/Pizza.smithy
@@ -4,12 +4,11 @@ namespace alloy.test
 
 use alloy#simpleRestJson
 
-
 @simpleRestJson
 service PizzaAdminService {
     version: "1.0.0",
     errors: [GenericServerError, GenericClientError],
-    operations: [AddMenuItem, GetMenu, Version, Health, HeaderEndpoint, RoundTrip, GetEnum, GetIntEnum, CustomCode, HttpPayloadWithDefault]
+    operations: [AddMenuItem, GetMenu, Version, Health, HeaderEndpoint, RoundTrip, GetEnum, GetIntEnum, CustomCode, HttpPayloadWithDefault, HttpPayloadRequiredWithDefault]
 }
 
 @http(method: "POST", uri: "/restaurant/{restaurant}/menu/item", code: 201)
@@ -316,3 +315,17 @@ structure HttpPayloadWithDefaultInputOutput {
     @default("default value")
     body: String,
 }
+
+@idempotent
+@http(uri: "/httpPayloadRequiredWithDefault", method: "PUT")
+operation HttpPayloadRequiredWithDefault {
+    input: HttpPayloadRequiredWithDefaultInputOutput
+    output: HttpPayloadRequiredWithDefaultInputOutput
+}
+
+structure HttpPayloadRequiredWithDefaultInputOutput {
+    @httpPayload
+    @default("default value")
+    @required
+    body: String
+}