fix: pass grpc metadata to helpers. #189

testinggospels · Sep 2, 2022 · 3ff8421 · 3ff8421
1 parent 0cdcc8e
commit 3ff8421
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/src/handlebar/RequestHelper.ts b/src/handlebar/RequestHelper.ts
@@ -79,6 +79,8 @@ export class RequestHelper {
                 return null;
             }
           }
+        case "metadata":
+          return context.data.root.metadata.get(context.hash.key);
         default:
           if (
             typeof context.hash.using === "undefined" ||

diff --git a/src/parser/GrpcParser.ts b/src/parser/GrpcParser.ts
@@ -35,8 +35,8 @@ export default class GrpcParser {
       const mockFilePath = path.join(this.config.protocols.grpc.mocks_dir, mockFile + ".mock");
       if (fs.existsSync(mockFilePath)) {
         const template = Handlebars.compile(fs.readFileSync(mockFilePath, "utf-8").toString());
-        logger.debug(`Unary Request: ${JSON.stringify(call.request)}`)
-        const fileContent = await template({ request: call.request });
+        logger.debug(`Unary Request: ${JSON.stringify(call.request)}. Metadata: ${JSON.stringify(call.metadata)}`)
+        const fileContent = await template({ request: call.request, metadata: call.metadata });
         logger.debug(`Mock file path: ${mockFilePath}`);
         logger.debug(`Response: ${fileContent}`);
         const response = JSON.parse(fileContent);
@@ -72,8 +72,8 @@ export default class GrpcParser {
     if (fs.existsSync(mockFilePath)) {
       try {
         const template = Handlebars.compile(fs.readFileSync(mockFilePath, "utf-8").toString());
-        logger.debug(`Server Stream Request: ${JSON.stringify(call.request)}`)
-        const fileContent = await template({ request: call.request });
+        logger.debug(`Server Stream Request: ${JSON.stringify(call.request)}. Metadata ${JSON.stringify(call.metadata)}`)
+        const fileContent = await template({ request: call.request, metadata: call.metadata });
         logger.debug(`Mock file path: ${mockFilePath}`);
         const streamArr = fileContent.split("====");
         let delay = 0;