sql1

server/node_modules/pg-pool/test/ending.js

@@ -0,0 +1,40 @@
+'use strict'
+const co = require('co')
+const expect = require('expect.js')
+
+const describe = require('mocha').describe
+const it = require('mocha').it
+
+const Pool = require('../')
+
+describe('pool ending', () => {
+  it('ends without being used', (done) => {
+    const pool = new Pool()
+    pool.end(done)
+  })
+
+  it('ends with a promise', () => {
+    return new Pool().end()
+  })
+
+  it(
+    'ends with clients',
+    co.wrap(function* () {
+      const pool = new Pool()
+      const res = yield pool.query('SELECT $1::text as name', ['brianc'])
+      expect(res.rows[0].name).to.equal('brianc')
+      return pool.end()
+    })
+  )
+
+  it(
+    'allows client to finish',
+    co.wrap(function* () {
+      const pool = new Pool()
+      const query = pool.query('SELECT $1::text as name', ['brianc'])
+      yield pool.end()
+      const res = yield query
+      expect(res.rows[0].name).to.equal('brianc')
+    })
+  )
+})