From 6ddbe6ab60072d2e67ea5ab3a82fc7a0da72f5a2 Mon Sep 17 00:00:00 2001
From: Andrew Heuermann <aheuermann@gmail.com>
Date: Sat, 28 Dec 2019 09:24:45 -0600
Subject: [PATCH] Close connection after version check

---
 .../idle_in_transaction_session_timeout-tests.js     | 12 +++++++-----
 1 file changed, 7 insertions(+), 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
index 4435a40b6..18162f545 100644
--- 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();
+      }))
     }))
   }))
 }