fix: drop database after each test

danielpza · Sep 7, 2024 · b886a67 · b886a67
1 parent bf75cb9
commit b886a67
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/src/setupFile.ts b/src/setupFile.ts
@@ -1,5 +1,5 @@
 import { Db, MongoClient } from "mongodb";
-import { afterAll, beforeAll, beforeEach, inject } from "vitest";
+import { afterAll, afterEach, beforeAll, beforeEach, inject } from "vitest";
 // hack to keep imported vitest types
 export type { TestContext } from "vitest";
 // hack to fix pnpm build issue
@@ -29,5 +29,10 @@ afterAll(async () => {
 
 beforeEach(async (context) => {
   const db = mongoClient.db(randomUUID());
-  Object.assign(context, { mongoClient, db });
+  context.db = db;
+  context.mongoClient = mongoClient;
+});
+
+afterEach(async (context) => {
+  await context.db.dropDatabase();
 });