Function getTestMigration

packages/tlschema/src/__tests__/migrationTestUtils.ts:19–46

(migrationId: MigrationId)

}

export function getTestMigration(migrationId: MigrationId) {
	const migration = testSchema.sortedMigrations.find((m) => m.id === migrationId) as Migration
	if (!migration) {
		throw new Error(`Migration ${migrationId} not found`)
	}
	return {
		id: migrationId,
		up: (stuff: any) => {
			nextNanoId = 0
			if (migration.scope === 'record' || migration.scope === 'store') {
				const result = structuredClone(stuff)
				return migration.up(result) ?? result
			}
			const storage =
				typeof stuff.entries === 'function' ? stuff : new Map(Object.entries(stuff).map(devFreeze))
			migration.up(storage)
			return typeof stuff.entries === 'function' ? storage : Object.fromEntries(storage.entries())
		},
		down: (stuff: any) => {
			nextNanoId = 0
			if (typeof migration.down !== 'function') {
				throw new Error(`Migration ${migrationId} does not have a down function`)
			}
			const result = structuredClone(stuff)
			return migration.down(result) ?? result
		},
	}
}