diff --git a/plugins/catalog-backend/src/tests/migrations.test.ts b/plugins/catalog-backend/src/tests/migrations.test.ts index bb840f7a21..f58b174f93 100644 --- a/plugins/catalog-backend/src/tests/migrations.test.ts +++ b/plugins/catalog-backend/src/tests/migrations.test.ts @@ -521,4 +521,87 @@ describe('migrations', () => { await knex.destroy(); }, ); + it.each(databases.eachSupportedId())( + '20250401200503_update_refresh_state_columns.js, %p', + async databaseId => { + const knex = await databases.init(databaseId); + + // Run migrations up to just before the target migration + await migrateUntilBefore( + knex, + '20250401200503_update_refresh_state_columns.js', + ); + + // Insert a row with data into the refresh_state table + await knex('refresh_state').insert({ + entity_id: 'test-id', + entity_ref: 'k:ns/test', + unprocessed_entity: JSON.stringify({ key: 'value' }), + cache: JSON.stringify({ cacheKey: 'cacheValue' }), + errors: '[]', + next_update_at: knex.fn.now(), + last_discovery_at: knex.fn.now(), + }); + + // Verify the data before the migration + const preMigrationData = await knex('refresh_state') + .where({ entity_id: 'test-id' }) + .first(); + expect(preMigrationData).toEqual( + expect.objectContaining({ + entity_id: 'test-id', + entity_ref: 'k:ns/test', + unprocessed_entity: JSON.stringify({ key: 'value' }), + cache: JSON.stringify({ cacheKey: 'cacheValue' }), + }), + ); + + // Run the migration + await migrateUpOnce(knex); + + // Verify the schema after the migration + const columnInfo = await knex('refresh_state').columnInfo(); + const expectedType = knex.client.config.client.includes('mysql') + ? 'longtext' + : 'text'; + expect(columnInfo.unprocessed_entity.type).toBe(expectedType); + expect(columnInfo.cache.type).toBe(expectedType); + + // Verify the data after the migration + const postMigrationData = await knex('refresh_state') + .where({ entity_id: 'test-id' }) + .first(); + expect(postMigrationData).toEqual( + expect.objectContaining({ + entity_id: 'test-id', + entity_ref: 'k:ns/test', + unprocessed_entity: JSON.stringify({ key: 'value' }), + cache: JSON.stringify({ cacheKey: 'cacheValue' }), + }), + ); + + // Roll back the migration + await migrateDownOnce(knex); + + // Verify the schema after rolling back + const revertedColumnInfo = await knex('refresh_state').columnInfo(); + expect(revertedColumnInfo.unprocessed_entity.type).toBe('text'); + expect(revertedColumnInfo.cache.type).toBe('text'); + + // Verify the data after rolling back + const postRollbackData = await knex('refresh_state') + .where({ entity_id: 'test-id' }) + .first(); + expect(postRollbackData).toEqual( + expect.objectContaining({ + entity_id: 'test-id', + entity_ref: 'k:ns/test', + unprocessed_entity: JSON.stringify({ key: 'value' }), + cache: JSON.stringify({ cacheKey: 'cacheValue' }), + }), + ); + + await knex.destroy(); + }, + ); });