sqlite-views.test.ts — drizzle-orm Source File

Source Code

import { sql } from 'drizzle-orm';
import { int, sqliteTable, sqliteView } from 'drizzle-orm/sqlite-core';
import { expect, test } from 'vitest';
import { diffTestSchemasSqlite } from './schemaDiffer';

test('create view', async () => {
	const users = sqliteTable('users', { id: int('id').default(1) });
	const view = sqliteView('view').as((qb) => qb.select().from(users));
	const to = {
		users: users,
		testView: view,
	};

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

	expect(statements.length).toBe(2);
	expect(statements[0]).toStrictEqual({
		type: 'sqlite_create_table',
		tableName: 'users',
		columns: [{
			autoincrement: false,
			default: 1,
			name: 'id',
			type: 'integer',
			primaryKey: false,
			notNull: false,
		}],
		compositePKs: [],
		uniqueConstraints: [],
		referenceData: [],
		checkConstraints: [],
	});
	expect(statements[1]).toStrictEqual({
		type: 'sqlite_create_view',
		name: 'view',
		definition: 'select "id" from "users"',
	});

	expect(sqlStatements.length).toBe(2);
	expect(sqlStatements[0]).toBe(`CREATE TABLE \`users\` (
\t\`id\` integer DEFAULT 1
);\n`);
	expect(sqlStatements[1]).toBe(`CREATE VIEW \`view\` AS select "id" from "users";`);
});

test('drop view', async () => {
	const users = sqliteTable('users', {
		id: int('id').primaryKey({ autoIncrement: true }),
	});

	const from = {
		users: users,
		testView: sqliteView('view', { id: int('id') }).as(sql`SELECT * FROM users`),
	};
	const to = {
		users,
	};

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

// ... (159 more lines)