Add tests for Operation Context Params Identifier and SubExpression

Author: trivikr

smithy-typescript-codegen/src/test/java/software/amazon/smithy/typescript/codegen/CommandGeneratorTest.java

@@ -14,46 +14,60 @@ public class CommandGeneratorTest {
     public void addsCommandSpecificPlugins() {
         testCommmandCodegen(
             "output-structure.smithy",
-            "getSerdePlugin(config, this.serialize, this.deserialize)"
+            new String[] {"getSerdePlugin(config, this.serialize, this.deserialize)"}
         );
     }
 
     @Test
     public void writesSerializer() {
         testCommmandCodegen(
             "output-structure.smithy",
-            ".ser("
+            new String[] {".ser("}
         );
     }
 
     @Test
     public void writesDeserializer() {
         testCommmandCodegen(
             "output-structure.smithy",
-        ".de("
+            new String[] {".de("}
         );
     }
 
-    private void testCommmandCodegen(String file, String expectedType) {
-        Model model = Model.assembler()
-                .addImport(getClass().getResource(file))
-                .assemble()
-                .unwrap();
+    @Test
+    public void writesOperationContextParamValues() {
+        testCommmandCodegen(
+            "endpointsV2/endpoints-operation-context-params.smithy",
+            new String[] {
+                "opContextParamIdentifier: { type: \"operationContextParams\", name: this.input.fooString }",
+                "opContextParamSubExpression: { type: \"operationContextParams\", name: this.input.fooObj.bar }",
+            }
+        );
+    }
+
+    private void testCommmandCodegen(String filename, String[] expectedTypeArray) {
         MockManifest manifest = new MockManifest();
         PluginContext context = PluginContext.builder()
-                .model(model)
-                .fileManifest(manifest)
-                .settings(Node.objectNodeBuilder()
-                                  .withMember("service", Node.from("smithy.example#Example"))
-                                  .withMember("package", Node.from("example"))
-                                  .withMember("packageVersion", Node.from("1.0.0"))
-                                  .build())
-                .build();
+            .pluginClassLoader(getClass().getClassLoader())
+            .model(Model.assembler()
+                .addImport(getClass().getResource(filename))
+                .discoverModels()
+                .assemble()
+                .unwrap())
+            .fileManifest(manifest)
+            .settings(Node.objectNodeBuilder()
+                .withMember("service", Node.from("smithy.example#Example"))
+                .withMember("package", Node.from("example"))
+                .withMember("packageVersion", Node.from("1.0.0"))
+                .build())
+            .build();
 
         new TypeScriptCodegenPlugin().execute(context);
         String contents = manifest.getFileString(CodegenUtils.SOURCE_FOLDER + "//commands/GetFooCommand.ts").get();
 
         assertThat(contents, containsString("as __MetadataBearer"));
-        assertThat(contents, containsString(expectedType));
+        for (String expectedType : expectedTypeArray) {
+            assertThat(contents, containsString(expectedType));
+        }
     }
 }

smithy-typescript-codegen/src/test/resources/software/amazon/smithy/typescript/codegen/endpointsV2/endpoints-operation-context-params.smithy

@@ -0,0 +1,44 @@
+$version: "2.0"
+
+namespace smithy.example
+
+@smithy.rules#endpointRuleSet({
+    version: "1.0",
+    parameters: {
+        opContextParamIdentifier: {
+            type: "string",
+        },
+        opContextParamSubExpression: {
+            type: "string",
+        },
+    },
+    rules: []
+})
+service Example {
+    version: "1.0.0",
+    operations: [GetFoo]
+}
+
+@smithy.rules#operationContextParams(
+    "opContextParamIdentifier": { path: "fooString" }
+    "opContextParamSubExpression": { path: "fooObj.bar" }
+)
+operation GetFoo {
+    input: GetFooInput,
+    output: GetFooOutput,
+    errors: [GetFooError]
+}
+
+structure GetFooInput {
+    fooString: String,
+    fooObj: FooObject
+}
+
+structure FooObject {
+    bar: String
+}
+
+structure GetFooOutput {}
+
+@error("client")
+structure GetFooError {}