fix: add test

open-telemetry · Dec 19, 2019 · 6897a29 · 6897a29
1 parent 6730872
commit 6897a29
Showing 1 changed file with 27 additions and 0 deletions.
diff --git a/packages/opentelemetry-plugin-grpc/test/grpc.test.ts b/packages/opentelemetry-plugin-grpc/test/grpc.test.ts
@@ -45,6 +45,8 @@ type TestGrpcClient = grpc.Client & {
   // tslint:disable-next-line:no-any
   unaryMethod: any;
   // tslint:disable-next-line:no-any
+  UnaryMethod: any;
+  // tslint:disable-next-line:no-any
   clientStreamMethod: any;
   // tslint:disable-next-line:no-any
   serverStreamMethod: any;
@@ -93,6 +95,24 @@ const grpcClient = {
     });
   },
 
+  UnaryMethod: (
+    client: TestGrpcClient,
+    request: TestRequestResponse
+  ): Promise<TestRequestResponse> => {
+    return new Promise((resolve, reject) => {
+      return client.UnaryMethod(
+        request,
+        (err: grpc.ServiceError, response: TestRequestResponse) => {
+          if (err) {
+            reject(err);
+          } else {
+            resolve(response);
+          }
+        }
+      );
+    });
+  },
+
   clientStreamMethod: (
     client: TestGrpcClient,
     request: TestRequestResponse[]
@@ -318,6 +338,13 @@ describe('GrpcPlugin', () => {
       request: requestList[0],
       result: requestList[0],
     },
+    {
+      description: 'Unary call',
+      methodName: 'UnaryMethod',
+      method: grpcClient.UnaryMethod,
+      request: requestList[0],
+      result: requestList[0],
+    },
     {
       description: 'clientStream call',
       methodName: 'ClientStreamMethod',