Generate ssdk error protocol tests

Author: JordonPhillips

smithy-typescript-codegen/src/main/java/software/amazon/smithy/typescript/codegen/HttpProtocolTestGenerator.java

@@ -183,6 +183,17 @@ private void generateServerOperationTests(OperationShape operation, OperationInd
                     onlyIfProtocolMatches(testCase, () -> generateServerResponseTest(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.getTestCasesFor(AppliesTo.SERVER)) {
+                            onlyIfProtocolMatches(testCase,
+                                    () -> generateServerErrorResponseTest(operation, error, testCase));
+                        }
+                    });
+                }
+            }
         }
     }
 
@@ -288,8 +299,7 @@ private void generateServerRequestTest(OperationShape operation, HttpRequestTest
             });
 
             String getHandlerName = "get" + handlerSymbol.getName();
-            writer.addImport(getHandlerName, getHandlerName,
-                    "./protocols/" + ProtocolGenerator.getSanitizedName(protocolGenerator.getName()));
+            writer.addImport(getHandlerName, null, "./server/");
 
             // Cast the service as any so TS will ignore the fact that the type being passed in is incomplete.
             writer.write("const handler = $L(testService as $T);", getHandlerName, serviceSymbol);
@@ -477,8 +487,6 @@ private String registerBodyComparatorStub(String mediaType) {
     public void generateServerResponseTest(OperationShape operation, HttpResponseTestCase testCase) {
         Symbol serviceSymbol = serverSymbolProvider.toSymbol(service);
         Symbol operationSymbol = serverSymbolProvider.toSymbol(operation);
-        Symbol handlerSymbol = serviceSymbol.expectProperty("handler", Symbol.class);
-        Symbol serviceOperationsSymbol = serviceSymbol.expectProperty("operations", Symbol.class);
         testCase.getDocumentation().ifPresent(writer::writeDocs);
         String testName = testCase.getId() + ":ServerResponse";
         writer.openBlock("it($S, async () => {", "});\n", testName, () -> {
@@ -497,42 +505,7 @@ public void generateServerResponseTest(OperationShape operation, HttpResponseTes
                     }
                 });
             });
-
-            writer.write("const service: any = new TestService()");
-
-            // There's a lot of setup here, including creating our own mux, serializers list, and ultimately
-            // our own service handler. This is largely in service of avoiding having to go through the
-            // request deserializer
-            writer.addImport("httpbinding", null, "@aws-smithy/server-common");
-            writer.openBlock("const testMux = new httpbinding.HttpBindingMux<$S, keyof $T>([", "]);",
-                service.getId().getName(), serviceSymbol, () -> {
-                    writer.openBlock("new httpbinding.UriSpec<$S, $S>('POST', [], [], {", "}),",
-                        service.getId().getName(), operation.getId().getName(), () -> {
-                            writer.write("service: $S,", service.getId().getName());
-                            writer.write("operation: $S,", operation.getId().getName());
-                        });
-                });
-
-            writer.write("const request = new HttpRequest({method: 'POST', hostname: 'example.com'});");
-
-            String serializerName = ProtocolGenerator.getGenericSerFunctionName(operationSymbol) + "Response";
-            writer.addImport(serializerName, serializerName,
-                    "./protocols/" + ProtocolGenerator.getSanitizedName(protocolGenerator.getName()));
-
-            writer.addImport("OperationSerializer", "__OperationSerializer", "@aws-smithy/server-common");
-            writer.openBlock("const serFn: (op: $1T) => __OperationSerializer<$2T, $1T> = (op) => {", "};",
-                    serviceOperationsSymbol, serviceSymbol, () -> {
-                writer.openBlock("return {", "};", () -> {
-                    writer.write("serialize: $L,", serializerName);
-                    writer.openBlock("deserialize: (output: any, context: any): Promise<any> => {", "},", () -> {
-                        writer.write("return Promise.resolve({});");
-                    });
-                });
-            });
-
-            writer.write("const handler = new $T(service, testMux, serFn);", handlerSymbol);
-            writer.write("let r = await handler.handle(request)").write("");
-            writeHttpResponseAssertions(testCase);
+            writeServerResponseTest(operation, testCase);
         });
     }
 
@@ -554,6 +527,76 @@ private void generateResponseTest(OperationShape operation, HttpResponseTestCase
         });
     }
 
+    private void generateServerErrorResponseTest(
+            OperationShape operation,
+            StructureShape error,
+            HttpResponseTestCase testCase
+    ) {
+        Symbol serviceSymbol = serverSymbolProvider.toSymbol(service);
+        Symbol operationSymbol = serverSymbolProvider.toSymbol(operation);
+        Symbol outputType = operationSymbol.expectProperty("outputType", Symbol.class);
+        Symbol errorSymbol = serverSymbolProvider.toSymbol(error);
+        ErrorTrait errorTrait = error.expectTrait(ErrorTrait.class);
+        testCase.getDocumentation().ifPresent(writer::writeDocs);
+        String testName = testCase.getId() + ":ServerErrorResponse";
+        writer.openBlock("it($S, async () => {", "});\n", testName, () -> {
+            writer.openBlock("class TestService implements Partial<$T> {", "}", serviceSymbol, () -> {
+                writer.openBlock("$L(input: any, request: HttpRequest): $T {", "}",
+                    operationSymbol.getName(), outputType, () -> {
+                        writer.writeInline("const response = ");
+                        testCase.getParams().accept(new CommandInputNodeVisitor(error, true));
+                        writer.openBlock("const error: $T = {", "};", errorSymbol, () -> {
+                            writer.write("...response,");
+                            writer.write("name: $S,", error.getId().getName());
+                            writer.write("$$fault: $S,", errorTrait.isClientError() ? "client" : "server");
+                            writer.write("$$metadata: {},");
+                        });
+                        writer.write("throw error;");
+                    });
+            });
+            writeServerResponseTest(operation, testCase);
+        });
+    }
+
+    private void writeServerResponseTest(OperationShape operation, HttpResponseTestCase testCase) {
+        Symbol serviceSymbol = serverSymbolProvider.toSymbol(service);
+        Symbol operationSymbol = serverSymbolProvider.toSymbol(operation);
+        Symbol handlerSymbol = serviceSymbol.expectProperty("handler", Symbol.class);
+        Symbol serializerSymbol = operationSymbol.expectProperty("serializerType", Symbol.class);
+        Symbol serviceOperationsSymbol = serviceSymbol.expectProperty("operations", Symbol.class);
+        writer.write("const service: any = new TestService()");
+
+        // There's a lot of setup here, including creating our own mux, serializers list, and ultimately
+        // our own service handler. This is largely in service of avoiding having to go through the
+        // request deserializer
+        writer.addImport("httpbinding", null, "@aws-smithy/server-common");
+        writer.openBlock("const testMux = new httpbinding.HttpBindingMux<$S, keyof $T>([", "]);",
+            service.getId().getName(), serviceSymbol, () -> {
+                writer.openBlock("new httpbinding.UriSpec<$S, $S>('POST', [], [], {", "}),",
+                    service.getId().getName(), operation.getId().getName(), () -> {
+                        writer.write("service: $S,", service.getId().getName());
+                        writer.write("operation: $S,", operation.getId().getName());
+                    });
+            });
+
+        writer.write("const request = new HttpRequest({method: 'POST', hostname: 'example.com'});");
+
+        writer.openBlock("class TestSerializer extends $T {", "}", serializerSymbol, () -> {
+            writer.openBlock("deserialize = (output: any, context: any): Promise<any> => {", "};", () -> {
+                writer.write("return Promise.resolve({});");
+            });
+        });
+
+        writer.addImport("SmithyException", "__SmithyException", "@aws-sdk/smithy-client");
+        writer.addImport("OperationSerializer", "__OperationSerializer", "@aws-smithy/server-common");
+        writer.openBlock("const serFn: (op: $1T) => __OperationSerializer<$2T, $1T, __SmithyException> = (op) =>"
+                        + " { return new TestSerializer(); };", serviceOperationsSymbol, serviceSymbol);
+
+        writer.write("const handler = new $T(service, testMux, serFn);", handlerSymbol);
+        writer.write("let r = await handler.handle(request)").write("");
+        writeHttpResponseAssertions(testCase);
+    }
+
     private void generateErrorResponseTest(
             OperationShape operation,
             StructureShape error,

smithy-typescript-codegen/src/main/java/software/amazon/smithy/typescript/codegen/ServerCommandGenerator.java

@@ -182,18 +182,18 @@ private void writeErrorChecker() {
     private void writeErrorHandler() {
         writer.addImport("SerdeContext", null, "@aws-sdk/types");
         writer.openBlock("serializeError(error: $T, ctx: Omit<SerdeContext, 'endpoint'>): Promise<$T> {", "}",
-            errorsType, applicationProtocol.getResponseType(), () -> {
-                if (operation.getErrors().isEmpty()) {
-                    writer.write("throw error;");
-                } else {
-                    writer.openBlock("switch (error.name) {", "}", () -> {
-                        for (ShapeId errorId : operation.getErrors()) {
-                            writeErrorHandlerCase(errorId);
-                        }
-                        writer.openBlock("default: {", "}", () -> writer.write("throw error;"));
-                    });
-                }
-            });
+                errorsType, applicationProtocol.getResponseType(), () -> {
+            if (operation.getErrors().isEmpty()) {
+                writer.write("throw error;");
+            } else {
+                writer.openBlock("switch (error.name) {", "}", () -> {
+                    for (ShapeId errorId : operation.getErrors()) {
+                        writeErrorHandlerCase(errorId);
+                    }
+                    writer.openBlock("default: {", "}", () -> writer.write("throw error;"));
+                });
+            }
+        });
         writer.write("");
     }