fix: wait for pool.end()

Because when you don't pass a callback to .end() it always returns a promise

Author: asadbek-raimov

packages/pg-pool/index.js

@@ -356,18 +356,17 @@ class Pool extends EventEmitter {
     if (isExpired) {
       this.log('remove expired client')
       this._expired.delete(client)
-      this._remove(client, this._pulseQueue.bind(this))
-      return
+      return this._remove(client, this._pulseQueue.bind(this))
     }
 
     // idle timeout
     let tid
     if (this.options.idleTimeoutMillis) {
       tid = setTimeout(() => {
         this.log('remove idle client')
-        this._remove(client)
+        this._remove(client, this._pulseQueue.bind(this))
       }, this.options.idleTimeoutMillis)
-
+      
       if (this.options.allowExitOnIdle) {
         // allow Node to exit if this is all that's left
         tid.unref()

packages/pg-pool/test/idle-timeout.js

@@ -50,7 +50,7 @@ describe('idle timeout', () => {
       try {
         yield Promise.race([removal, timeout])
       } finally {
-        pool.end()
+        yield pool.end()
       }
     })
   )