feat(experimentalIdentityAndAuth): add endpointRuleSet trait to gen…

…eric client test (#5573)
aws · Dec 12, 2023 · 03d6230 · 03d6230
1 parent cdcbe7a
commit 03d6230
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 0 deletions.
diff --git a/codegen/generic-client-test-codegen/build.gradle.kts b/codegen/generic-client-test-codegen/build.gradle.kts
@@ -37,6 +37,7 @@ plugins {
 dependencies {
     implementation("software.amazon.smithy:smithy-aws-protocol-tests:$smithyVersion")
     implementation("software.amazon.smithy:smithy-aws-traits:$smithyVersion")
+    implementation("software.amazon.smithy:smithy-rules-engine:$smithyVersion")
     implementation(project(":smithy-aws-typescript-codegen"))
 }
 

diff --git a/codegen/generic-client-test-codegen/model/weather.smithy b/codegen/generic-client-test-codegen/model/weather.smithy
@@ -3,6 +3,7 @@ $version: "2.0"
 namespace example.weather
 
 use aws.auth#sigv4
+use aws.api#service
 
 @authDefinition
 @trait
@@ -12,6 +13,7 @@ structure customAuth {}
 @protocolDefinition
 structure fakeProtocol {}
 
+@service(sdkId: "weather")
 @fakeProtocol
 @httpApiKeyAuth(name: "X-Api-Key", in: "header")
 @httpBearerAuth
@@ -86,3 +88,30 @@ operation SameAsService {
     service: String
   }
 }
+
+apply Weather @smithy.rules#endpointRuleSet({
+  "version": "1.3",
+  "parameters": {
+    "Region": {
+      "required": true,
+      "type": "String",
+      "documentation": "docs"
+    }
+  },
+  "rules": [
+    {
+      "conditions": [],
+      "documentation": "base rule",
+      "endpoint": {
+        "url": "https://{Region}.amazonaws.com",
+        "properties": {},
+        "headers": {}
+      },
+      "type": "endpoint"
+    }
+  ]
+})
+
+apply Weather @smithy.rules#clientContextParams(
+  Region: {type: "string", documentation: "docs"}
+)