123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107
import { assertEquals } from "https://deno.land/x/testing/asserts.ts";import { runTests, test } from "https://deno.land/x/testing/mod.ts";import { Client } from "./mod.ts";import "./tests/query.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,            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",    1  ]);  assertEquals(result, { affectedRows: 1, lastInsertId: 0 });});
test(async function testQuery() {  let result = await client.query("select ??,name from ?? where id = ?", [    "id",    "users",    1  ]);  assertEquals(result, [{ id: 1, name: "MYR" }]);});
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.connections.length);  const expect = new Array(10).fill([    {      "1": 1    }  ]);  const result = await Promise.all(expect.map(() => client.query(`select 1`)));  assertEquals(client.config.pool, client.connections.length);  assertEquals(result, expect);});
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();