diff --git a/plugins/catalog-backend/src/next/database/DefaultProcessingDatabase.test.ts b/plugins/catalog-backend/src/next/database/DefaultProcessingDatabase.test.ts index 05c10da188..2a28951e6b 100644 --- a/plugins/catalog-backend/src/next/database/DefaultProcessingDatabase.test.ts +++ b/plugins/catalog-backend/src/next/database/DefaultProcessingDatabase.test.ts @@ -395,15 +395,24 @@ describe('Default Processing Database', () => { expect(mockLogger.warn).not.toHaveBeenCalled(); } - const [, unprocessed] = await knexTx( + const states = await knexTx( 'refresh_state', ).select(); - expect(unprocessed).toEqual( - expect.objectContaining({ - entity_ref: 'component:default/1', - location_key: step.expectedLocationKey, - }), + expect(states).toEqual( + expect.arrayContaining([ + expect.objectContaining({ + entity_ref: 'component:default/1', + location_key: step.expectedLocationKey, + }), + expect.objectContaining({ + entity_ref: 'location:default/fakelocation', + location_key: null, + }), + ]), ); + const unprocessed = states.find( + state => state.entity_ref === 'component:default/1', + )!; const entity = JSON.parse(unprocessed.unprocessed_entity) as Entity; expect(entity.spec?.type).toEqual(step.expectedTestKey);