From eacf4decaba8867da5a3b9a34e48f721c1fd4037 Mon Sep 17 00:00:00 2001
From: Paul Makles <me@insrt.uk>
Date: Sat, 22 Apr 2023 16:01:07 +0100
Subject: [PATCH] refactor: clean up database before and after tests

---
 crates/core/database/src/lib.rs | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/crates/core/database/src/lib.rs b/crates/core/database/src/lib.rs
index 625563f6..e53c8231 100644
--- a/crates/core/database/src/lib.rs
+++ b/crates/core/database/src/lib.rs
@@ -63,13 +63,12 @@ macro_rules! database_test {
         .await
         .expect("Database connection failed.");
 
+        db.drop_database().await;
+
         #[allow(clippy::redundant_closure_call)]
         (|$db: $crate::Database| $test)(db.clone()).await;
 
-        match db {
-            $crate::Database::Reference(_) => {}
-            $crate::Database::MongoDb(db) => db.0.database(&db.1).drop(None).await.unwrap(),
-        }
+        db.drop_database().await
     };
 }