From ac8c2022c8517f186bedf93107b571cb428692dc Mon Sep 17 00:00:00 2001
From: Paul van Brenk <paul.van.brenk@microsoft.com>
Date: Tue, 21 Mar 2017 15:50:50 -0700
Subject: [PATCH] Fix promise usage

---
 pg-query-stream/pg-query-stream-tests.ts | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/pg-query-stream/pg-query-stream-tests.ts b/pg-query-stream/pg-query-stream-tests.ts
index 6c4c034102..04796580de 100644
--- a/pg-query-stream/pg-query-stream-tests.ts
+++ b/pg-query-stream/pg-query-stream-tests.ts
@@ -10,10 +10,10 @@ const query = new QueryStream('SELECT * FROM generate_series(0, $1) num', [10000
 
 pg.connect('', (err, client, done) => {
     const stream = client.query(query);
-    stream.on('end', () => {
+    stream.then(() => {
         client.end();
     });
-    stream.on('data', (data: any) => {
+    stream.then((data: any) => {
         console.log(data);
     });
 });