Add DgraphClientStub#waitForReady

Author: gpahal

src/clientStub.ts

@@ -16,6 +16,7 @@ export class DgraphClientStub {
         mutate(mu: messages.Mutation): Promise<messages.Assigned>;
         commitOrAbort(ctx: messages.TxnContext): Promise<messages.TxnContext>;
         checkVersion(check: messages.Check): Promise<messages.Version>;
+        waitForReady(deadline: grpc.Deadline): Promise<void>;
     };
 
     constructor(addr?: string | null, credentials?: grpc.ChannelCredentials | null) {
@@ -35,6 +36,7 @@ export class DgraphClientStub {
             mutate: promisify(this.stub.mutate, this.stub),
             commitOrAbort: promisify(this.stub.commitOrAbort, this.stub),
             checkVersion: promisify(this.stub.checkVersion, this.stub),
+            waitForReady: promisify(this.stub.waitForReady, this.stub),
         };
     }
 
@@ -58,6 +60,10 @@ export class DgraphClientStub {
         return this.promisified.checkVersion(check);
     }
 
+    public waitForReady(deadline: grpc.Deadline): Promise<void> {
+        return this.promisified.waitForReady(deadline);
+    }
+
     public close(): void {
         return this.stub.close();
     }

tests/clientStub.spec.ts

@@ -1,6 +1,6 @@
 import * as dgraph from "../src";
 
-import { setup } from "./helper";
+import { SERVER_ADDR, SERVER_CREDENTIALS, setup } from "./helper";
 
 async function checkVersion(stub: dgraph.DgraphClientStub): Promise<void> {
     const v = await stub.checkVersion(new dgraph.Check());
@@ -24,10 +24,17 @@ describe("clientStub", () => {
         });
     });
 
+    describe("waitForReady", () => {
+        it("should provide a promisified version of grpc.Client#waitForReady", async () => {
+            const clientStub = new dgraph.DgraphClientStub(SERVER_ADDR, SERVER_CREDENTIALS);
+            await clientStub.waitForReady(Date.now() + 500);
+            await checkVersion(clientStub);
+        });
+    });
+
     describe("close", () => {
         it("should close channel", async () => {
-            const client = await setup();
-            const clientStub = await client.anyClient();
+            const clientStub = new dgraph.DgraphClientStub(SERVER_ADDR, SERVER_CREDENTIALS);
             clientStub.close();
             const p = clientStub.checkVersion(new dgraph.Check());
             await expect(p).rejects.toBeDefined();