@contember/schema-migrations 1.3.6 → 1.4.0-alpha.2

package/tests/cases/integration/createUnique.test.ts

@@ -0,0 +1,146 @@
+import { c, createSchema } from '@contember/schema-definition'
+import { SQL } from '../../src/tags'
+import { testMigrations } from '../../src/tests'
+import { createUniqueConstraintModification, removeUniqueConstraintModification } from '../../../src'
+import { createDatabaseMetadata } from '@contember/database'
+
+testMigrations('create unique (immediate)', {
+	original: createSchema({
+		Author: class Author {
+			email = c.stringColumn()
+		},
+	}),
+	updated: createSchema({
+		Author: class Author {
+			email = c.stringColumn().unique()
+		},
+	}),
+	diff: [
+		createUniqueConstraintModification.createModification({
+			entityName: 'Author',
+			unique: {
+				fields: ['email'],
+			},
+		}),
+	],
+	sql: SQL`ALTER TABLE "author" ADD UNIQUE ("email");`,
+})
+
+testMigrations('create unique (deferrable)', {
+	original: createSchema({
+		Author: class Author {
+			email = c.stringColumn()
+		},
+	}),
+	updated: createSchema({
+		Author: class Author {
+			email = c.stringColumn().unique({ timing: 'deferrable' })
+		},
+	}),
+	diff: [
+		createUniqueConstraintModification.createModification({
+			entityName: 'Author',
+			unique: {
+				fields: ['email'],
+				timing: 'deferrable',
+			},
+		}),
+	],
+	sql: SQL`ALTER TABLE "author"
+        ADD UNIQUE ("email") DEFERRABLE;`,
+})
+
+
+testMigrations('create unique (deferred)', {
+	original: createSchema({
+		Author: class Author {
+			email = c.stringColumn()
+		},
+	}),
+	updated: createSchema({
+		Author: class Author {
+			email = c.stringColumn().unique({ timing: 'deferred' })
+		},
+	}),
+	diff: [
+
+		createUniqueConstraintModification.createModification({
+			entityName: 'Author',
+			unique: {
+				fields: ['email'],
+				timing: 'deferred',
+			},
+		}),
+	],
+	sql: SQL`ALTER TABLE "author"
+        ADD UNIQUE ("email") INITIALLY DEFERRED;`,
+})
+
+
+testMigrations('change unique timing', {
+	original: createSchema({
+		Author: class Author {
+			email = c.stringColumn().unique()
+		},
+	}),
+	updated: createSchema({
+		Author: class Author {
+			email = c.stringColumn().unique({ timing: 'deferrable' })
+		},
+	}),
+	diff: [
+		removeUniqueConstraintModification.createModification({
+			entityName: 'Author',
+			fields: ['email'],
+		}),
+		createUniqueConstraintModification.createModification({
+			entityName: 'Author',
+			unique: {
+				fields: ['email'],
+				timing: 'deferrable',
+			},
+		}),
+	],
+	sql: SQL`ALTER TABLE "author" DROP CONSTRAINT "uniq_author_email"; 
+ALTER TABLE "author" ADD UNIQUE ("email") DEFERRABLE;`,
+	databaseMetadata: createDatabaseMetadata({
+		foreignKeys: [],
+		indexes: [],
+		uniqueConstraints: [{
+			constraintName: 'uniq_author_email',
+			columnNames: ['email'],
+			tableName: 'author',
+			deferred: false,
+			deferrable: false,
+		}],
+	}),
+})
+
+
+namespace SchemaWithUniqueDecorator {
+
+	@c.Unique({ fields: ['email'], timing: 'deferred' })
+	export class Author {
+		email = c.stringColumn()
+	}
+}
+
+testMigrations('create unique using decorator', {
+	original: createSchema({
+		Author: class Author {
+			email = c.stringColumn()
+		},
+	}),
+	updated: createSchema(SchemaWithUniqueDecorator),
+	diff: [
+		createUniqueConstraintModification.createModification({
+			entityName: 'Author',
+			unique: {
+				fields: ['email'],
+				timing: 'deferred',
+			},
+		}),
+	],
+	sql: SQL`ALTER TABLE "author"
+        ADD UNIQUE ("email") INITIALLY DEFERRED;`,
+})

package/tests/cases/integration/removeField.test.ts

@@ -3,6 +3,7 @@ import { createSchema, SchemaBuilder } from '@contember/schema-definition'
 import { Model } from '@contember/schema'
 import { SQL } from '../../src/tags'
 import { SchemaDefinition as def } from '@contember/schema-definition'
+import { createDatabaseMetadata } from '@contember/database'
 
 
 testMigrations('remove relation (many has one)', {
@@ -67,6 +68,17 @@ testMigrations('remove relation (one has many)', {
 	],
 	sql: SQL`ALTER TABLE "post_locale" DROP CONSTRAINT "uniq_post_locale_post_id_locale";
 						ALTER TABLE "post_locale" DROP "post_id";`,
+	databaseMetadata: createDatabaseMetadata({
+		foreignKeys: [],
+		indexes: [],
+		uniqueConstraints: [{
+			constraintName: 'uniq_post_locale_post_id_locale',
+			columnNames: ['post_id', 'locale'],
+			tableName: 'post_locale',
+			deferrable: false,
+			deferred: false,
+		}],
+	}),
 })
 
 testMigrations('remove relation (many has many)', {

package/tests/cases/integration/updateRelationOnDelete.test.ts

@@ -2,6 +2,7 @@ import { testMigrations } from '../../src/tests'
 import { SchemaBuilder } from '@contember/schema-definition'
 import { Model } from '@contember/schema'
 import { SQL } from '../../src/tags'
+import { ForeignKeyDeleteAction, createDatabaseMetadata } from '@contember/database'
 
 testMigrations('update relation ondelete to cascade', {
 	original: {
@@ -32,4 +33,20 @@ testMigrations('update relation ondelete to cascade', {
 	],
 	sql: SQL`ALTER TABLE "post" DROP CONSTRAINT "fk_post_category_id_category_id"; 
 ALTER TABLE "post" ADD FOREIGN KEY ("category_id") REFERENCES "category"("id") ON DELETE CASCADE DEFERRABLE INITIALLY IMMEDIATE;`,
+	databaseMetadata: createDatabaseMetadata({
+		foreignKeys: [
+			{
+				constraintName: 'fk_post_category_id_category_id',
+				deleteAction: ForeignKeyDeleteAction.cascade,
+				fromColumn: 'category_id',
+				fromTable: 'post',
+				toColumn: 'id',
+				toTable: 'category',
+				deferred: false,
+				deferrable: false,
+			},
+		],
+		indexes: [],
+		uniqueConstraints: [],
+	}),
 })

package/tests/src/tests.ts

@@ -2,7 +2,8 @@ import { Migration, ModificationHandlerFactory, SchemaDiffer, SchemaMigrator, VE
 import { Schema } from '@contember/schema'
 import { createMigrationBuilder } from '@contember/database-migrations'
 import { assert, describe, it } from 'vitest'
-import { dummySchemaDatabaseMetadata, emptySchema } from '@contember/schema-utils'
+import { emptySchema } from '@contember/schema-utils'
+import { DatabaseMetadata, emptyDatabaseMetadata } from '@contember/database'
 
 const modificationFactory = new ModificationHandlerFactory(ModificationHandlerFactory.defaultFactoryMap)
 const schemaMigrator = new SchemaMigrator(modificationFactory)
@@ -14,6 +15,7 @@ export interface TestContext {
 	updated: Partial<Schema>
 	sql: string
 	noDiff?: boolean
+	databaseMetadata?: DatabaseMetadata
 }
 
 export function testDiffSchemas(
@@ -61,7 +63,12 @@ export function testApplyDiff(
 	})
 }
 
-export function testGenerateSql(original: Partial<Schema>, diff: Migration.Modification[], expectedSql: string) {
+export function testGenerateSql(
+	original: Partial<Schema>,
+	diff: Migration.Modification[],
+	expectedSql: string,
+	databaseMetadata: DatabaseMetadata = emptyDatabaseMetadata,
+) {
 	let schema = { ...emptySchema, ...original }
 	const builder = createMigrationBuilder()
 	for (let { modification, ...data } of diff) {
@@ -70,7 +77,7 @@ export function testGenerateSql(original: Partial<Schema>, diff: Migration.Modif
 		})
 		modificationHandler.createSql(builder, {
 			systemSchema: 'system',
-			databaseMetadata: dummySchemaDatabaseMetadata,
+			databaseMetadata,
 			invalidateDatabaseMetadata: () => null,
 		})
 		schema = modificationHandler.getSchemaUpdater()({ schema })
@@ -79,7 +86,7 @@ export function testGenerateSql(original: Partial<Schema>, diff: Migration.Modif
 	assert.equal(actual, expectedSql)
 }
 
-export function testMigrations(title: string, { original, updated, diff, noDiff, sql }: TestContext) {
+export function testMigrations(title: string, { original, updated, diff, noDiff, sql, databaseMetadata }: TestContext) {
 	describe(title, () => {
 		it('diff schemas', () => {
 			if (noDiff) {
@@ -91,7 +98,7 @@ export function testMigrations(title: string, { original, updated, diff, noDiff,
 			testApplyDiff(original, updated, diff)
 		})
 		it('generate sql', () => {
-			testGenerateSql(original, diff, sql)
+			testGenerateSql(original, diff, sql, databaseMetadata)
 		})
 	})
 }