Skip to content

Commit a668184

Browse files
ianbotsfianbotsf
committed
fix: correctly deserialize/render input values for smoke tests
1 parent 1d3028e commit a668184

File tree

4 files changed

+24
-92
lines changed

4 files changed

+24
-92
lines changed

‎codegen/smithy-kotlin-codegen/src/main/kotlin/software/amazon/smithy/kotlin/codegen/rendering/ShapeValueGenerator.kt‎

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -78,7 +78,7 @@ class ShapeValueGenerator(
7878
private fun classDeclaration(writer: KotlinWriter, shape: StructureShape, block: () -> Unit) {
7979
val symbol = symbolProvider.toSymbol(shape)
8080
// invoke the generated DSL builder for the class
81-
writer.writeInline("#L {", symbol.name)
81+
writer.writeInline("#T {", symbol)
8282
.ensureNewline()
8383
.indent()
8484
.call { block() }
@@ -129,7 +129,7 @@ class ShapeValueGenerator(
129129
val suffix = when {
130130
shape.isEnum -> {
131131
val symbol = symbolProvider.toSymbol(shape)
132-
writer.writeInline("#L.fromValue(", symbol.name)
132+
writer.writeInline("#T.fromValue(", symbol)
133133
")"
134134
}
135135

@@ -243,14 +243,14 @@ class ShapeValueGenerator(
243243
ShapeType.DOUBLE,
244244
ShapeType.FLOAT,
245245
-> {
246-
val symbolName = generator.symbolProvider.toSymbol(currShape).name
246+
val symbol = generator.symbolProvider.toSymbol(currShape)
247247
val symbolMember = when (node.value) {
248248
"Infinity" -> "POSITIVE_INFINITY"
249249
"-Infinity" -> "NEGATIVE_INFINITY"
250250
"NaN" -> "NaN"
251-
else -> throw CodegenException("""Cannot interpret $symbolName value "${node.value}".""")
251+
else -> throw CodegenException("""Cannot interpret $symbol value "${node.value}".""")
252252
}
253-
writer.writeInline("#L", "$symbolName.$symbolMember")
253+
writer.writeInline("#T.#L", symbol, symbolMember)
254254
}
255255

256256
ShapeType.BIG_INTEGER -> writer.writeInline("#T(#S)", RuntimeTypes.Core.Content.BigInteger, node.value)

‎codegen/smithy-kotlin-codegen/src/main/kotlin/software/amazon/smithy/kotlin/codegen/rendering/smoketests/SmokeTestsRunnerGenerator.kt‎

Lines changed: 12 additions & 80 deletions
Original file line numberDiff line numberDiff line change
@@ -6,18 +6,16 @@ import software.amazon.smithy.kotlin.codegen.integration.SectionId
66
import software.amazon.smithy.kotlin.codegen.integration.SectionKey
77
import software.amazon.smithy.kotlin.codegen.model.getTrait
88
import software.amazon.smithy.kotlin.codegen.model.hasTrait
9-
import software.amazon.smithy.kotlin.codegen.model.isStringEnumShape
9+
import software.amazon.smithy.kotlin.codegen.rendering.ShapeValueGenerator
1010
import software.amazon.smithy.kotlin.codegen.rendering.endpoints.EndpointParametersGenerator
1111
import software.amazon.smithy.kotlin.codegen.rendering.endpoints.EndpointProviderGenerator
12-
import software.amazon.smithy.kotlin.codegen.rendering.protocol.stringToNumber
1312
import software.amazon.smithy.kotlin.codegen.rendering.smoketests.SmokeTestSectionIds.ClientConfig.EndpointParams
1413
import software.amazon.smithy.kotlin.codegen.rendering.smoketests.SmokeTestSectionIds.ClientConfig.EndpointProvider
1514
import software.amazon.smithy.kotlin.codegen.rendering.smoketests.SmokeTestSectionIds.ClientConfig.Name
1615
import software.amazon.smithy.kotlin.codegen.rendering.smoketests.SmokeTestSectionIds.ClientConfig.Value
1716
import software.amazon.smithy.kotlin.codegen.rendering.util.format
1817
import software.amazon.smithy.kotlin.codegen.utils.dq
1918
import software.amazon.smithy.kotlin.codegen.utils.toCamelCase
20-
import software.amazon.smithy.kotlin.codegen.utils.toPascalCase
2119
import software.amazon.smithy.kotlin.codegen.utils.topDownOperations
2220
import software.amazon.smithy.model.node.*
2321
import software.amazon.smithy.model.shapes.*
@@ -144,7 +142,7 @@ class SmokeTestsRunnerGenerator(
144142
}
145143

146144
private fun renderClient(testCase: SmokeTestCase) {
147-
writer.withInlineBlock("#L {", "}", service) {
145+
writer.withInlineBlock("#T {", "}", service) {
148146
renderClientConfig(testCase)
149147
}
150148
}
@@ -180,27 +178,19 @@ class SmokeTestsRunnerGenerator(
180178
}
181179

182180
private fun renderOperation(operation: OperationShape, testCase: SmokeTestCase) {
183-
val operationSymbol = symbolProvider.toSymbol(model.getShape(operation.input.get()).get())
184-
185181
writer.withBlock(".#T { client ->", "}", RuntimeTypes.Core.IO.use) {
186-
withBlock("client.#L(", ")", operation.defaultName()) {
187-
withBlock("#L {", "}", operationSymbol) {
188-
renderOperationParameters(operation, testCase)
189-
}
190-
}
191-
}
192-
}
193-
194-
private fun renderOperationParameters(operation: OperationShape, testCase: SmokeTestCase) {
195-
if (!testCase.hasOperationParameters) return
182+
val inputParams = testCase.params.getOrNull()
196183

197-
val paramsToShapes = mapOperationParametersToModeledShapes(operation)
184+
writeInline("client.#L", operation.defaultName())
198185

199-
testCase.operationParameters.forEach { param ->
200-
val paramName = param.key.value.toCamelCase()
201-
writer.writeInline("#L = ", paramName)
202-
val paramShape = paramsToShapes[paramName] ?: throw IllegalArgumentException("Unable to find shape for operation parameter '$paramName' in smoke test '${testCase.functionName}'.")
203-
renderOperationParameter(paramName, param.value, paramShape, testCase)
186+
if (inputParams == null) {
187+
write("()")
188+
} else {
189+
withBlock("(", ")") {
190+
val inputShape = model.expectShape(operation.input.get())
191+
ShapeValueGenerator(model, symbolProvider).instantiateShapeInline(writer, inputShape, inputParams)
192+
}
193+
}
204194
}
205195
}
206196

@@ -228,56 +218,6 @@ class SmokeTestsRunnerGenerator(
228218
}
229219

230220
// Helpers
231-
/**
232-
* Renders a [SmokeTestCase] operation parameter
233-
*/
234-
private fun renderOperationParameter(
235-
paramName: String,
236-
node: Node,
237-
shape: Shape,
238-
testCase: SmokeTestCase,
239-
) {
240-
when {
241-
// String enum
242-
node is StringNode && shape.isStringEnumShape -> {
243-
val enumSymbol = symbolProvider.toSymbol(shape)
244-
val enumValue = node.value.toPascalCase()
245-
writer.write("#T.#L", enumSymbol, enumValue)
246-
}
247-
// Int enum
248-
node is NumberNode && shape is IntEnumShape -> {
249-
val enumSymbol = symbolProvider.toSymbol(shape)
250-
val enumValue = node.format()
251-
writer.write("#T.fromValue(#L.toInt())", enumSymbol, enumValue)
252-
}
253-
// Number
254-
node is NumberNode && shape is NumberShape -> writer.write("#L.#L", node.format(), stringToNumber(shape))
255-
// Object
256-
node is ObjectNode -> {
257-
val shapeSymbol = symbolProvider.toSymbol(shape)
258-
writer.withBlock("#T {", "}", shapeSymbol) {
259-
node.members.forEach { member ->
260-
val memberName = member.key.value.toCamelCase()
261-
val memberShape = shape.allMembers[member.key.value] ?: throw IllegalArgumentException("Unable to find shape for operation parameter '$paramName' in smoke test '${testCase.functionName}'.")
262-
writer.writeInline("#L = ", memberName)
263-
renderOperationParameter(memberName, member.value, memberShape, testCase)
264-
}
265-
}
266-
}
267-
// List
268-
node is ArrayNode && shape is CollectionShape -> {
269-
writer.withBlock("listOf(", ")") {
270-
node.elements.forEach { element ->
271-
renderOperationParameter(paramName, element, model.expectShape(shape.member.target), testCase)
272-
writer.write(",")
273-
}
274-
}
275-
}
276-
// Everything else
277-
else -> writer.write("#L", node.format())
278-
}
279-
}
280-
281221
/**
282222
* Tries to get the specific exception required in the failure criterion of a test.
283223
* If no specific exception is required we default to the generic smoke tests failure exception.
@@ -304,14 +244,6 @@ class SmokeTestsRunnerGenerator(
304244
writer.write("println(#S)", testResult)
305245
}
306246

307-
/**
308-
* Maps an operations parameters to their shapes
309-
*/
310-
private fun mapOperationParametersToModeledShapes(operation: OperationShape): Map<String, Shape> =
311-
model.getShape(operation.inputShape).get().allMembers.map { (key, value) ->
312-
key.toCamelCase() to model.getShape(value.target).get()
313-
}.toMap()
314-
315247
/**
316248
* Derives a function name for a [SmokeTestCase]
317249
*/

‎codegen/smithy-kotlin-codegen/src/main/resources/META-INF/services/software.amazon.smithy.kotlin.codegen.integration.KotlinIntegration‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,4 +12,4 @@ software.amazon.smithy.kotlin.codegen.rendering.endpoints.discovery.EndpointDisc
1212
software.amazon.smithy.kotlin.codegen.rendering.endpoints.SdkEndpointBuiltinIntegration
1313
software.amazon.smithy.kotlin.codegen.rendering.compression.RequestCompressionIntegration
1414
software.amazon.smithy.kotlin.codegen.rendering.auth.SigV4AsymmetricAuthSchemeIntegration
15-
# software.amazon.smithy.kotlin.codegen.rendering.smoketests.SmokeTestsIntegration
15+
software.amazon.smithy.kotlin.codegen.rendering.smoketests.SmokeTestsIntegration

‎codegen/smithy-kotlin-codegen/src/test/kotlin/software/amazon/smithy/kotlin/codegen/rendering/smoketests/SmokeTestsRunnerGeneratorTest.kt‎

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -106,12 +106,12 @@ class SmokeTestsRunnerGeneratorTest {
106106
}
107107
108108
try {
109-
com.test.TestClient {
109+
TestClient {
110110
interceptors.add(SmokeTestsInterceptor())
111111
region = "eu-central-1"
112112
}.use { client ->
113113
client.testOperation(
114-
com.test.model.TestOperationRequest {
114+
TestOperationRequest {
115115
bar = "2"
116116
}
117117
)
@@ -143,11 +143,11 @@ class SmokeTestsRunnerGeneratorTest {
143143
}
144144
145145
try {
146-
com.test.TestClient {
146+
TestClient {
147147
148148
}.use { client ->
149149
client.testOperation(
150-
com.test.model.TestOperationRequest {
150+
TestOperationRequest {
151151
bar = "föö"
152152
}
153153
)
@@ -179,12 +179,12 @@ class SmokeTestsRunnerGeneratorTest {
179179
}
180180
181181
try {
182-
com.test.TestClient {
182+
TestClient {
183183
interceptors.add(SmokeTestsInterceptor())
184184
185185
}.use { client ->
186186
client.testOperation(
187-
com.test.model.TestOperationRequest {
187+
TestOperationRequest {
188188
bar = "föö"
189189
}
190190
)

0 commit comments

Comments
 (0)