Explicitly add KOTLIN_TEST dependency

smithy-lang · Oct 2, 2023 · 773f558 · 773f558
1 parent 6292691
commit 773f558
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/...e/amazon/smithy/kotlin/codegen/rendering/protocol/HttpProtocolUnitTestRequestGenerator.kt b/...e/amazon/smithy/kotlin/codegen/rendering/protocol/HttpProtocolUnitTestRequestGenerator.kt
@@ -22,7 +22,9 @@ open class HttpProtocolUnitTestRequestGenerator protected constructor(builder: B
     override fun renderTestBody(test: HttpRequestTestCase) {
         writer.addImport(KotlinDependency.SMITHY_TEST.namespace, "*")
         writer.addImport(KotlinDependency.HTTP.namespace, "HttpMethod")
+        writer.addImport(KotlinDependency.KOTLIN_TEST.namespace, "*")
         writer.dependencies.addAll(KotlinDependency.SMITHY_TEST.dependencies)
+        writer.dependencies.addAll(KotlinDependency.KOTLIN_TEST.dependencies)
         renderExpectedBlock(test)
         writer.write("")
         renderOperationBlock(test)

diff --git a/.../amazon/smithy/kotlin/codegen/rendering/protocol/HttpProtocolUnitTestResponseGenerator.kt b/.../amazon/smithy/kotlin/codegen/rendering/protocol/HttpProtocolUnitTestResponseGenerator.kt
@@ -40,7 +40,9 @@ open class HttpProtocolUnitTestResponseGenerator protected constructor(builder:
     override fun renderTestBody(test: HttpResponseTestCase) {
         writer.addImport(KotlinDependency.SMITHY_TEST.namespace, "*")
         writer.addImport(KotlinDependency.HTTP.namespace, "HttpStatusCode")
+        writer.addImport(KotlinDependency.KOTLIN_TEST.namespace, "*")
         writer.dependencies.addAll(KotlinDependency.SMITHY_TEST.dependencies)
+        writer.dependencies.addAll(KotlinDependency.KOTLIN_TEST.dependencies)
         renderExpectedBlock(test)
         writer.write("")
         renderTestBlock(test)