@credo-ts/drizzle-storage 0.6.0-pr-2324-20250708123651 → 0.6.0-pr-2324-20250709075226

package/cli/drizzle.config.ts

@@ -48,7 +48,7 @@ if (drizzleDialect === 'sqlite') {
           dbCredentials: { url: drizzleDatabaseUrl },
         }
       : {
-          driver: 'pglite',
+          driver: 'node-postgres',
         }),
   })
 }

package/cli-build/drizzle.config.js

@@ -46,7 +46,7 @@ else {
                 dbCredentials: { url: drizzleDatabaseUrl },
             }
             : {
-                driver: 'pglite',
+                driver: 'node-postgres',
             }),
     });
 }

package/cli-build/drizzle.config.js.map

@@ -1 +1 @@
-{"version":3,"file":"drizzle.config.js","sourceRoot":"","sources":["../cli/drizzle.config.ts"],"names":[],"mappings":";;AAAA,6CAAkD;AAElD,MAAM,kBAAkB,GAAG,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAA;AAE3D,MAAM,cAAc,GAAG,OAAO,CAAC,GAAG,CAAC,eAAe,CAAA;AAClD,IAAI,CAAC,cAAc,IAAI,CAAC,cAAc,KAAK,QAAQ,IAAI,cAAc,KAAK,UAAU,CAAC,EAAE,CAAC;IACtF,MAAM,IAAI,KAAK,CAAC,gFAAgF,CAAC,CAAA;AACnG,CAAC;AAED,MAAM,iBAAiB,GAAG,OAAO,CAAC,GAAG,CAAC,mBAAmB,CAAA;AACzD,MAAM,uBAAuB,GAAG,OAAO,CAAC,GAAG,CAAC,yBAAyB,CAAA;AACrE,MAAM,sBAAsB,GAAG,OAAO,CAAC,GAAG,CAAC,wBAAwB,CAAA;AAEnE,IAAI,aAAqB,CAAA;AAEzB,IAAI,cAAc,KAAK,QAAQ,EAAE,CAAC;IAChC,aAAa,GAAG,IAAA,0BAAY,EAAC;QAC3B,OAAO,EAAE,cAAc;QACvB,gFAAgF;QAChF,gEAAgE;QAChE,MAAM,EAAE,iBAAiB;QACzB,GAAG,EAAE,uBAAuB;QAC5B,UAAU,EAAE,sBAAsB;YAChC,CAAC,CAAC;gBACE,KAAK,EAAE,sBAAsB;aAC9B;YACH,CAAC,CAAC,SAAS;QACb,GAAG,CAAC,kBAAkB;YACpB,CAAC,CAAC;gBACE,aAAa,EAAE,EAAE,GAAG,EAAE,kBAAkB,EAAE;aAC3C;YACH,CAAC,CAAC;gBACE,MAAM,EAAE,MAAM;aACf,CAAC;KACP,CAAC,CAAA;AACJ,CAAC;KAAM,CAAC;IACN,aAAa,GAAG,IAAA,0BAAY,EAAC;QAC3B,OAAO,EAAE,YAAY;QACrB,MAAM,EAAE,iBAAiB;QACzB,GAAG,EAAE,uBAAuB;QAC5B,UAAU,EAAE,sBAAsB;YAChC,CAAC,CAAC;gBACE,KAAK,EAAE,sBAAsB;aAC9B;YACH,CAAC,CAAC,SAAS;QACb,GAAG,CAAC,kBAAkB;YACpB,CAAC,CAAC;gBACE,aAAa,EAAE,EAAE,GAAG,EAAE,kBAAkB,EAAE;aAC3C;YACH,CAAC,CAAC;gBACE,MAAM,EAAE,QAAQ;aACjB,CAAC;KACP,CAAC,CAAA;AACJ,CAAC;AAED,kBAAe,aAAa,CAAA"}
+{"version":3,"file":"drizzle.config.js","sourceRoot":"","sources":["../cli/drizzle.config.ts"],"names":[],"mappings":";;AAAA,6CAAkD;AAElD,MAAM,kBAAkB,GAAG,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAA;AAE3D,MAAM,cAAc,GAAG,OAAO,CAAC,GAAG,CAAC,eAAe,CAAA;AAClD,IAAI,CAAC,cAAc,IAAI,CAAC,cAAc,KAAK,QAAQ,IAAI,cAAc,KAAK,UAAU,CAAC,EAAE,CAAC;IACtF,MAAM,IAAI,KAAK,CAAC,gFAAgF,CAAC,CAAA;AACnG,CAAC;AAED,MAAM,iBAAiB,GAAG,OAAO,CAAC,GAAG,CAAC,mBAAmB,CAAA;AACzD,MAAM,uBAAuB,GAAG,OAAO,CAAC,GAAG,CAAC,yBAAyB,CAAA;AACrE,MAAM,sBAAsB,GAAG,OAAO,CAAC,GAAG,CAAC,wBAAwB,CAAA;AAEnE,IAAI,aAAqB,CAAA;AAEzB,IAAI,cAAc,KAAK,QAAQ,EAAE,CAAC;IAChC,aAAa,GAAG,IAAA,0BAAY,EAAC;QAC3B,OAAO,EAAE,cAAc;QACvB,gFAAgF;QAChF,gEAAgE;QAChE,MAAM,EAAE,iBAAiB;QACzB,GAAG,EAAE,uBAAuB;QAC5B,UAAU,EAAE,sBAAsB;YAChC,CAAC,CAAC;gBACE,KAAK,EAAE,sBAAsB;aAC9B;YACH,CAAC,CAAC,SAAS;QACb,GAAG,CAAC,kBAAkB;YACpB,CAAC,CAAC;gBACE,aAAa,EAAE,EAAE,GAAG,EAAE,kBAAkB,EAAE;aAC3C;YACH,CAAC,CAAC;gBACE,MAAM,EAAE,MAAM;aACf,CAAC;KACP,CAAC,CAAA;AACJ,CAAC;KAAM,CAAC;IACN,aAAa,GAAG,IAAA,0BAAY,EAAC;QAC3B,OAAO,EAAE,YAAY;QACrB,MAAM,EAAE,iBAAiB;QACzB,GAAG,EAAE,uBAAuB;QAC5B,UAAU,EAAE,sBAAsB;YAChC,CAAC,CAAC;gBACE,KAAK,EAAE,sBAAsB;aAC9B;YACH,CAAC,CAAC,SAAS;QACb,GAAG,CAAC,kBAAkB;YACpB,CAAC,CAAC;gBACE,aAAa,EAAE,EAAE,GAAG,EAAE,kBAAkB,EAAE;aAC3C;YACH,CAAC,CAAC;gBACE,MAAM,EAAE,eAAe;aACxB,CAAC;KACP,CAAC,CAAA;AACJ,CAAC;AAED,kBAAe,aAAa,CAAA"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@credo-ts/drizzle-storage",
-  "version": "0.6.0-pr-2324-20250708123651",
+  "version": "0.6.0-pr-2324-20250709075226",
   "files": [
     "build",
     "cli-build",
@@ -52,24 +52,24 @@
     "ts-node": "^10.9.2",
     "tsyringe": "^4.8.0",
     "zod": "^3.24.2",
-    "@credo-ts/action-menu": "0.6.0-pr-2324-20250708123651",
-    "@credo-ts/anoncreds": "0.6.0-pr-2324-20250708123651",
-    "@credo-ts/core": "0.6.0-pr-2324-20250708123651",
-    "@credo-ts/didcomm": "0.6.0-pr-2324-20250708123651",
-    "@credo-ts/drpc": "0.6.0-pr-2324-20250708123651",
-    "@credo-ts/openid4vc": "0.6.0-pr-2324-20250708123651",
-    "@credo-ts/question-answer": "0.6.0-pr-2324-20250708123651",
-    "@credo-ts/tenants": "0.6.0-pr-2324-20250708123651"
+    "@credo-ts/action-menu": "0.6.0-pr-2324-20250709075226",
+    "@credo-ts/anoncreds": "0.6.0-pr-2324-20250709075226",
+    "@credo-ts/core": "0.6.0-pr-2324-20250709075226",
+    "@credo-ts/didcomm": "0.6.0-pr-2324-20250709075226",
+    "@credo-ts/drpc": "0.6.0-pr-2324-20250709075226",
+    "@credo-ts/openid4vc": "0.6.0-pr-2324-20250709075226",
+    "@credo-ts/question-answer": "0.6.0-pr-2324-20250709075226",
+    "@credo-ts/tenants": "0.6.0-pr-2324-20250709075226"
   },
   "peerDependencies": {
     "drizzle-orm": "^0.44.2"
   },
   "devDependencies": {
-    "@electric-sql/pglite": "^0.3.1",
     "@libsql/client": "^0.15.6",
+    "@types/pg": "^8.15.4",
     "@types/yargs": "^17.0.33",
-    "pg": "^8.16.3",
     "drizzle-orm": "^0.44.2",
+    "pg": "^8.16.3",
     "rimraf": "^4.4.0",
     "typescript": "~5.5.2"
   },
@@ -84,6 +84,6 @@
     "cli:dev:migrate:sqlite": "pnpm cli:dev migrate --dialect sqlite --database-url \"file:local.db\"",
     "cli:dev:studio:postgres": "pnpm cli:dev studio --dialect postgres --database-url \"postgresql://postgres:postgres@localhost:5432/postgres\"",
     "cli:dev:studio:sqlite": "pnpm cli:dev studio --dialect sqlite --database-url \"file:local.db\"",
-    "test": "NODE_OPTIONS=--experimental-vm-modules jest"
+    "test": "jest"
   }
 }

package/src/adapter/__tests__/{queryToDrizzlePostgres.test.ts → queryToDrizzlePostgres.drizzle.e2e.test.ts}

@@ -1,8 +1,6 @@
 import { ConnectionRecord, DidExchangeRole, DidExchangeState } from '@credo-ts/didcomm'
 import { pushSchema } from 'drizzle-kit/api'
-import { PgDatabase } from 'drizzle-orm/pg-core'
-import { drizzle } from 'drizzle-orm/pglite'
-import { DrizzlePostgresDatabase } from '../../DrizzleDatabase'
+import { DrizzlePostgresTestDatabase, createDrizzlePostgresTestDatabase } from '../../../tests/testDatabase'
 import * as coreContextSchema from '../../core/context/postgres'
 import * as didcommConnectionSchema from '../../didcomm/connection/postgres'
 import { queryToDrizzlePostgres } from '../queryToDrizzlePostgres'
@@ -10,23 +8,23 @@ import { queryToDrizzlePostgres } from '../queryToDrizzlePostgres'
 const { didcommConnection } = didcommConnectionSchema
 const { context } = coreContextSchema
 
-const db = drizzle('memory://', {
-  schema: { ...didcommConnectionSchema, ...coreContextSchema },
-}) as unknown as DrizzlePostgresDatabase<typeof didcommConnectionSchema>
-
 describe('queryToDrizzlePostgres', () => {
+  let postgresDatabase: DrizzlePostgresTestDatabase
+
   beforeAll(async () => {
+    postgresDatabase = await createDrizzlePostgresTestDatabase()
+
     const { apply } = await pushSchema(
       { ...didcommConnectionSchema, ...coreContextSchema },
-      // biome-ignore lint/suspicious/noExplicitAny: <explanation>
-      db as unknown as PgDatabase<any>
+      // @ts-ignore
+      postgresDatabase.drizzle
     )
     await apply()
 
-    await db.insert(context).values({
+    await postgresDatabase.drizzle.insert(context).values({
       contextCorrelationId: 'b2fc0867-d0d1-4182-ade7-813b695d43c2',
     })
-    await db.insert(didcommConnection).values({
+    await postgresDatabase.drizzle.insert(didcommConnection).values({
       createdAt: new Date(),
       updatedAt: new Date(),
       contextCorrelationId: 'b2fc0867-d0d1-4182-ade7-813b695d43c2',
@@ -42,9 +40,13 @@ describe('queryToDrizzlePostgres', () => {
     })
   })
 
+  afterAll(async () => {
+    await postgresDatabase.teardown()
+  })
+
   test('should correctly query column', async () => {
     expect(
-      await db
+      await postgresDatabase.drizzle
         .select()
         .from(didcommConnection)
         .where(
@@ -58,7 +60,7 @@ describe('queryToDrizzlePostgres', () => {
     ).toHaveLength(1)
 
     expect(
-      await db
+      await postgresDatabase.drizzle
         .select()
         .from(didcommConnection)
         .where(
@@ -74,7 +76,7 @@ describe('queryToDrizzlePostgres', () => {
 
   test('should correctly query array value column', async () => {
     expect(
-      await db
+      await postgresDatabase.drizzle
         .select()
         .from(didcommConnection)
         .where(
@@ -89,7 +91,7 @@ describe('queryToDrizzlePostgres', () => {
     ).toHaveLength(1)
 
     expect(
-      await db
+      await postgresDatabase.drizzle
         .select()
         .from(didcommConnection)
         .where(
@@ -104,7 +106,7 @@ describe('queryToDrizzlePostgres', () => {
     ).toHaveLength(0)
 
     expect(
-      await db
+      await postgresDatabase.drizzle
         .select()
         .from(didcommConnection)
         .where(
@@ -121,7 +123,7 @@ describe('queryToDrizzlePostgres', () => {
 
   test('should correctly query custom tag column value', async () => {
     expect(
-      await db
+      await postgresDatabase.drizzle
         .select()
         .from(didcommConnection)
         .where(
@@ -136,7 +138,7 @@ describe('queryToDrizzlePostgres', () => {
     ).toHaveLength(1)
 
     expect(
-      await db
+      await postgresDatabase.drizzle
         .select()
         .from(didcommConnection)
         .where(
@@ -153,7 +155,7 @@ describe('queryToDrizzlePostgres', () => {
 
   test('should correctly query custom tag column array value', async () => {
     expect(
-      await db
+      await postgresDatabase.drizzle
         .select()
         .from(didcommConnection)
         .where(
@@ -168,7 +170,7 @@ describe('queryToDrizzlePostgres', () => {
     ).toHaveLength(1)
 
     expect(
-      await db
+      await postgresDatabase.drizzle
         .select()
         .from(didcommConnection)
         .where(
@@ -183,7 +185,7 @@ describe('queryToDrizzlePostgres', () => {
     ).toHaveLength(0)
 
     expect(
-      await db
+      await postgresDatabase.drizzle
         .select()
         .from(didcommConnection)
         .where(
@@ -200,7 +202,7 @@ describe('queryToDrizzlePostgres', () => {
 
   test('should correctly query with and/or', async () => {
     expect(
-      await db
+      await postgresDatabase.drizzle
         .select()
         .from(didcommConnection)
         .where(
@@ -227,7 +229,7 @@ describe('queryToDrizzlePostgres', () => {
     ).toHaveLength(1)
 
     expect(
-      await db
+      await postgresDatabase.drizzle
         .select()
         .from(didcommConnection)
         .where(
@@ -259,7 +261,7 @@ describe('queryToDrizzlePostgres', () => {
 
   test('should correctly query with not', async () => {
     expect(
-      await db
+      await postgresDatabase.drizzle
         .select()
         .from(didcommConnection)
         .where(
@@ -291,7 +293,7 @@ describe('queryToDrizzlePostgres', () => {
     ).toHaveLength(1)
 
     expect(
-      await db
+      await postgresDatabase.drizzle
         .select()
         .from(didcommConnection)
         .where(

package/src/anoncreds/schema/__tests__/{schema.test.ts → schema.drizzle.e2e.test.ts}

@@ -1,14 +1,18 @@
 import { AnonCredsSchemaRecord, AnonCredsSchemaRepository } from '@credo-ts/anoncreds'
-import { Agent, JsonTransformer } from '@credo-ts/core'
+import { JsonTransformer } from '@credo-ts/core'
 
-import { setupDrizzleRecordTest } from '../../../../tests/testDatabase'
+import { DrizzleRecordTest, setupDrizzleRecordTest } from '../../../../tests/testDatabase'
 import { anonCredsSchemaDrizzleRecord } from '../index'
 
 describe.each(['postgres', 'sqlite'] as const)('AnonCredsSchemaRecord with %s', (type) => {
-  let agent: Agent
+  let recordTest: DrizzleRecordTest
 
   beforeAll(async () => {
-    agent = await setupDrizzleRecordTest(type, anonCredsSchemaDrizzleRecord)
+    recordTest = await setupDrizzleRecordTest(type, anonCredsSchemaDrizzleRecord)
+  })
+
+  afterAll(async () => {
+    await recordTest.teardown()
   })
 
   test('create, retrieve, update, query and delete schema record', async () => {
@@ -28,20 +32,20 @@ describe.each(['postgres', 'sqlite'] as const)('AnonCredsSchemaRecord with %s',
       },
       AnonCredsSchemaRecord
     )
-    const anoncredsSchemaRepository = agent.context.resolve(AnonCredsSchemaRepository)
+    const anoncredsSchemaRepository = recordTest.agent.context.resolve(AnonCredsSchemaRepository)
 
-    await anoncredsSchemaRepository.save(agent.context, schema)
+    await anoncredsSchemaRepository.save(recordTest.agent.context, schema)
 
-    const schema2 = await anoncredsSchemaRepository.findById(agent.context, schema.id)
+    const schema2 = await anoncredsSchemaRepository.findById(recordTest.agent.context, schema.id)
     expect(schema).toEqual(schema2)
 
     schema.setTags({
       myCustomTag: 'hello',
       isMorning: false,
     })
-    await anoncredsSchemaRepository.update(agent.context, schema)
+    await anoncredsSchemaRepository.update(recordTest.agent.context, schema)
 
-    const [schema3] = await anoncredsSchemaRepository.findByQuery(agent.context, {
+    const [schema3] = await anoncredsSchemaRepository.findByQuery(recordTest.agent.context, {
       // TODO: we should allow null values in the query
       // unqualifiedSchemaId: null,
       isMorning: false,
@@ -54,13 +58,13 @@ describe.each(['postgres', 'sqlite'] as const)('AnonCredsSchemaRecord with %s',
     expect(schema3).toEqual(schema)
 
     expect(
-      await anoncredsSchemaRepository.findByQuery(agent.context, {
+      await anoncredsSchemaRepository.findByQuery(recordTest.agent.context, {
         myCustomTag: 'not-hello',
       })
     ).toHaveLength(0)
 
-    await anoncredsSchemaRepository.deleteById(agent.context, schema.id)
+    await anoncredsSchemaRepository.deleteById(recordTest.agent.context, schema.id)
 
-    expect(await anoncredsSchemaRepository.findByQuery(agent.context, {})).toHaveLength(0)
+    expect(await anoncredsSchemaRepository.findByQuery(recordTest.agent.context, {})).toHaveLength(0)
   })
 })

package/src/didcomm/credential-exchange/__tests__/{credentialExchange.test.ts → credentialExchange.drizzle.e2e.test.ts}

@@ -1,4 +1,4 @@
-import { Agent, JsonTransformer } from '@credo-ts/core'
+import { JsonTransformer } from '@credo-ts/core'
 
 import {
   AutoAcceptCredential,
@@ -7,14 +7,18 @@ import {
   CredentialRole,
   CredentialState,
 } from '@credo-ts/didcomm'
-import { setupDrizzleRecordTest } from '../../../../tests/testDatabase'
+import { DrizzleRecordTest, setupDrizzleRecordTest } from '../../../../tests/testDatabase'
 import { didcommCredentialExchangeDrizzleRecord } from '../index'
 
 describe.each(['postgres', 'sqlite'] as const)('CredentialExchangeRecord with %s', (type) => {
-  let agent: Agent
+  let recordTest: DrizzleRecordTest
 
   beforeAll(async () => {
-    agent = await setupDrizzleRecordTest(type, didcommCredentialExchangeDrizzleRecord)
+    recordTest = await setupDrizzleRecordTest(type, didcommCredentialExchangeDrizzleRecord)
+  })
+
+  afterAll(async () => {
+    await recordTest.teardown()
   })
 
   test('create, retrieve, update, query and delete credential exchange record', async () => {
@@ -58,26 +62,29 @@ describe.each(['postgres', 'sqlite'] as const)('CredentialExchangeRecord with %s
       },
       CredentialExchangeRecord
     )
-    const credentialExchangeRepository = agent.context.resolve(CredentialRepository)
+    const credentialExchangeRepository = recordTest.agent.context.resolve(CredentialRepository)
 
-    await credentialExchangeRepository.save(agent.context, credentialExchange)
+    await credentialExchangeRepository.save(recordTest.agent.context, credentialExchange)
 
-    const credentialExchange2 = await credentialExchangeRepository.findById(agent.context, credentialExchange.id)
+    const credentialExchange2 = await credentialExchangeRepository.findById(
+      recordTest.agent.context,
+      credentialExchange.id
+    )
     expect(credentialExchange).toEqual(credentialExchange2)
 
     credentialExchange.setTags({
       myCustomTag: 'hello',
       isMorning: false,
     })
-    await credentialExchangeRepository.update(agent.context, credentialExchange)
+    await credentialExchangeRepository.update(recordTest.agent.context, credentialExchange)
 
-    const [credentialExchange3] = await credentialExchangeRepository.findByQuery(agent.context, {
+    const [credentialExchange3] = await credentialExchangeRepository.findByQuery(recordTest.agent.context, {
       isMorning: false,
       credentialIds: ['8633f56d-abc9-4229-ba09-9ca6611ad8e4'],
     })
     expect(credentialExchange3).toEqual(credentialExchange)
 
-    const [credentialExchange4] = await credentialExchangeRepository.findByQuery(agent.context, {
+    const [credentialExchange4] = await credentialExchangeRepository.findByQuery(recordTest.agent.context, {
       isMorning: false,
       // Id does not exist
       credentialIds: ['a0b7c554-0beb-4a4c-808a-fd4495241770'],
@@ -85,13 +92,13 @@ describe.each(['postgres', 'sqlite'] as const)('CredentialExchangeRecord with %s
     expect(credentialExchange4).toBeUndefined()
 
     expect(
-      await credentialExchangeRepository.findByQuery(agent.context, {
+      await credentialExchangeRepository.findByQuery(recordTest.agent.context, {
         myCustomTag: 'not-hello',
       })
     ).toHaveLength(0)
 
-    await credentialExchangeRepository.deleteById(agent.context, credentialExchange.id)
+    await credentialExchangeRepository.deleteById(recordTest.agent.context, credentialExchange.id)
 
-    expect(await credentialExchangeRepository.findByQuery(agent.context, {})).toHaveLength(0)
+    expect(await credentialExchangeRepository.findByQuery(recordTest.agent.context, {})).toHaveLength(0)
   })
 })

package/src/didcomm/out-of-band/__tests__/{outOfBand.test.ts → outOfBand.drizzle.e2e.test.ts}

@@ -1,14 +1,18 @@
-import { Agent, JsonTransformer } from '@credo-ts/core'
+import { JsonTransformer } from '@credo-ts/core'
 
 import { OutOfBandRecord, OutOfBandRepository } from '@credo-ts/didcomm'
-import { setupDrizzleRecordTest } from '../../../../tests/testDatabase'
+import { DrizzleRecordTest, setupDrizzleRecordTest } from '../../../../tests/testDatabase'
 import { didcommOutOfBandDrizzleRecord } from '../index'
 
 describe.each(['postgres', 'sqlite'] as const)('OutOfBandRecord with %s', (type) => {
-  let agent: Agent
+  let recordTest: DrizzleRecordTest
 
   beforeAll(async () => {
-    agent = await setupDrizzleRecordTest(type, didcommOutOfBandDrizzleRecord)
+    recordTest = await setupDrizzleRecordTest(type, didcommOutOfBandDrizzleRecord)
+  })
+
+  afterAll(async () => {
+    await recordTest.teardown()
   })
 
   test('create, retrieve, update, query and delete out of band record', async () => {
@@ -43,20 +47,20 @@ describe.each(['postgres', 'sqlite'] as const)('OutOfBandRecord with %s', (type)
       },
       OutOfBandRecord
     )
-    const outOfBandRepository = agent.context.resolve(OutOfBandRepository)
+    const outOfBandRepository = recordTest.agent.context.resolve(OutOfBandRepository)
 
-    await outOfBandRepository.save(agent.context, outOfBand)
+    await outOfBandRepository.save(recordTest.agent.context, outOfBand)
 
-    const outOfBand2 = await outOfBandRepository.findById(agent.context, outOfBand.id)
+    const outOfBand2 = await outOfBandRepository.findById(recordTest.agent.context, outOfBand.id)
     expect(outOfBand.toJSON()).toEqual(outOfBand2?.toJSON())
 
     outOfBand.setTags({
       myCustomTag: 'hello',
       isMorning: false,
     })
-    await outOfBandRepository.update(agent.context, outOfBand)
+    await outOfBandRepository.update(recordTest.agent.context, outOfBand)
 
-    const [outOfBand3] = await outOfBandRepository.findByQuery(agent.context, {
+    const [outOfBand3] = await outOfBandRepository.findByQuery(recordTest.agent.context, {
       isMorning: false,
 
       // Tests custom tag mapping (invitationId -> outOfBandInvitation[@id])
@@ -64,7 +68,7 @@ describe.each(['postgres', 'sqlite'] as const)('OutOfBandRecord with %s', (type)
     })
     expect(outOfBand3.toJSON()).toEqual(outOfBand.toJSON())
 
-    const [outOfBand4] = await outOfBandRepository.findByQuery(agent.context, {
+    const [outOfBand4] = await outOfBandRepository.findByQuery(recordTest.agent.context, {
       isMorning: false,
       // Id does not exist
       invitationId: 'a0b7c554-0beb-4a4c-808a-fd4495241770',
@@ -72,13 +76,13 @@ describe.each(['postgres', 'sqlite'] as const)('OutOfBandRecord with %s', (type)
     expect(outOfBand4).toBeUndefined()
 
     expect(
-      await outOfBandRepository.findByQuery(agent.context, {
+      await outOfBandRepository.findByQuery(recordTest.agent.context, {
         myCustomTag: 'not-hello',
       })
     ).toHaveLength(0)
 
-    await outOfBandRepository.deleteById(agent.context, outOfBand.id)
+    await outOfBandRepository.deleteById(recordTest.agent.context, outOfBand.id)
 
-    expect(await outOfBandRepository.findByQuery(agent.context, {})).toHaveLength(0)
+    expect(await outOfBandRepository.findByQuery(recordTest.agent.context, {})).toHaveLength(0)
   })
 })