mysql.planetscale.test.ts — drizzle-orm Source File

Source Code

import 'dotenv/config';

import { Client } from '@planetscale/database';
import { desc, DrizzleError, eq, gt, gte, or, placeholder, sql, TransactionRollbackError } from 'drizzle-orm';
import { drizzle, type PlanetScaleDatabase } from 'drizzle-orm/planetscale-serverless';
import { beforeAll, beforeEach, expect, expectTypeOf, test } from 'vitest';
import * as schema from './mysql.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: PlanetScaleDatabase<typeof schema>;

beforeAll(async () => {
	db = drizzle(
		new Client({
			url: process.env['PLANETSCALE_CONNECTION_STRING']!,
			// host: process.env['DATABASE_HOST']!,
			// username: process.env['DATABASE_USERNAME']!,
			// password: process.env['DATABASE_PASSWORD']!,
		}),
		{ schema, logger: ENABLE_LOGGING },
	);

	await Promise.all([
		db.execute(sql`drop table if exists \`users\``),
		db.execute(sql`drop table if exists \`groups\``),
		db.execute(sql`drop table if exists \`users_to_groups\``),
		db.execute(sql`drop table if exists \`posts\``),
		db.execute(sql`drop table if exists \`comments\``),
		db.execute(sql`drop table if exists \`comment_likes\``),
	]);
	await Promise.all([
		db.execute(
			sql`
				CREATE TABLE IF NOT EXISTS \`users\` (
					  \`id\` bigint PRIMARY KEY AUTO_INCREMENT NOT NULL,
					  \`name\` text NOT NULL,
					  \`verified\` boolean DEFAULT false NOT NULL,
					  \`invited_by\` bigint
				);
			`,
		),
		db.execute(
			sql`
				CREATE TABLE IF NOT EXISTS \`groups\` (
					\`id\` bigint PRIMARY KEY AUTO_INCREMENT NOT NULL,
					\`name\` text NOT NULL,
					\`description\` text
				);
			`,
		),
		db.execute(
			sql`
// ... (6051 more lines)