From 8819a05e421c046746baf146130e29928420e806 Mon Sep 17 00:00:00 2001
From: Steven Yuan <yuasteve@amazon.com>
Date: Mon, 29 Jan 2024 14:41:48 -0800
Subject: [PATCH] Add Go Server Codegen test

---
 codegen/smithy-go-codegen-test/build.gradle.kts  | 1 +
 codegen/smithy-go-codegen-test/model/main.smithy | 1 +
 codegen/smithy-go-codegen-test/smithy-build.json | 7 +++++++
 3 files changed, 9 insertions(+)

diff --git a/codegen/smithy-go-codegen-test/build.gradle.kts b/codegen/smithy-go-codegen-test/build.gradle.kts
index 5b8a4f29b..319499c67 100644
--- a/codegen/smithy-go-codegen-test/build.gradle.kts
+++ b/codegen/smithy-go-codegen-test/build.gradle.kts
@@ -43,5 +43,6 @@ repositories {
 
 dependencies {
     implementation("software.amazon.smithy:smithy-protocol-test-traits:$smithyVersion")
+    implementation("software.amazon.smithy:smithy-aws-traits:$smithyVersion")
     implementation(project(":smithy-go-codegen"))
 }
diff --git a/codegen/smithy-go-codegen-test/model/main.smithy b/codegen/smithy-go-codegen-test/model/main.smithy
index d41e76f8b..5bfeb1c7a 100644
--- a/codegen/smithy-go-codegen-test/model/main.smithy
+++ b/codegen/smithy-go-codegen-test/model/main.smithy
@@ -8,6 +8,7 @@ use smithy.waiters#waitable
 /// Provides weather forecasts.
 @httpBearerAuth
 @fakeProtocol
+@aws.protocols#awsJson1_0
 @paginated(inputToken: "nextToken", outputToken: "nextToken", pageSize: "pageSize")
 service Weather {
     version: "2006-03-01",
diff --git a/codegen/smithy-go-codegen-test/smithy-build.json b/codegen/smithy-go-codegen-test/smithy-build.json
index b495c9292..9f7bd5308 100644
--- a/codegen/smithy-go-codegen-test/smithy-build.json
+++ b/codegen/smithy-go-codegen-test/smithy-build.json
@@ -7,6 +7,13 @@
             "moduleVersion": "0.0.1",
             "generateGoMod": true,
             "goDirective": "1.18"
+        },
+        "go-server-codegen": {
+            "service": "example.weather#Weather",
+            "module": "github.com/aws/smithy-go/internal/tests/service/weather",
+            "moduleVersion": "0.0.1",
+            "generateGoMod": true,
+            "goDirective": "1.18"
         }
     }
 }