fix: fix test initialization

snuhcs-course · Dec 10, 2023 · 7f45d81 · 7f45d81
1 parent 23cb64b
commit 7f45d81
Show file tree

Hide file tree

Showing 4 changed files with 37 additions and 2 deletions.
diff --git a/springboot/src/test/kotlin/com/goliath/emojihub/springboot/domain/emoji/dao/EmojiDaoTest.kt b/springboot/src/test/kotlin/com/goliath/emojihub/springboot/domain/emoji/dao/EmojiDaoTest.kt
@@ -79,6 +79,12 @@ internal class EmojiDaoTest {
             for (emoji in testDto.emojiList) {
                 testDB.collection(EMOJI_COLLECTION_NAME.string).document(emoji.id).set(emoji)
             }
+            var docs = testDB.collection(EMOJI_COLLECTION_NAME.string).get().get().documents
+            var a = 1
+            while (docs.size != testDto.emojiList.size && a <= 5) {
+                docs = testDB.collection(EMOJI_COLLECTION_NAME.string).get().get().documents
+                a++
+            }
         }
     }
 
@@ -236,6 +242,13 @@ internal class EmojiDaoTest {
         testDB.collection(EMOJI_COLLECTION_NAME.string)
             .document(emojiId)
             .set(emojiList[3])
+        result = emojiDao.existsEmoji(emojiId)
+        var b = 1
+        while (!result && b <= 5) {
+            result = emojiDao.existsEmoji(emojiId)
+            b++
+        }
+        assertEquals(result, true)
     }
 
     @Test

diff --git a/springboot/src/test/kotlin/com/goliath/emojihub/springboot/domain/post/dao/PostDaoTest.kt b/springboot/src/test/kotlin/com/goliath/emojihub/springboot/domain/post/dao/PostDaoTest.kt
@@ -65,6 +65,12 @@ internal class PostDaoTest {
             for (post in testDto.postList) {
                 testDB.collection(POST_COLLECTION_NAME.string).document(post.id).set(post)
             }
+            var docs = testDB.collection(POST_COLLECTION_NAME.string).get().get().documents
+            var a = 1
+            while (docs.size != testDto.postList.size && a <= 5) {
+                docs = testDB.collection(POST_COLLECTION_NAME.string).get().get().documents
+                a++
+            }
         }
     }
 
@@ -124,8 +130,12 @@ internal class PostDaoTest {
         val index = 1
         val count = testDto.postSize
         val postListForUser = mutableListOf<PostDto>()
-        postListForUser.add(postList[1])
-        postListForUser.add(postList[0])
+        for (post in testDto.postList) {
+            if (post.created_by == username) {
+                postListForUser.add(post)
+            }
+        }
+        postListForUser.sortByDescending { it.created_at }
         Mockito.`when`(db.collection(POST_COLLECTION_NAME.string))
             .thenReturn(testDB.collection(POST_COLLECTION_NAME.string))
 

diff --git a/...ot/src/test/kotlin/com/goliath/emojihub/springboot/domain/reaction/dao/ReactionDaoTest.kt b/...ot/src/test/kotlin/com/goliath/emojihub/springboot/domain/reaction/dao/ReactionDaoTest.kt
@@ -62,6 +62,12 @@ internal class ReactionDaoTest {
             for (reaction in testDto.reactionList) {
                 testDB.collection(REACTION_COLLECTION_NAME.string).document(reaction.id).set(reaction)
             }
+            var docs = testDB.collection(REACTION_COLLECTION_NAME.string).get().get().documents
+            var a = 1
+            while (docs.size != testDto.reactionList.size && a <= 5) {
+                docs = testDB.collection(REACTION_COLLECTION_NAME.string).get().get().documents
+                a++
+            }
         }
     }
 

diff --git a/springboot/src/test/kotlin/com/goliath/emojihub/springboot/domain/user/dao/UserDaoTest.kt b/springboot/src/test/kotlin/com/goliath/emojihub/springboot/domain/user/dao/UserDaoTest.kt
@@ -64,6 +64,12 @@ internal class UserDaoTest {
             for (user in testDto.userList) {
                 testDB.collection(USER_COLLECTION_NAME.string).document(user.username).set(user)
             }
+            var docs = testDB.collection(USER_COLLECTION_NAME.string).get().get().documents
+            var a = 1
+            while (docs.size != testDto.userList.size && a <= 5) {
+                docs = testDB.collection(USER_COLLECTION_NAME.string).get().get().documents
+                a++
+            }
         }
     }