test: dropAll for mongo migration tests

cordiverse · Feb 21, 2024 · 5833775 · 5833775
1 parent 3a356d2
commit 5833775
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/packages/core/package.json b/packages/core/package.json
@@ -42,7 +42,7 @@
     "postgres"
   ],
   "dependencies": {
-    "cordis": "^3.9.2",
+    "cordis": "^3.11.0",
     "cosmokit": "^1.5.2"
   }
 }
diff --git a/packages/core/src/database.ts b/packages/core/src/database.ts
@@ -47,7 +47,10 @@ export namespace Join2 {
 
 const kTransaction = Symbol('transaction')
 
-export class Database<S = any, C extends Context = Context> extends Service<C> {
+export class Database<S = any, C extends Context = Context> extends Service<undefined, C> {
+  static [Service.provide] = 'model'
+  static [Service.immediate] = true
+
   public tables: { [K in Keys<S>]: Model<S[K]> } = Object.create(null)
   public drivers: Record<keyof any, Driver> = Object.create(null)
   public migrating = false
@@ -56,10 +59,6 @@ export class Database<S = any, C extends Context = Context> extends Service<C> {
 
   private stashed = new Set<string>()
 
-  constructor(ctx?: C) {
-    super(ctx, 'model', true)
-  }
-
   async connect<T = undefined>(driver: Driver.Constructor<T>, ...args: Spread<T>) {
     this.ctx.plugin(driver, args[0] as any)
     await this.ctx.start()

diff --git a/packages/mongo/tests/migration.spec.ts b/packages/mongo/tests/migration.spec.ts
@@ -67,6 +67,7 @@ describe('@minatojs/driver-mongo/migrate-virtualKey', () => {
   })
 
   afterEach(async () => {
+    await database.dropAll()
     fork?.dispose()
     logger.level = 2
   })