123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217
import {  assertEquals,  assertThrowsAsync,  semver,} from "./test.deps.ts";import { WriteError } from "./src/constant/errors.ts";import { createTestDB, testWithClient, isMariaDB } from "./test.util.ts";
testWithClient(async function testCreateDb(client) {  await client.query(`CREATE DATABASE IF NOT EXISTS enok`);});
testWithClient(async function testCreateTable(client) {  await client.query(`DROP TABLE IF EXISTS users`);  await client.query(`        CREATE TABLE users (            id int(11) NOT NULL AUTO_INCREMENT,            name varchar(100) NOT NULL,            is_top tinyint(1) default 0,            created_at timestamp not null default current_timestamp,            PRIMARY KEY (id)        ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4;    `);});
testWithClient(async function testInsert(client) {  let result = await client.execute(`INSERT INTO users(name) values(?)`, [    "manyuanrong",  ]);  assertEquals(result, { affectedRows: 1, lastInsertId: 1 });  result = await client.execute(`INSERT INTO users ?? values ?`, [    ["id", "name"],    [2, "MySQL"],  ]);  assertEquals(result, { affectedRows: 1, lastInsertId: 2 });});
testWithClient(async function testUpdate(client) {  let result = await client.execute(    `update users set ?? = ?, ?? = ? WHERE id = ?`,    ["name", "MYR🦕", "created_at", new Date(), 1],  );  assertEquals(result, { affectedRows: 1, lastInsertId: 0 });});
testWithClient(async function testQuery(client) {  let result = await client.query(    "select ??,`is_top`,`name` from ?? where id = ?",    ["id", "users", 1],  );  assertEquals(result, [{ id: 1, name: "MYR🦕", is_top: false }]);});
testWithClient(async function testQueryErrorOccurred(client) {  assertEquals(client.pool, {    size: 0,    maxSize: client.config.poolSize,    available: 0,  });  await assertThrowsAsync(    () => client.query("select unknownfield from `users`"),    Error,  );  await client.query("select 1");  assertEquals(client.pool, {    size: 1,    maxSize: client.config.poolSize,    available: 1,  });});
testWithClient(async function testQueryList(client) {  const sql = "select ??,?? from ??";  let result = await client.query(sql, ["id", "name", "users"]);  assertEquals(result, [    { id: 1, name: "MYR🦕" },    { id: 2, name: "MySQL" },  ]);});
testWithClient(async function testQueryTime(client) {  const sql = `SELECT CAST("09:04:10" AS time) as time`;  let result = await client.query(sql);  assertEquals(result, [{ time: "09:04:10" }]);});
testWithClient(async function testQueryBigint(client) {  await client.query(`DROP TABLE IF EXISTS test_bigint`);  await client.query(`CREATE TABLE test_bigint (    id int(11) NOT NULL AUTO_INCREMENT,    bigint_column bigint NOT NULL,    PRIMARY KEY (id)  ) ENGINE=MEMORY DEFAULT CHARSET=utf8mb4`);
  const value = "9223372036854775807";  await client.execute(    "INSERT INTO test_bigint(bigint_column) VALUES (?)",    [value],  );
  const result = await client.query("SELECT bigint_column FROM test_bigint");  assertEquals(result, [{ bigint_column: BigInt(value) }]);});
testWithClient(async function testQueryDecimal(client) {  await client.query(`DROP TABLE IF EXISTS test_decimal`);  await client.query(`CREATE TABLE test_decimal (    id int(11) NOT NULL AUTO_INCREMENT,    decimal_column decimal(65,30) NOT NULL,    PRIMARY KEY (id)  ) ENGINE=MEMORY DEFAULT CHARSET=utf8mb4`);
  const value = "0.012345678901234567890123456789";  await client.execute(    "INSERT INTO test_decimal(decimal_column) VALUES (?)",    [value],  );
  const result = await client.query("SELECT decimal_column FROM test_decimal");  assertEquals(result, [{ decimal_column: value }]);});
testWithClient(async function testQueryDatetime(client) {  await client.useConnection(async (connection) => {    if (isMariaDB(connection) || semver.lt(connection.serverVersion, "5.6.0")) {      return;    }
    await client.query(`DROP TABLE IF EXISTS test_datetime`);    await client.query(`CREATE TABLE test_datetime (      id int(11) NOT NULL AUTO_INCREMENT,      datetime datetime(6) NOT NULL,      PRIMARY KEY (id)    ) ENGINE=MEMORY DEFAULT CHARSET=utf8mb4`);    const datetime = new Date();    await client.execute(      `      INSERT INTO test_datetime (datetime)      VALUES (?)`,      [datetime],    );
    const [row] = await client.query("SELECT datetime FROM test_datetime");    assertEquals(row.datetime.toISOString(), datetime.toISOString()); // See https://github.com/denoland/deno/issues/6643  });});
testWithClient(async function testDelete(client) {  let result = await client.execute(`delete from users where ?? = ?`, [    "id",    1,  ]);  assertEquals(result, { affectedRows: 1, lastInsertId: 0 });});
testWithClient(async function testPool(client) {  assertEquals(client.pool, {    maxSize: client.config.poolSize,    available: 0,    size: 0,  });  const expect = new Array(10).fill([{ "1": 1 }]);  const result = await Promise.all(expect.map(() => client.query(`select 1`)));
  assertEquals(client.pool, {    maxSize: client.config.poolSize,    available: 3,    size: 3,  });  assertEquals(result, expect);});
testWithClient(async function testQueryOnClosed(client) {  for (const i of [0, 0, 0]) {    await assertThrowsAsync(async () => {      await client.transaction(async (conn) => {        conn.close();        await conn.query("SELECT 1");      });    }, WriteError);  }  assertEquals(client.pool?.size, 0);  await client.query("select 1");});
testWithClient(async function testTransactionSuccess(client) {  const success = await client.transaction(async (connection) => {    await connection.execute("insert into users(name) values(?)", [      "transaction1",    ]);    await connection.execute("delete from users where id = ?", [2]);    return true;  });  assertEquals(true, success);  const result = await client.query("select name,id from users");  assertEquals([{ name: "transaction1", id: 3 }], result);});
testWithClient(async function testTransactionRollback(client) {  let success;  await assertThrowsAsync(async () => {    success = await client.transaction(async (connection) => {      // Insert an existing id      await connection.execute("insert into users(name,id) values(?,?)", [        "transaction2",        3,      ]);      return true;    });  });  assertEquals(undefined, success);  const result = await client.query("select name from users");  assertEquals([{ name: "transaction1" }], result);});
await createTestDB();