Update protocol test case generation (#275)

smithy-lang · Feb 23, 2021 · 62f7953 · 62f7953
1 parent 8ea5676
commit 62f7953
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/...en/src/main/java/software/amazon/smithy/typescript/codegen/HttpProtocolTestGenerator.java b/...en/src/main/java/software/amazon/smithy/typescript/codegen/HttpProtocolTestGenerator.java
@@ -52,6 +52,7 @@
 import software.amazon.smithy.model.traits.HttpPrefixHeadersTrait;
 import software.amazon.smithy.model.traits.IdempotencyTokenTrait;
 import software.amazon.smithy.model.traits.StreamingTrait;
+import software.amazon.smithy.protocoltests.traits.AppliesTo;
 import software.amazon.smithy.protocoltests.traits.HttpMessageTestCase;
 import software.amazon.smithy.protocoltests.traits.HttpRequestTestCase;
 import software.amazon.smithy.protocoltests.traits.HttpRequestTestsTrait;
@@ -118,21 +119,21 @@ public void run() {
             if (!operation.hasTag("server-only")) {
                 // 1. Generate test cases for each request.
                 operation.getTrait(HttpRequestTestsTrait.class).ifPresent(trait -> {
-                    for (HttpRequestTestCase testCase : trait.getTestCases()) {
+                    for (HttpRequestTestCase testCase : trait.getTestCasesFor(AppliesTo.CLIENT)) {
                         onlyIfProtocolMatches(testCase, () -> generateRequestTest(operation, testCase));
                     }
                 });
                 // 2. Generate test cases for each response.
                 operation.getTrait(HttpResponseTestsTrait.class).ifPresent(trait -> {
-                    for (HttpResponseTestCase testCase : trait.getTestCases()) {
+                    for (HttpResponseTestCase testCase : trait.getTestCasesFor(AppliesTo.CLIENT)) {
                         onlyIfProtocolMatches(testCase, () -> generateResponseTest(operation, testCase));
                     }
                 });
                 // 3. Generate test cases for each error on each operation.
                 for (StructureShape error : operationIndex.getErrors(operation)) {
                     if (!error.hasTag("server-only")) {
                         error.getTrait(HttpResponseTestsTrait.class).ifPresent(trait -> {
-                            for (HttpResponseTestCase testCase : trait.getTestCases()) {
+                            for (HttpResponseTestCase testCase : trait.getTestCasesFor(AppliesTo.CLIENT)) {
                                 onlyIfProtocolMatches(testCase,
                                         () -> generateErrorResponseTest(operation, error, testCase));
                             }