bettersqlite.test.ts — drizzle-orm Source File

Source Code

import 'dotenv/config';
import Database from 'better-sqlite3';
import { desc, DrizzleError, eq, gt, gte, or, placeholder, sql, TransactionRollbackError } from 'drizzle-orm';
import { type BetterSQLite3Database, drizzle } from 'drizzle-orm/better-sqlite3';
import { beforeAll, beforeEach, expect, expectTypeOf, test } from 'vitest';
import * as schema from './sqlite.schema.ts';

const { usersTable, postsTable, commentsTable, usersToGroupsTable, groupsTable } = schema;

const ENABLE_LOGGING = false;

/*
	Test cases:
	- querying nested relation without PK with additional fields
*/

let db: BetterSQLite3Database<typeof schema>;

beforeAll(() => {
	const dbPath = process.env['SQLITE_DB_PATH'] ?? ':memory:';

	db = drizzle(new Database(dbPath), { schema, logger: ENABLE_LOGGING });
});

beforeEach(() => {
	db.run(sql`drop table if exists \`groups\``);
	db.run(sql`drop table if exists \`users\``);
	db.run(sql`drop table if exists \`users_to_groups\``);
	db.run(sql`drop table if exists \`posts\``);
	db.run(sql`drop table if exists \`comments\``);
	db.run(sql`drop table if exists \`comment_likes\``);

	db.run(
		sql`
			CREATE TABLE \`users\` (
			    \`id\` integer PRIMARY KEY AUTOINCREMENT NOT NULL,
			    \`name\` text NOT NULL,
			    \`verified\` integer DEFAULT 0 NOT NULL,
			    \`invited_by\` integer
			);
		`,
	);
	db.run(
		sql`
			CREATE TABLE \`groups\` (
			    \`id\` integer PRIMARY KEY AUTOINCREMENT NOT NULL,
			    \`name\` text NOT NULL,
			    \`description\` text
			);
		`,
	);
	db.run(
		sql`
			CREATE TABLE \`users_to_groups\` (
			    \`id\` integer PRIMARY KEY AUTOINCREMENT NOT NULL,
			    \`user_id\` integer NOT NULL,
			    \`group_id\` integer NOT NULL
			);
		`,
	);
// ... (6043 more lines)