diff --git a/packages/pg-query-stream/test/pool-query-stream-close.ts b/packages/pg-query-stream/test/pool-query-stream-close.ts new file mode 100644 index 000000000..7a36558d9 --- /dev/null +++ b/packages/pg-query-stream/test/pool-query-stream-close.ts @@ -0,0 +1,29 @@ +import helper from "./helper"; +import QueryStream from "../src"; +import pg from "pg"; + + +describe("Pool Query-Stream", function() { + + it('releases pool connection after end of query stream', async function() { + const pool = new pg.Pool(); + // const client = await pool.connect(); + const query = new QueryStream('SELECT NOW()'); + const stream = pool.query(query); + + query.on('end',async(res)=>{ + console.log('query stream ended'); + }); + + // bellow code raising error as stream .on is not a function + console.log(await stream); // this never gets resolved; + stream.on('end',async (res)=>{ + console.log(res); + await pool.end(); + console.log('stream ended'); + await Promise.resolve(); + }) + + }) + +})