refactor(test): refactor mongodb tests

SmartOperatingBlock · Mar 1, 2023 · f0793f6 · f0793f6
1 parent 3d84e6b
commit f0793f6
Showing 1 changed file with 22 additions and 6 deletions.
diff --git a/src/test/kotlin/infrastructure/database/TestMongo.kt b/src/test/kotlin/infrastructure/database/TestMongo.kt
@@ -15,17 +15,33 @@ import io.kotest.matchers.shouldNotBe
 
 class TestMongo : StringSpec({
 
-    "Test user operations on mongo db" {
-        withMongo {
-            val mockUser = User("1234", "MockPassword!")
-
-            val mongoClient = MongoClient("mongodb://localhost:27017")
+    val mockUser = User("1234", "MockPassword!")
 
+    "Test user creation on mongo db" {
+        withMongo {
+            val mongoClient = MongoClient("mongodb://localhost:27017").also {
+                it.getDatabase("user_management").drop()
+            }
             mongoClient.createUser(mockUser) shouldNotBe null
+        }
+    }
 
+    "Test user deletion on mongo db" {
+        withMongo {
+            val mongoClient = MongoClient("mongodb://localhost:27017").also {
+                it.getDatabase("user_management").drop()
+            }
+            mongoClient.createUser(mockUser)
             mongoClient.deleteUser(mockUser.userId) shouldBe true
+        }
+    }
 
-            mongoClient.createUser(mockUser) shouldNotBe null
+    "Test user retrieve on mongo db" {
+        withMongo {
+            val mongoClient = MongoClient("mongodb://localhost:27017").also {
+                it.getDatabase("user_management").drop()
+            }
+            mongoClient.createUser(mockUser)
             mongoClient.getUser(mockUser.userId).also {
                 it shouldNotBe null
                 it?.userId shouldBe mockUser.userId