gel-custom.test.ts — drizzle-orm Source File

Source Code

import retry from 'async-retry';
import type Docker from 'dockerode';
import { asc, eq, sql } from 'drizzle-orm';
import { drizzle, type GelJsDatabase } from 'drizzle-orm/gel';
import { alias, customType, gelTable, gelTableCreator } from 'drizzle-orm/gel-core';
import * as gel from 'gel';
import { afterAll, afterEach, beforeAll, beforeEach, expect, test } from 'vitest';
import { createDockerDB } from './createInstance';
import 'zx/globals';

$.quiet = true;

const ENABLE_LOGGING = false;

let db: GelJsDatabase;
let client: gel.Client;
let container: Docker.Container | undefined;

let dsn: string;
const tlsSecurity = '--tls-security=insecure';

function sleep(ms: number) {
	return new Promise((resolve) => setTimeout(resolve, ms));
}

beforeAll(async () => {
	let connectionString;
	if (process.env['GEL_CONNECTION_STRING']) {
		connectionString = process.env['GEL_CONNECTION_STRING'];
	} else {
		const { connectionString: conStr, container: contrainerObj } = await createDockerDB();
		connectionString = conStr;
		container = contrainerObj;
	}
	await sleep(15 * 1000);
	client = await retry(async () => {
		client = gel.createClient({ dsn: connectionString, tlsSecurity: 'insecure' });
		return client;
	}, {
		retries: 20,
		factor: 1,
		minTimeout: 250,
		maxTimeout: 250,
		randomize: false,
		onRetry() {
			client?.close();
		},
	});
	db = drizzle(client, { logger: ENABLE_LOGGING });

	dsn = connectionString;
	await $`gel query "CREATE TYPE default::users_custom {
		create property id1: int16 {
			create constraint exclusive;
		};
		create required property name: str;
		create property verified: bool {
		  SET default := false;
		};
		create property json: json;
// ... (620 more lines)