123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167
import {  assertEquals,  assertThrowsAsync} from "https://deno.land/std/testing/asserts.ts";import { runTests, test } from "https://deno.land/std/testing/mod.ts";import { Client } from "./mod.ts";import { WriteError } from "./src/consttants/errors.ts";
let client: Client;
test(async function testCreateDb() {  await client.query(`CREATE DATABASE IF NOT EXISTS enok`);});
test(async function testCreateTable() {  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=utf8;    `);});
test(async function testInsert() {  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 });});
test(async function testUpdate() {  let result = await client.execute(`update users set ?? = ?, ?? = ? WHERE id = ?`, [    "name",    "MYR",    "created_at",    new Date(),    1  ]);  assertEquals(result, { affectedRows: 1, lastInsertId: 0 });});
test(async function testQuery() {  let result = await client.query(    "select ??,`is_top`,`name` from ?? where id = ?",    ["id", "users", 1]  );  assertEquals(result, [{ id: 1, name: "MYR", is_top: false }]);});
test(async function testQueryErrorOccurred() {  assertEquals(1, client.poolSize);  await assertThrowsAsync(    () => client.query("select unknownfield from `users`"),    Error  );  await client.query("select 1");  assertEquals(client.poolSize, 1);});
test(async function testQueryList() {  const sql = "select ??,?? from ??";  let result = await client.query(sql, ["id", "name", "users"]);  assertEquals(result, [{ id: 1, name: "MYR" }, { id: 2, name: "MySQL" }]);});
test(async function testDelete() {  let result = await client.execute(`delete from users where ?? = ?`, [    "id",    1  ]);  assertEquals(result, { affectedRows: 1, lastInsertId: 0 });});
test(async function testPool() {  assertEquals(1, client.poolLength);  assertEquals(1, client.poolSize);  const expect = new Array(10).fill([    {      "1": 1    }  ]);  const result = await Promise.all(expect.map(() => client.query(`select 1`)));  assertEquals(client.config.pool, client.poolLength);  assertEquals(client.config.pool, client.poolSize);  assertEquals(result, expect);});
test(async function testQueryOnClosed() {  for (const i of [0, 0, 0]) {    await assertThrowsAsync(async () => {      await client.transaction(async conn => {        await conn.close();        await conn.query("SELECT 1");      });    }, WriteError);  }  assertEquals(client.poolSize, 0);  const result = client.query("select 1");});
test(async function testTransactionSuccess() {  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);});
test(async function testTransactionRollback() {  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);});
async function main() {  const { DB_PORT, DB_NAME, DB_PASSWORD, DB_USER, DB_HOST } = Deno.env();  const port = DB_PORT ? parseInt(DB_PORT) : 3306;  const db = DB_NAME || "test";  const password = DB_PASSWORD;  const username = DB_USER || "root";  const hostname = DB_HOST || "127.0.0.1";
  const config = {    timeout: 10000,    pool: 3,    debug: false,    hostname,    username,    port,    db,    password  };  client = await new Client().connect({ ...config, pool: 1, db: null });  await client.execute(`CREATE DATABASE IF NOT EXISTS ${db}`);  await client.close();  client = await new Client().connect(config);  await runTests();  console.log("end");}
main();