Close connection after version check

brianc · Dec 28, 2019 · 6ddbe6a · 6ddbe6a
1 parent 8390432
commit 6ddbe6a
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/packages/pg/test/integration/client/idle_in_transaction_session_timeout-tests.js b/packages/pg/test/integration/client/idle_in_transaction_session_timeout-tests.js
@@ -14,11 +14,13 @@ function testClientVersion(cb) {
   var client = new Client({})
   client.connect(assert.success(function () {
     helper.versionGTE(client, 100000, assert.success(function(isGreater) {
-      if (!isGreater) {
-        console.log('skip idle_in_transaction_session_timeout at client-level is only available in v10 and above');
-        return client.end();
-      }
-      cb();
+      return client.end(assert.success(function () {
+        if (!isGreater) {
+          console.log('skip idle_in_transaction_session_timeout at client-level is only available in v10 and above');
+          return;
+        }
+        cb();
+      }))
     }))
   }))
 }