123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145
import {  assertEquals,  assertThrowsAsync,} from "../test_deps.ts";import { Pool } from "../pool.ts";import { delay } from "../utils.ts";import { TEST_CONNECTION_PARAMS, DEFAULT_SETUP } from "./constants.ts";
async function testPool(  t: (pool: Pool) => void | Promise<void>,  setupQueries?: Array<string> | null,  lazy?: boolean,) {  // constructing Pool instantiates the connections,  // so this has to be constructed for each test.  const fn = async () => {    const POOL = new Pool(TEST_CONNECTION_PARAMS, 10, lazy);    try {      for (const q of setupQueries || DEFAULT_SETUP) {        await POOL.query(q);      }      await t(POOL);    } finally {      await POOL.end();    }  };  const name = t.name;  Deno.test({ fn, name });}
testPool(async function simpleQuery(POOL) {  const result = await POOL.query("SELECT * FROM ids;");  assertEquals(result.rows.length, 2);});
testPool(async function parametrizedQuery(POOL) {  const result = await POOL.query("SELECT * FROM ids WHERE id < $1;", 2);  assertEquals(result.rows.length, 1);
  const objectRows = result.rowsOfObjects();  const row = objectRows[0];
  assertEquals(row.id, 1);  assertEquals(typeof row.id, "number");});
testPool(async function nativeType(POOL) {  const result = await POOL.query("SELECT * FROM timestamps;");  const row = result.rows[0];
  const expectedDate = Date.UTC(2019, 1, 10, 6, 0, 40, 5);
  assertEquals(row[0].toUTCString(), new Date(expectedDate).toUTCString());
  await POOL.query("INSERT INTO timestamps(dt) values($1);", new Date());});
testPool(  async function lazyPool(POOL) {    await POOL.query("SELECT 1;");    assertEquals(POOL.available, 1);    const p = POOL.query("SELECT pg_sleep(0.1) is null, -1 AS id;");    await delay(1);    assertEquals(POOL.available, 0);    assertEquals(POOL.size, 1);    await p;    assertEquals(POOL.available, 1);
    const qs_thunks = [...Array(25)].map((_, i) =>      POOL.query("SELECT pg_sleep(0.1) is null, $1::text as id;", i)    );    const qs_promises = Promise.all(qs_thunks);    await delay(1);    assertEquals(POOL.available, 0);    const qs = await qs_promises;    assertEquals(POOL.available, 10);    assertEquals(POOL.size, 10);
    const result = qs.map((r) => r.rows[0][1]);    const expected = [...Array(25)].map((_, i) => i.toString());    assertEquals(result, expected);  },  null,  true,);
/** * @see https://github.com/bartlomieju/deno-postgres/issues/59 */testPool(async function returnedConnectionOnErrorOccurs(POOL) {  assertEquals(POOL.available, 10);  await assertThrowsAsync(async () => {    await POOL.query("SELECT * FROM notexists");  });  assertEquals(POOL.available, 10);});
testPool(async function manyQueries(POOL) {  assertEquals(POOL.available, 10);  const p = POOL.query("SELECT pg_sleep(0.1) is null, -1 AS id;");  await delay(1);  assertEquals(POOL.available, 9);  assertEquals(POOL.size, 10);  await p;  assertEquals(POOL.available, 10);
  const qs_thunks = [...Array(25)].map((_, i) =>    POOL.query("SELECT pg_sleep(0.1) is null, $1::text as id;", i)  );  const qs_promises = Promise.all(qs_thunks);  await delay(1);  assertEquals(POOL.available, 0);  const qs = await qs_promises;  assertEquals(POOL.available, 10);  assertEquals(POOL.size, 10);
  const result = qs.map((r) => r.rows[0][1]);  const expected = [...Array(25)].map((_, i) => i.toString());  assertEquals(result, expected);});
testPool(async function transaction(POOL) {  const client = await POOL.connect();  let errored;  let released;  assertEquals(POOL.available, 9);
  try {    await client.query("BEGIN");    await client.query("INSERT INTO timestamps(dt) values($1);", new Date());    await client.query("INSERT INTO ids(id) VALUES(3);");    await client.query("COMMIT");  } catch (e) {    await client.query("ROLLBACK");    errored = true;    throw e;  } finally {    client.release();    released = true;  }  assertEquals(errored, undefined);  assertEquals(released, true);  assertEquals(POOL.available, 10);});