refactor: fix database test

Signed-off-by: Jeroen Branje <[email protected]>
ASCS-eV · Jan 16, 2024 · 58c7e3e · 58c7e3e
1 parent 0be8149
commit 58c7e3e
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/apps/envited.ascs.digital/common/database/database.test.ts b/apps/envited.ascs.digital/common/database/database.test.ts
@@ -1,4 +1,5 @@
 import { initDb } from './database'
+import * as schema from './schema'
 
 describe('common/database', () => {
   describe('database', () => {
@@ -26,7 +27,7 @@ describe('common/database', () => {
       const db = await initDb({ drizzle, postgres, getSecret })()
 
       expect(postgres).toHaveBeenCalledWith(config)
-      expect(drizzle).toHaveBeenCalledWith(postgres(config))
+      expect(drizzle).toHaveBeenCalledWith(postgres(config), { schema })
       expect(getSecret).not.toHaveBeenCalled()
       expect(db).toEqual('DB_CONNECTION')
     })
@@ -59,7 +60,7 @@ describe('common/database', () => {
       const db = await initDb({ drizzle, postgres, getSecret })()
 
       expect(postgres).toHaveBeenCalledWith(config)
-      expect(drizzle).toHaveBeenCalledWith(postgres(config))
+      expect(drizzle).toHaveBeenCalledWith(postgres(config), { schema })
       expect(getSecret).toHaveBeenCalled()
       expect(db).toEqual('DB_CONNECTION')
     })