refactored generateServerStreamCall now passes basic tests

firecomm · Aug 6, 2019 · a3e00ee · a3e00ee
1 parent fe5ae40
commit a3e00ee
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/lib/callFactories/generateServerStreamCall.js b/lib/callFactories/generateServerStreamCall.js
@@ -2,6 +2,10 @@ const grpc = require("grpc");
 
 const generateMeta = require("../utils/generateMeta");
 
+const basicCallDecorator = require("./callDecorators/basicCallDecorator");
+const serverStreamDecorator = require("./callDecorators/serverStreamDecorator");
+const clientUnaryDecorator = require("./callDecorators/clientUnaryDecorator");
+
 function generateServerStreamCall(ServerWritableStream) {
   ServerWritableStreamClone = Object.create(ServerWritableStream);
   ServerWritableStreamClone.state = null;
@@ -67,4 +71,10 @@ function generateServerStreamCall(ServerWritableStream) {
 //   }
 // }
 
-module.exports = generateServerStreamCall;
+module.exports = function(ServerWritableStream) {
+  ServerWritableStreamClone = Object.create(ServerWritableStream);
+  basicCallDecorator(ServerWritableStreamClone, ServerWritableStream);
+  clientUnaryDecorator(ServerWritableStreamClone, ServerWritableStream);
+  serverStreamDecorator(ServerWritableStreamClone, ServerWritableStream);
+  return ServerWritableStreamClone;
+};