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

Source Code

import retry from 'async-retry';
import type Docker from 'dockerode';
import { relations, sql } from 'drizzle-orm';
import { drizzle, type GelJsDatabase } from 'drizzle-orm/gel';
import { foreignKey, gelSchema, gelTable, text, timestamptz, uniqueIndex, uuid } from 'drizzle-orm/gel-core';
import createClient, { type Client } from 'gel';
import { afterAll, afterEach, beforeAll, describe, expect, test } from 'vitest';
import 'zx/globals';
import { createDockerDB } from './createInstance';

$.quiet = true;

const ENABLE_LOGGING = false;

export const extauth = gelSchema('ext::auth');

export const identityInExtauth = extauth.table('Identity', {
	id: uuid().default(sql`uuid_generate_v4()`).primaryKey().notNull(),
	createdAt: timestamptz('created_at').default(sql`(clock_timestamp())`).notNull(),
	issuer: text().notNull(),
	modifiedAt: timestamptz('modified_at').notNull(),
	subject: text().notNull(),
}, (table) => [
	uniqueIndex('6bc2dd19-bce4-5810-bb1b-7007afe97a11;schemaconstr').using(
		'btree',
		table.id.asc().nullsLast().op('uuid_ops'),
	),
]);

export const user = gelTable('User', {
	id: uuid().default(sql`uuid_generate_v4()`).primaryKey().notNull(),
	email: text().notNull(),
	identityId: uuid('identity_id').notNull(),
	username: text().notNull(),
}, (table) => [
	uniqueIndex('d504514c-26a7-11f0-b836-81aa188c0abe;schemaconstr').using(
		'btree',
		table.id.asc().nullsLast().op('uuid_ops'),
	),
	foreignKey({
		columns: [table.identityId],
		foreignColumns: [identityInExtauth.id],
		name: 'User_fk_identity',
	}),
]);

export const userRelations = relations(user, ({ one }) => ({
	identity: one(identityInExtauth, { references: [identityInExtauth.id], fields: [user.identityId] }),
}));

const schema = { user, identityInExtauth, userRelations };

let client: Client;
let db: GelJsDatabase<typeof schema>;
const tlsSecurity: string = 'insecure';
let dsn: string;
let container: Docker.Container | undefined;

function sleep(ms: number) {
	return new Promise((resolve) => setTimeout(resolve, ms));
// ... (101 more lines)