Fix generation of aws-config related code in tests

smithy-lang · Dec 13, 2023 · d4db2be · d4db2be
1 parent c704d15
commit d4db2be
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/aws/sdk-codegen/src/main/kotlin/software/amazon/smithy/rustsdk/AwsDocs.kt b/aws/sdk-codegen/src/main/kotlin/software/amazon/smithy/rustsdk/AwsDocs.kt
@@ -19,7 +19,7 @@ object AwsDocs {
      * Also, STS and SSO must NOT reference `aws-config` since that would create a circular dependency.
      */
     fun canRelyOnAwsConfig(codegenContext: ClientCodegenContext): Boolean =
-        SdkSettings.from(codegenContext.settings).awsConfigVersion != null &&
+        SdkSettings.from(codegenContext.settings).generateReadme &&
             !setOf(
                 ShapeId.from("com.amazonaws.sso#SWBPortalService"),
                 ShapeId.from("com.amazonaws.ssooidc#AWSSSOOIDCService"),

diff --git a/aws/sdk-codegen/src/main/kotlin/software/amazon/smithy/rustsdk/SdkSettings.kt b/aws/sdk-codegen/src/main/kotlin/software/amazon/smithy/rustsdk/SdkSettings.kt
@@ -45,7 +45,7 @@ class SdkSettings private constructor(private val awsSdk: ObjectNode?) {
 
     /** Whether to generate a README */
     val generateReadme: Boolean
-        get() = awsSdkBuild
+        get() = awsSdkBuild && !(awsSdk?.getBooleanMember("suppressReadme")?.orNull()?.value ?: false)
 
     val requireEndpointResolver: Boolean
         get() = awsSdkBuild

diff --git a/aws/sdk-codegen/src/test/kotlin/software/amazon/smithy/rustsdk/TestUtil.kt b/aws/sdk-codegen/src/test/kotlin/software/amazon/smithy/rustsdk/TestUtil.kt
@@ -56,6 +56,7 @@ fun awsIntegrationTestParams() = IntegrationTestParams(
                 "awsSdk",
                 ObjectNode.builder()
                     .withMember("awsSdkBuild", true)
+                    .withMember("suppressReadme", true)
                     .withMember("integrationTestPath", "../sdk/integration-tests")
                     .withMember("partitionsConfigPath", "../sdk/aws-models/sdk-endpoints.json")
                     .withMember("awsConfigVersion", "1")