From b0f7c89a3493f78b34bbc10d56ffea89c442bbd4 Mon Sep 17 00:00:00 2001
From: Shiina Kin <shiina@sakurasou.io>
Date: Sun, 10 Nov 2024 04:11:16 +0800
Subject: [PATCH] fix: unit test of albumService

---
 .../service/album/AlbumServiceTest.kt         | 20 ++++++++++++++++++-
 1 file changed, 19 insertions(+), 1 deletion(-)

diff --git a/app/src/test/kotlin/io/sakurasou/service/album/AlbumServiceTest.kt b/app/src/test/kotlin/io/sakurasou/service/album/AlbumServiceTest.kt
index a8124bb8..7954ee9c 100644
--- a/app/src/test/kotlin/io/sakurasou/service/album/AlbumServiceTest.kt
+++ b/app/src/test/kotlin/io/sakurasou/service/album/AlbumServiceTest.kt
@@ -11,9 +11,11 @@ import io.sakurasou.exception.service.album.AlbumUpdateFailedException
 import io.sakurasou.model.DatabaseSingleton
 import io.sakurasou.model.dao.album.AlbumDao
 import io.sakurasou.model.dao.image.ImageDao
+import io.sakurasou.model.dao.user.UserDao
 import io.sakurasou.model.dto.AlbumInsertDTO
 import io.sakurasou.model.dto.AlbumUpdateDTO
 import io.sakurasou.model.entity.Album
+import io.sakurasou.model.entity.User
 import kotlinx.coroutines.runBlocking
 import kotlinx.datetime.*
 import kotlin.test.BeforeTest
@@ -26,6 +28,7 @@ import kotlin.test.assertFailsWith
  * 2024/9/13 21:47
  */
 class AlbumServiceTest {
+    private lateinit var userDao: UserDao
     private lateinit var albumDao: AlbumDao
     private lateinit var imageDao: ImageDao
     private lateinit var albumService: AlbumServiceImpl
@@ -36,9 +39,10 @@ class AlbumServiceTest {
     fun setUp() {
         mockkObject(DatabaseSingleton)
         mockkObject(Clock.System)
+        userDao = mockk()
         albumDao = mockk()
         imageDao = mockk()
-        albumService = AlbumServiceImpl(albumDao, imageDao)
+        albumService = AlbumServiceImpl(userDao, albumDao, imageDao)
         instant = Clock.System.now()
         every { Clock.System.now() } returns instant
         now = instant.toLocalDateTime(TimeZone.currentSystemDefault())
@@ -205,18 +209,32 @@ class AlbumServiceTest {
             isUncategorized = false,
             createTime = now
         )
+        val user = User(
+            id = userId,
+            name = "testUser",
+            password = "testPassword",
+            email = "test@example.com",
+            groupId = 1,
+            isDefaultImagePrivate = true,
+            defaultAlbumId = albumId,
+            isBanned = false,
+            createTime = now,
+            updateTime = now
+        )
         val expected = AlbumVO(
             id = albumId,
             name = "My Album",
             description = "Test Album",
             imageCount = 0,
             isUncategorized = false,
+            isDefault = true,
             createTime = album.createTime
         )
 
         coEvery { DatabaseSingleton.dbQuery<AlbumVO>(any()) } coAnswers {
             this.arg<suspend () -> AlbumVO>(0).invoke()
         }
+        every { userDao.findUserById(userId) } returns user
         every { albumDao.findAlbumById(albumId) } returns album
         every { imageDao.countImageByAlbumId(albumId) } returns 0