working on adding tests to decorators

__tests__/callFactories/callDecorators/serverUnaryDecorator.js

@@ -1,11 +1,21 @@
 const object = {};
+
+const mockSendCalback = jest.fn();
+const mockSendMetadata = jest.fn();
+
 const originalCall = {
-  hello: "hello"
+  hello: "hello",
+  sendMetadata: mockSendMetadata
 };
+
 const serverUnaryDecorator = require("../../../lib/callFactories/callDecorators/serverUnaryDecorator");
 
 describe("Server unary decorator tests.", () => {
-  describe("decorator adds the right methods", () => {
+  beforeEach(() => {
+    mockSendCalback.mockClear();
+  });
+
+  xdescribe("decorator adds the right methods", () => {
     //adds properties
     beforeEach(() => {
       serverUnaryDecorator(object, originalCall);
@@ -20,4 +30,57 @@ describe("Server unary decorator tests.", () => {
       expect(object.hasOwnProperty("throw")).toBeTruthy();
     });
   });
+
+  xdescribe("Tests for throw method", () => {
+    it("Expect throw to throw if not type error passed.", () => {
+      serverUnaryDecorator(object, mockSendCalback);
+      const fakeError = "hello";
+      expect(() => {
+        object.throw(fakeError);
+      }).toThrow();
+    });
+
+    it("Throw passes error into sendCallback", () => {
+      serverUnaryDecorator(object, mockSendCalback);
+      const fakeError = new Error("fake error");
+      object.throw(fakeError);
+      expect(mockSendCalback.mock.calls[0][0] instanceof Error).toBeTruthy();
+    });
+  });
+
+  xdescribe("Tests for send method", () => {
+    it("Passes send message into the send callback as the second parameter.", () => {
+      serverUnaryDecorator(object, mockSendCalback);
+      const fakeMessage = { message: "hello" };
+      object.send(fakeMessage);
+      expect(
+        mockSendCalback.mock.calls[0][1].hasOwnProperty("message")
+      ).toBeTruthy();
+    });
+  });
+
+  describe("Tests for set method", () => {
+    it("Adds new properties to the metadata object", () => {
+      serverUnaryDecorator(object, mockSendCalback);
+      const fakeMetadata = { message: "hello" };
+      object.set(fakeMetadata);
+      expect(
+        mockSendCalback.mock.calls[0][1].hasOwnProperty("message")
+      ).toBeTruthy();
+    });
+  });
+
+  describe("Tests for metadata passing", () => {
+    it("Passes send message into the send callback as the second parameter.", () => {
+      // add metadata to the call with set
+      //send
+      //check if send metadata was called
+      // serverUnaryDecorator(object, mockSendCalback);
+      // const fakeMessage = { message: "hello" };
+      // object.send(fakeMessage);
+      // expect(
+      //   mockSendCalback.mock.calls[0][1].hasOwnProperty("message")
+      // ).toBeTruthy();
+    });
+  });
 });

examples/firecomm/client.js

@@ -38,10 +38,10 @@ const firstChat = {
 
 const { log: c } = console;
 
-stub
-  .unaryChat(firstChat)
-  .then(res => console.log(res))
-  .catch(err => console.log(err));
+// stub
+//   .unaryChat(firstChat)
+//   .then(res => console.log(res))
+//   .catch(err => console.log(err));
 
 const testUnaryChat = () => {
   // console.log(stub.getChannel().getConnectivityState(true))
@@ -62,7 +62,7 @@ const testClientStream = () => {
   clientStream.end();
 };
 
-// testClientStream();
+testClientStream();
 
 const testServerStream = () => {
   const serverStream = stub.serverStream(firstChat);

examples/firecomm/methodHandlers.js

@@ -23,8 +23,8 @@ function clientStream(context) {
   // console.log('serverStream context: ', context);
   context.on("data", data => {
     console.log(data);
+    context.send({ message: "world" });
   });
-  context.send({ message: "world" });
 }
 
 function bidiChat(context) {

lib/callFactories/callDecorators/serverUnaryDecorator.js

@@ -39,16 +39,20 @@ module.exports = function serverUnaryDecorator(customCall, sendCallback) {
       );
     }
 
-    if (typeof trailers !== "object") {
-      throw new Error(
-        "Please input trailers, second parameter to Throw as an object."
-      );
-    }
-    this.trailers = generateMeta(trailers);
+    if (typeof trailers !== "undefined") {
+      if (typeof trailers !== "object") {
+        throw new Error(
+          "Please input trailers, second parameter to Throw as an object."
+        );
+      }
+      this.trailers = generateMeta(trailers);
 
-    const keys = Object.keys(trailers);
-    for (let i = 0; i < keys.length; i++) {
-      this.trailer.set(keys[i], metaObject[keys[i]]);
+      const keys = Object.keys(trailers);
+      for (let i = 0; i < keys.length; i++) {
+        this.trailer.set(keys[i], metaObject[keys[i]]);
+      }
+    } else {
+      this.trailers = trailers;
     }
 
     this.sendCallback(err, {}, this.trailers);