123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155
import { assertEquals, delay } from "./test_deps.ts";import { getMainConfiguration } from "./config.ts";import { generatePoolClientTest } from "./helpers.ts";
const testPool = generatePoolClientTest(getMainConfiguration());
Deno.test(  "Pool handles simultaneous connections correcly",  testPool(    async (POOL) => {      assertEquals(POOL.available, 10);      const client = await POOL.connect();      const p = client.queryArray("SELECT pg_sleep(0.1) is null, -1 AS id");      await delay(1);      assertEquals(POOL.available, 9);      assertEquals(POOL.size, 10);      await p;      client.release();      assertEquals(POOL.available, 10);
      const qsThunks = [...Array(25)].map(async (_, i) => {        const client = await POOL.connect();        const query = await client.queryArray(          "SELECT pg_sleep(0.1) is null, $1::text as id",          [i],        );        client.release();        return query;      });      const qsPromises = Promise.all(qsThunks);      await delay(1);      assertEquals(POOL.available, 0);      const qs = await qsPromises;      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);    },  ),);
Deno.test(  "Pool initializes lazy connections on demand",  testPool(    async (POOL, size) => {      const client_1 = await POOL.connect();      await client_1.queryArray("SELECT 1");      await client_1.release();      assertEquals(await POOL.initialized(), 1);
      const client_2 = await POOL.connect();      const p = client_2.queryArray("SELECT pg_sleep(0.1) is null, -1 AS id");      await delay(1);      assertEquals(POOL.size, size);      assertEquals(POOL.available, size - 1);      assertEquals(await POOL.initialized(), 0);      await p;      await client_2.release();      assertEquals(await POOL.initialized(), 1);
      // Test stack repletion as well      const requested_clients = size + 5;      const qsThunks = Array.from(        { length: requested_clients },        async (_, i) => {          const client = await POOL.connect();          const query = await client.queryArray(            "SELECT pg_sleep(0.1) is null, $1::text as id",            [i],          );          client.release();          return query;        },      );      const qsPromises = Promise.all(qsThunks);      await delay(1);      assertEquals(POOL.available, 0);      assertEquals(await POOL.initialized(), 0);      const qs = await qsPromises;      assertEquals(POOL.available, size);      assertEquals(await POOL.initialized(), size);
      const result = qs.map((r) => r.rows[0][1]);      const expected = Array.from(        { length: requested_clients },        (_, i) => i.toString(),      );      assertEquals(result, expected);    },    10,    true,  ),);
Deno.test(  "Pool can be reinitialized after termination",  testPool(async (POOL) => {    await POOL.end();    assertEquals(POOL.available, 0);
    const client = await POOL.connect();    await client.queryArray`SELECT 1`;    client.release();    assertEquals(POOL.available, 10);  }),);
Deno.test(  "Lazy pool can be reinitialized after termination",  testPool(    async (POOL, size) => {      await POOL.end();      assertEquals(POOL.available, 0);      assertEquals(await POOL.initialized(), 0);
      const client = await POOL.connect();      await client.queryArray`SELECT 1`;      client.release();      assertEquals(await POOL.initialized(), 1);      assertEquals(POOL.available, size);    },    10,    true,  ),);
Deno.test(  "Concurrent connect-then-release cycles do not throw",  testPool(async (POOL) => {    async function connectThenRelease() {      let client = await POOL.connect();      client.release();      client = await POOL.connect();      client.release();    }    await Promise.all(      Array.from({ length: POOL.size + 1 }, connectThenRelease),    );  }),);
Deno.test(  "Pool client will be released after `using` block",  testPool(async (POOL) => {    const initialPoolAvailable = POOL.available;    {      using _client = await POOL.connect();      assertEquals(POOL.available, initialPoolAvailable - 1);    }    assertEquals(POOL.available, initialPoolAvailable);  }),);