cyclicTables.test.ts — drizzle-orm Source File

Source Code

import { PGlite } from '@electric-sql/pglite';
import { sql } from 'drizzle-orm';
import type { PgliteDatabase } from 'drizzle-orm/pglite';
import { drizzle } from 'drizzle-orm/pglite';
import { afterAll, afterEach, beforeAll, expect, test } from 'vitest';
import { reset, seed } from '../../../src/index.ts';
import * as schema from './pgSchema.ts';

let client: PGlite;
let db: PgliteDatabase;

beforeAll(async () => {
	client = new PGlite();

	db = drizzle(client);

	await db.execute(
		sql`
			create table model_image
			(
			    id        serial
			        primary key,
			    url       varchar not null,
			    caption   varchar,
			    "modelId" integer not null
			);
		`,
	);

	await db.execute(
		sql`
			create table model
			      (
			          id               serial
			              primary key,
			          name             varchar not null,
			          "defaultImageId" integer
			              constraint "model_defaultImageId_model_image_id_fk"
			                  references model_image
			      );
		`,
	);

	await db.execute(
		sql`
			alter table model_image
						    add constraint "model_image_modelId_model_id_fk"
						        foreign key ("modelId") references model;
		`,
	);

	// 3 tables case
	await db.execute(
		sql`
			create table model_image1
			(
			    id        serial
			        primary key,
			    url       varchar not null,
			    caption   varchar,
// ... (98 more lines)