misc: set up initial tests

Author: 0marperez

codegen/smithy-kotlin-codegen-testutils/src/main/kotlin/software/amazon/smithy/kotlin/codegen/test/CodegenTestUtils.kt

@@ -63,8 +63,8 @@ fun codegenSerializerForShape(
     val resolvedSettings = settings ?: model.defaultSettings(TestModelDefault.SERVICE_NAME, TestModelDefault.NAMESPACE)
     val ctx = model.newTestContext(settings = resolvedSettings)
 
-    val op = ctx.generationCtx.model.expectShape(ShapeId.from(shapeId))
-    return testRender(ctx.requestMembers(op, location)) { members, writer ->
+    val shape = ctx.generationCtx.model.expectShape(ShapeId.from(shapeId))
+    return testRender(ctx.shapeMembers(shape, location)) { members, writer ->
         SerializeStructGenerator(
             ctx.generationCtx,
             members,
@@ -121,15 +121,30 @@ fun TestContext.responseMembers(shape: Shape, location: HttpBinding.Location = H
         .map { it.member }
 }
 
-/** Retrieves Request Document members for HttpTrait-enabled protocols */
-fun TestContext.requestMembers(shape: Shape, location: HttpBinding.Location = HttpBinding.Location.DOCUMENT): List<MemberShape> {
-    val bindingIndex = HttpBindingIndex.of(this.generationCtx.model)
-    val responseBindings = bindingIndex.getRequestBindings(shape)
-
-    return responseBindings.values
-        .filter { it.location == location }
-        .sortedBy { it.memberName }
-        .map { it.member }
+/**
+ * If the shape is an operation:
+ * Attempts to retrieve request document members for HttpTrait-enabled protocols.
+ *
+ * Otherwise:
+ * Retrieves shape members
+ *
+ * TODO: Make this nicer
+ * */
+fun TestContext.shapeMembers(shape: Shape, location: HttpBinding.Location = HttpBinding.Location.DOCUMENT): List<MemberShape> {
+    when (shape) {
+        is OperationShape -> {
+            val bindingIndex = HttpBindingIndex.of(this.generationCtx.model)
+            val responseBindings = bindingIndex.getRequestBindings(shape)
+
+            return responseBindings.values
+                .filter { it.location == location }
+                .sortedBy { it.memberName }
+                .map { it.member }
+        }
+        else -> {
+            return shape.members().toList()
+        }
+    }
 }
 
 fun TestContext.toGenerationContext(): GenerationContext =

codegen/smithy-kotlin-codegen/src/test/kotlin/software/amazon/smithy/kotlin/codegen/rendering/serde/SerializeStructGeneratorTest.kt

@@ -2027,4 +2027,52 @@ class SerializeStructGeneratorTest {
 
         actual.shouldContainOnlyOnceWithDiff(expected)
     }
+
+    @Test
+    fun `it serializes an un-nested idempotency token`() {
+        val model = (
+            modelPrefix + """            
+            structure FooRequest { 
+                @idempotencyToken
+                bar: String
+            }
+        """
+            ).toSmithyModel()
+
+        val expected = """
+            serializer.serializeStruct(OBJ_DESCRIPTOR) {
+                input.bar?.let { field(BAR_DESCRIPTOR, it) } ?: field(BAR_DESCRIPTOR, context.idempotencyTokenProvider.generateToken())
+            }
+        """.trimIndent()
+
+        val actual = codegenSerializerForShape(model, "com.test#Foo").stripCodegenPrefix()
+
+        actual.shouldContainOnlyOnceWithDiff(expected)
+    }
+
+    @Test
+    fun `it serializes a nested idempotency token`() {
+        val model = (
+            modelPrefix + """            
+            structure FooRequest { 
+                bar: Bar
+            }
+            
+            structure Bar {
+                @idempotencyToken
+                baz: String
+            }
+        """
+            ).toSmithyModel()
+
+        val expected = """
+            serializer.serializeStruct(OBJ_DESCRIPTOR) {
+                input.baz?.let { field(BAZ_DESCRIPTOR, it) } ?: field(BAZ_DESCRIPTOR, context.idempotencyTokenProvider.generateToken())
+            }
+        """.trimIndent()
+
+        val actual = codegenSerializerForShape(model, "com.test#Bar").stripCodegenPrefix()
+
+        actual.shouldContainOnlyOnceWithDiff(expected)
+    }
 }