From a9771a25f55c4b34a65777e966668f68a8f25cbe Mon Sep 17 00:00:00 2001
From: Mayur Kale <mayurkale@google.com>
Date: Wed, 18 Dec 2019 20:28:13 -0800
Subject: [PATCH] fix: add test

---
 .../test/grpc.test.ts                         | 27 +++++++++++++++++++
 1 file changed, 27 insertions(+)

diff --git a/packages/opentelemetry-plugin-grpc/test/grpc.test.ts b/packages/opentelemetry-plugin-grpc/test/grpc.test.ts
index 4c7d5cb73f7..a6d4f354509 100644
--- 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',