From d2a5819d31ead08a2d8731eceeed0c6d29028218 Mon Sep 17 00:00:00 2001
From: Christopher Brown <io@henrian.com>
Date: Wed, 23 Mar 2016 16:25:28 -0500
Subject: [PATCH] Add test for calling done with an argument

---
 pg/pg-tests.ts | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/pg/pg-tests.ts b/pg/pg-tests.ts
index 087c60834d..99d1fac9ba 100644
--- a/pg/pg-tests.ts
+++ b/pg/pg-tests.ts
@@ -12,10 +12,13 @@ pg.connect(conString, (err, client, done) => {
         return console.error("Error fetching client from pool", err);
     }
     client.query("SELECT $1::int AS number", ["1"], (err, result) => {
-        done();
         if (err) {
+            done(err);
             return console.error("Error running query", err);
         }
+        else {
+            done();
+        }
         console.log(result.rows[0]["number"]);
         return null;
     });
@@ -37,4 +40,4 @@ client.connect((err) => {
         return null;
     });
     return null;
-});
\ No newline at end of file
+});