pg-sequences.test.ts — drizzle-orm Source File

Source Code

import { pgSchema, pgSequence } from 'drizzle-orm/pg-core';
import { expect, test } from 'vitest';
import { diffTestSchemas } from './schemaDiffer';

test('create sequence', async () => {
	const from = {};
	const to = {
		seq: pgSequence('name', { startWith: 100 }),
	};

	const { statements, sqlStatements } = await diffTestSchemas(from, to, []);

	expect(statements).toStrictEqual([
		{
			name: 'name',
			schema: 'public',
			type: 'create_sequence',
			values: {
				cache: '1',
				cycle: false,
				increment: '1',
				maxValue: '9223372036854775807',
				minValue: '1',
				startWith: '100',
			},
		},
	]);
	expect(sqlStatements).toStrictEqual([
		'CREATE SEQUENCE "public"."name" INCREMENT BY 1 MINVALUE 1 MAXVALUE 9223372036854775807 START WITH 100 CACHE 1;',
	]);
});

test('create sequence: all fields', async () => {
	const from = {};
	const to = {
		seq: pgSequence('name', {
			startWith: 100,
			maxValue: 10000,
			minValue: 100,
			cycle: true,
			cache: 10,
			increment: 2,
		}),
	};

	const { statements, sqlStatements } = await diffTestSchemas(from, to, []);

	expect(statements).toStrictEqual([
		{
			type: 'create_sequence',
			name: 'name',
			schema: 'public',
			values: {
				startWith: '100',
				maxValue: '10000',
				minValue: '100',
				cycle: true,
				cache: '10',
				increment: '2',
			},
// ... (239 more lines)