index-queries.test.js — astro Source File

Source Code

import assert from 'node:assert/strict';
import { describe, it } from 'node:test';
import { getTableChangeQueries } from '../../dist/core/cli/migration-queries.js';
import { dbConfigSchema, tableSchema } from '../../dist/core/schemas.js';
import { column } from '../../dist/runtime/virtual.js';

const userInitial = tableSchema.parse({
	columns: {
		name: column.text(),
		age: column.number(),
		email: column.text({ unique: true }),
		mi: column.text({ optional: true }),
	},
	indexes: {},
	writable: false,
});

describe('index queries', () => {
	it('generates index names by table and combined column names', async () => {
		// Use dbConfigSchema.parse to resolve generated idx names
		const dbConfig = dbConfigSchema.parse({
			tables: {
				oldTable: userInitial,
				newTable: {
					...userInitial,
					indexes: [
						{ on: ['name', 'age'], unique: false },
						{ on: ['email'], unique: true },
					],
				},
			},
		});

		const { queries } = await getTableChangeQueries({
			tableName: 'user',
			oldTable: dbConfig.tables.oldTable,
			newTable: dbConfig.tables.newTable,
		});

		assert.deepEqual(queries, [
			'CREATE INDEX "newTable_age_name_idx" ON "user" ("age", "name")',
			'CREATE UNIQUE INDEX "newTable_email_idx" ON "user" ("email")',
		]);
	});

	it('generates index names with consistent column ordering', async () => {
		const initial = dbConfigSchema.parse({
			tables: {
				user: {
					...userInitial,
					indexes: [
						{ on: ['email'], unique: true },
						{ on: ['name', 'age'], unique: false },
					],
				},
			},
		});

		const final = dbConfigSchema.parse({
			tables: {
// ... (224 more lines)