Merge branch 'fahadzub/member-constraints' of https://github.com/awslabs/smithy-rs into fahadzub/member-constraints

Author: Fahad Zubair

codegen-server/src/test/kotlin/software/amazon/smithy/rust/codegen/server/smithy/ConstraintsMemberShapeTest.kt

@@ -6,18 +6,21 @@ import software.amazon.smithy.model.shapes.ShapeId
 import software.amazon.smithy.model.traits.RequiredTrait
 import software.amazon.smithy.rust.codegen.core.rustlang.RustWriter
 import software.amazon.smithy.rust.codegen.core.rustlang.Writable
+import software.amazon.smithy.rust.codegen.core.smithy.DirectedWalker
 import software.amazon.smithy.rust.codegen.core.testutil.asSmithyModel
 import software.amazon.smithy.rust.codegen.server.smithy.transformers.ConstrainedMemberTransform
 import software.amazon.smithy.rust.codegen.core.smithy.RuntimeConfig
 import software.amazon.smithy.rust.codegen.core.smithy.RuntimeCrateLocation
 import software.amazon.smithy.rust.codegen.core.smithy.RustCrate
 import software.amazon.smithy.rust.codegen.core.smithy.transformers.OperationNormalizer
+import software.amazon.smithy.rust.codegen.core.testutil.TestWorkspace
 import software.amazon.smithy.rust.codegen.core.testutil.generatePluginContext
 import software.amazon.smithy.rust.codegen.core.testutil.unitTest
 import software.amazon.smithy.rust.codegen.core.util.runCommand
 import software.amazon.smithy.rust.codegen.core.util.toPascalCase
 import software.amazon.smithy.rust.codegen.server.smithy.customize.CombinedServerCodegenDecorator
 import software.amazon.smithy.rust.codegen.server.smithy.testutil.serverTestCodegenContext
+import software.amazon.smithy.rust.codegen.server.smithy.testutil.serverTestSymbolProvider
 import java.io.File
 import java.nio.file.Path
 
@@ -267,14 +270,29 @@ class ConstraintsMemberShapeTest {
         )
         val codegenDecorator: CombinedServerCodegenDecorator =
             CombinedServerCodegenDecorator.fromClasspath(context)
-        ServerCodegenVisitor(context, codegenDecorator)
-            .execute()
 
+        val transformed = model
+            .let(OperationNormalizer::transform)
+            .let(ConstrainedMemberTransform::transform)
+
+        val testProject = TestWorkspace.testProject(serverTestSymbolProvider(transformed))
         val codegenContext = serverTestCodegenContext(model)
+
+        testProject.initializeInlineModuleWriter(codegenContext)
+        val settings = ServerRustSettings.from(context.model, context.settings)
+        val service = settings.getService(model)
+        val serviceShapes = DirectedWalker(model).walkShapes(service)
+
+        serviceShapes.forEach { it.accept(this) }
+        codegenDecorator.extras(codegenContext, rustCrate)
+
+        rustCrate.getInlineModuleWriter().render()
+
+
         RustCrate(
             context.fileManifest,
             codegenContext.symbolProvider,
-            ServerRustSettings.from(context.model, context.settings).codegenConfig
+            ServerRustSettings.from(context.model, context.settings).codegenConfig,
         )
             .lib { writable(this) }