added .on metadata functions to client unary

firecomm · Aug 13, 2019 · c224c10 · c224c10
1 parent 45ccc12
commit c224c10
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/__tests__/callFactories/callDecorators/clientUnaryDecorator.js b/__tests__/callFactories/callDecorators/clientUnaryDecorator.js
@@ -33,6 +33,12 @@ describe("tests for client unary decorator", () => {
       mockOn.mockClear();
       mockCallback.mockClear();
     });
+
+    it("if passed metadata as a string, calls callback on the head", () => {
+      customCall.on("metadata", mockCallback);
+      expect(mockCallback.mock.calls.length).toBe(1);
+    });
+
     it("on should accept a data string and callback", () => {
       customCall.on("data", mockCallback);
       expect(mockCallback.mock.calls.length).toBe(1);

diff --git a/lib/callFactories/callDecorators/clientUnaryDecorator.js b/lib/callFactories/callDecorators/clientUnaryDecorator.js
@@ -23,6 +23,8 @@ module.exports = function clientUnaryDecorator(customCall, originalCall) {
       } else {
         if (string === "data") {
           callback(this.body);
+        } else if (string === "metadata") {
+          callback(this.head);
         } else {
           throw new Error('Only the "data" event is support by unary .on');
         }