DPP-1 test fix

Sphereon-Opensource · Sep 12, 2023 · dd09f38 · dd09f38
1 parent 8422945
commit dd09f38
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/packages/data-store/src/__tests__/contact.store.test.ts b/packages/data-store/src/__tests__/contact.store.test.ts
@@ -1,5 +1,4 @@
 import { DataSource } from 'typeorm'
-
 import { DataStoreMigrations, DataStoreContactEntities } from '../index'
 import { ContactStore } from '../contact/ContactStore'
 import {
@@ -1676,12 +1675,15 @@ describe('Contact store tests', (): void => {
     const resultPartyType: PartyType = await contactStore.getPartyType({ partyTypeId: savedPartyType.id })
     expect(resultPartyType).toBeDefined()
 
+    const includingMigrationPartyTypes: Array<PartyType> = await contactStore.getPartyTypes()
+    expect(includingMigrationPartyTypes.length).toEqual(2)
+
     await contactStore.removePartyType({ partyTypeId: savedPartyType.id })
 
     const result: Array<PartyType> = await contactStore.getPartyTypes()
 
     expect(result).toBeDefined()
-    expect(result.length).toEqual(0)
+    expect(result.length).toEqual(1)
   })
 
   it('should throw error when removing party type attached to contact', async (): Promise<void> => {