neon-ws.test.mjs — drizzle-orm Source File

Source Code

import 'dotenv/config';
import { Client, neonConfig, Pool } from '@neondatabase/serverless';
import { drizzle } from 'drizzle-orm/neon-serverless';
import { describe, expect } from 'vitest';
import ws from 'ws';
import { pg as schema } from './schema.mjs';

neonConfig.webSocketConstructor = ws;

if (!process.env['NEON_CONNECTION_STRING']) {
	throw new Error('NEON_CONNECTION_STRING is not defined');
}

describe('neon-ws', async (it) => {
	it('drizzle(string)', async () => {
		const db = drizzle(
			process.env['NEON_CONNECTION_STRING'],
		);

		await db.$client.query('SELECT 1;');
	});

	it('drizzle(string, config)', async () => {
		const db = drizzle(
			process.env['NEON_CONNECTION_STRING'],
			{
				schema,
			},
		);

		await db.$client.query('SELECT 1;');

		expect(db.query.User).not.toStrictEqual(undefined);
		expect(db.$client).toBeInstanceOf(Pool);
	});

	it('drizzle({connection: string, ...config})', async () => {
		const db = drizzle({
			connection: process.env['NEON_CONNECTION_STRING'],
			schema,
		});

		await db.$client.query('SELECT 1;');

		expect(db.query.User).not.toStrictEqual(undefined);
		expect(db.$client).toBeInstanceOf(Pool);
	});

	it('drizzle({connection: params, ...config})', async () => {
		const db = drizzle({
			connection: {
				connectionString: process.env['NEON_CONNECTION_STRING'],
			},
			schema,
		});

		await db.$client.query('SELECT 1;');

		expect(db.query.User).not.toStrictEqual(undefined);
		expect(db.$client).toBeInstanceOf(Pool);
// ... (152 more lines)