From 6092201a8b488adcbe5524770e209ce641c06207 Mon Sep 17 00:00:00 2001
From: ywceric <eric@tractable.io>
Date: Fri, 15 Dec 2017 11:46:21 +0000
Subject: [PATCH] [cassandra-driver] Add signature for Promise-based
 connect/shutdown

---
 types/cassandra-driver/index.d.ts | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/types/cassandra-driver/index.d.ts b/types/cassandra-driver/index.d.ts
index bdaed6559c..5c0578ccf2 100644
--- a/types/cassandra-driver/index.d.ts
+++ b/types/cassandra-driver/index.d.ts
@@ -513,6 +513,7 @@ export interface Client extends events.EventEmitter {
   batch(queries: Array<string> | Array<{ query: string, params?: any }>, options?: QueryOptions): Promise<types.ResultSet>;
 
   connect(callback: Callback): void;
+  connect(): Promise<void>;
   eachRow(query: string, params?: any, options?: QueryOptions, rowCallback?: Callback, callback?: Callback): void;
   execute(query: string, params: any, options: QueryOptions, callback: ResultCallback): void;
   execute(query: string, params: any, callback: ResultCallback): void;
@@ -520,6 +521,7 @@ export interface Client extends events.EventEmitter {
   execute(query: string, params?: any, options?: QueryOptions): Promise<types.ResultSet>;
   getReplicas(keyspace: string, token: Buffer): Array<any>; // TODO: Should this be a more explicit return?
   shutdown(callback?: Callback): void;
+  shutdown(): Promise<void>;
   stream(query: string, params?: any, options?: QueryOptions, callback?: Callback): NodeJS.ReadableStream;
 }