fix: Fix tests related to operation doc header (#1231)

awslabs · Nov 22, 2023 · a4998bc · a4998bc
1 parent 1732f96
commit a4998bc
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/...wift-codegen/src/test/kotlin/software/amazon/smithy/aws/swift/codegen/EventStreamTests.kt b/...wift-codegen/src/test/kotlin/software/amazon/smithy/aws/swift/codegen/EventStreamTests.kt
@@ -193,6 +193,8 @@ extension EventStreamTestClientTypes.TestStream: ClientRuntime.MessageUnmarshall
         val contents = getFileContents(context.manifest, "/Example/EventStreamTestClient.swift")
         var expected = """
 extension EventStreamTestClient: EventStreamTestClientProtocol {
+    /// Performs the `TestStreamOp` operation on the `TestService` service.
+    ///
     /// This operation is cool.
     ///
     /// - Parameter TestStreamOpInput : [no documentation found]
@@ -239,7 +241,7 @@ extension EventStreamTestClient: EventStreamTestClientProtocol {
     }
 
 }
-        """.trimIndent()
+"""
         contents.shouldContainOnlyOnce(expected)
     }
 

diff --git a/...st/kotlin/software/amazon/smithy/aws/swift/codegen/awsquery/AWSQueryOperationStackTest.kt b/...st/kotlin/software/amazon/smithy/aws/swift/codegen/awsquery/AWSQueryOperationStackTest.kt
@@ -19,9 +19,10 @@ class AWSQueryOperationStackTest {
         val context = setupTests("awsquery/query-empty-input-output.smithy", "aws.protocoltests.query#AwsQuery")
         val contents = getFileContents(context.manifest, "/Example/QueryProtocolClient.swift")
         contents.shouldSyntacticSanityCheck()
-        val expectedContents =
-            """
+        val expectedContents = """
 extension QueryProtocolClient: QueryProtocolClientProtocol {
+    /// Performs the `NoInputAndOutput` operation on the `AwsQuery` service.
+    ///
     /// This is a very cool operation.
     ///
     /// - Parameter NoInputAndOutputInput : [no documentation found]
@@ -57,7 +58,8 @@ extension QueryProtocolClient: QueryProtocolClientProtocol {
         return result
     }
 
-}"""
+}
+"""
         contents.shouldContainOnlyOnce(expectedContents)
     }