fix: Generate CBOR smoke tests as AWS services (#928)

jbelkins · web-flow · commit c7379e8ce0df · 2025-05-08T19:05:35.000-05:00
diff --git a/smithy-swift-codegen/src/main/kotlin/software/amazon/smithy/swift/codegen/integration/DefaultHTTPProtocolCustomizations.kt b/smithy-swift-codegen/src/main/kotlin/software/amazon/smithy/swift/codegen/integration/DefaultHTTPProtocolCustomizations.kt
@@ -47,4 +47,6 @@ abstract class DefaultHTTPProtocolCustomizations : HTTPProtocolCustomizable {
     override val unknownServiceErrorSymbol: Symbol = ClientRuntimeTypes.Http.UnknownHttpServiceError
 
     override val defaultTimestampFormat = TimestampFormatTrait.Format.EPOCH_SECONDS
+
+    override fun smokeTestGenerator(ctx: ProtocolGenerator.GenerationContext): SmokeTestGenerator = SmokeTestGenerator(ctx)
 }
diff --git a/smithy-swift-codegen/src/main/kotlin/software/amazon/smithy/swift/codegen/integration/HTTPBindingProtocolGenerator.kt b/smithy-swift-codegen/src/main/kotlin/software/amazon/smithy/swift/codegen/integration/HTTPBindingProtocolGenerator.kt
@@ -406,7 +406,7 @@ abstract class HTTPBindingProtocolGenerator(
         }
     }
 
-    override fun generateSmokeTests(ctx: ProtocolGenerator.GenerationContext) = SmokeTestGenerator(ctx).generateSmokeTests()
+    override fun generateSmokeTests(ctx: ProtocolGenerator.GenerationContext) = customizations.smokeTestGenerator(ctx).generateSmokeTests()
 
     override fun initializeMiddleware(ctx: ProtocolGenerator.GenerationContext) {
         val resolver = getProtocolHttpBindingResolver(ctx, defaultContentType)
diff --git a/smithy-swift-codegen/src/main/kotlin/software/amazon/smithy/swift/codegen/integration/HTTPProtocolCustomizable.kt b/smithy-swift-codegen/src/main/kotlin/software/amazon/smithy/swift/codegen/integration/HTTPProtocolCustomizable.kt
@@ -60,4 +60,6 @@ interface HTTPProtocolCustomizable {
     val unknownServiceErrorSymbol: Symbol
 
     val defaultTimestampFormat: TimestampFormatTrait.Format
+
+    fun smokeTestGenerator(ctx: ProtocolGenerator.GenerationContext): SmokeTestGenerator
 }

Original file line number	Diff line number	Diff line change
`@@ -47,4 +47,6 @@ abstract class DefaultHTTPProtocolCustomizations : HTTPProtocolCustomizable {`
`47`	`47`	`override val unknownServiceErrorSymbol: Symbol = ClientRuntimeTypes.Http.UnknownHttpServiceError`
`48`	`48`
`49`	`49`	`override val defaultTimestampFormat = TimestampFormatTrait.Format.EPOCH_SECONDS`
	`50`	`+`
	`51`	`+ override fun smokeTestGenerator(ctx: ProtocolGenerator.GenerationContext): SmokeTestGenerator = SmokeTestGenerator(ctx)`
`50`	`52`	`}`
Original file line number	Diff line number	Diff line change
`@@ -406,7 +406,7 @@ abstract class HTTPBindingProtocolGenerator(`
`406`	`406`	`}`
`407`	`407`	`}`
`408`	`408`
`409`		`- override fun generateSmokeTests(ctx: ProtocolGenerator.GenerationContext) = SmokeTestGenerator(ctx).generateSmokeTests()`
	`409`	`+ override fun generateSmokeTests(ctx: ProtocolGenerator.GenerationContext) = customizations.smokeTestGenerator(ctx).generateSmokeTests()`
`410`	`410`
`411`	`411`	`override fun initializeMiddleware(ctx: ProtocolGenerator.GenerationContext) {`
`412`	`412`	`val resolver = getProtocolHttpBindingResolver(ctx, defaultContentType)`
Original file line number	Diff line number	Diff line change
`@@ -60,4 +60,6 @@ interface HTTPProtocolCustomizable {`
`60`	`60`	`val unknownServiceErrorSymbol: Symbol`
`61`	`61`
`62`	`62`	`val defaultTimestampFormat: TimestampFormatTrait.Format`
	`63`	`+`
	`64`	`+ fun smokeTestGenerator(ctx: ProtocolGenerator.GenerationContext): SmokeTestGenerator`
`63`	`65`	`}`