Unverified Commit 7a307666 authored by 82marbag's avatar 82marbag Committed by GitHub
Browse files

Test decorators in server tests (#3107) 



This change allows to test decorators by adding them to the
CodegenContext.

----

_By submitting this pull request, I confirm that you can use, modify,
copy, and redistribute this contribution, under the terms of your
choice._

---------

Signed-off-by: default avatarDaniele Ahmed <ahmeddan@amazon.de>
parent a9b9cbab

codegen-server/src/main/kotlin/software/amazon/smithy/rust/codegen/server/smithy/testutil/ServerTestHelpers.kt

+5 −7
Original line number Diff line number Diff line
@@ -28,6 +28,7 @@ import software.amazon.smithy.rust.codegen.server.smithy.ServerModuleProvider 
import software.amazon.smithy.rust.codegen.server.smithy.ServerRustSettings

import software.amazon.smithy.rust.codegen.server.smithy.ServerSymbolProviders

import software.amazon.smithy.rust.codegen.server.smithy.customizations.SmithyValidationExceptionConversionGenerator

import software.amazon.smithy.rust.codegen.server.smithy.customize.CombinedServerCodegenDecorator

import software.amazon.smithy.rust.codegen.server.smithy.customize.ServerCodegenDecorator

import software.amazon.smithy.rust.codegen.server.smithy.generators.ServerBuilderGenerator

import software.amazon.smithy.rust.codegen.server.smithy.generators.protocol.ServerProtocol
@@ -47,15 +48,11 @@ private fun testServiceShapeFor(model: Model) = 
fun serverTestSymbolProvider(model: Model, serviceShape: ServiceShape? = null) =

    serverTestSymbolProviders(model, serviceShape).symbolProvider



private class ServerTestCodegenDecorator : ServerCodegenDecorator {

    override val name = "test"

    override val order: Byte = 0

}



fun serverTestSymbolProviders(

    model: Model,

    serviceShape: ServiceShape? = null,

    settings: ServerRustSettings? = null,

    decorators: List<ServerCodegenDecorator> = emptyList(),

) =

    ServerSymbolProviders.from(

        serverTestRustSettings(),
@@ -67,7 +64,7 @@ fun serverTestSymbolProviders( 
                (serviceShape ?: testServiceShapeFor(model)).id,

            )

            ).codegenConfig.publicConstrainedTypes,

        ServerTestCodegenDecorator(),

        CombinedServerCodegenDecorator(decorators),

        RustServerCodegenPlugin::baseSymbolProvider,

    )
@@ -102,6 +99,7 @@ fun serverTestCodegenContext( 
    serviceShape: ServiceShape? = null,

    settings: ServerRustSettings = serverTestRustSettings(),

    protocolShapeId: ShapeId? = null,

    decorators: List<ServerCodegenDecorator> = emptyList(),

): ServerCodegenContext {

    val service = serviceShape ?: testServiceShapeFor(model)

    val protocol = protocolShapeId ?: ShapeId.from("test#Protocol")
@@ -111,7 +109,7 @@ fun serverTestCodegenContext( 
        service,

        ServerTestRustSymbolProviderConfig,

        settings.codegenConfig.publicConstrainedTypes,

        ServerTestCodegenDecorator(),

        CombinedServerCodegenDecorator(decorators),

        RustServerCodegenPlugin::baseSymbolProvider,

    )