From 3b8d4eca72c9bae1cfee0aeebb6be89505be6616 Mon Sep 17 00:00:00 2001
From: jorgemucientesfayos <jorge.mucientes@automattic.com>
Date: Thu, 19 Sep 2024 16:48:34 +0200
Subject: [PATCH] Fix unit tests compile issues after dao refactor

---
 .../fluxc/store/blaze/BlazeCampaignsStoreTest.kt  | 15 ++++++++++-----
 1 file changed, 10 insertions(+), 5 deletions(-)

diff --git a/example/src/test/java/org/wordpress/android/fluxc/store/blaze/BlazeCampaignsStoreTest.kt b/example/src/test/java/org/wordpress/android/fluxc/store/blaze/BlazeCampaignsStoreTest.kt
index 1c6c1a0353..c6bbdac8f4 100644
--- a/example/src/test/java/org/wordpress/android/fluxc/store/blaze/BlazeCampaignsStoreTest.kt
+++ b/example/src/test/java/org/wordpress/android/fluxc/store/blaze/BlazeCampaignsStoreTest.kt
@@ -39,7 +39,8 @@ import org.wordpress.android.fluxc.network.rest.wpcom.blaze.BlazeCreationRestCli
 import org.wordpress.android.fluxc.network.rest.wpcom.blaze.CampaignImage
 import org.wordpress.android.fluxc.persistence.blaze.BlazeCampaignsDao
 import org.wordpress.android.fluxc.persistence.blaze.BlazeCampaignsDao.BlazeCampaignEntity
-import org.wordpress.android.fluxc.persistence.blaze.BlazeCampaignsDao.BlazeCampaignObjectiveEntity
+import org.wordpress.android.fluxc.persistence.blaze.BlazeObjectivesDao
+import org.wordpress.android.fluxc.persistence.blaze.BlazeObjectivesDao.BlazeCampaignObjectiveEntity
 import org.wordpress.android.fluxc.persistence.blaze.BlazeTargetingDao
 import org.wordpress.android.fluxc.persistence.blaze.BlazeTargetingDeviceEntity
 import org.wordpress.android.fluxc.persistence.blaze.BlazeTargetingLanguageEntity
@@ -123,6 +124,7 @@ class BlazeCampaignsStoreTest {
     private val creationRestClient: BlazeCreationRestClient = mock()
     private val blazeCampaignsDao: BlazeCampaignsDao = mock()
     private val blazeTargetingDao: BlazeTargetingDao = mock()
+    private val blazeObjectivesDao: BlazeObjectivesDao = mock()
     private val siteModel = SiteModel().apply { siteId = SITE_ID }
 
     private lateinit var store: BlazeCampaignsStore
@@ -137,7 +139,8 @@ class BlazeCampaignsStoreTest {
             creationRestClient = creationRestClient,
             campaignsDao = blazeCampaignsDao,
             targetingDao = blazeTargetingDao,
-            coroutineEngine = initCoroutineEngine()
+            coroutineEngine = initCoroutineEngine(),
+            blazeObjectivesDao = blazeObjectivesDao
         )
     }
 
@@ -224,7 +227,8 @@ class BlazeCampaignsStoreTest {
                     BlazeCampaignObjective(
                         id = it.toString(),
                         title = "Title $it",
-                        description = "Description $it"
+                        description = "Description $it",
+                        suitableForDescription = "Suitable for description $it"
                     )
                 }
             )
@@ -232,18 +236,19 @@ class BlazeCampaignsStoreTest {
 
         store.fetchBlazeCampaignObjectives(siteModel)
 
-        verify(blazeCampaignsDao).replaceObjectives(any())
+        verify(blazeObjectivesDao).replaceObjectives(any())
     }
 
     @Test
     fun `when observing campaign objectives, then return data from DB`() = test {
-        whenever(blazeCampaignsDao.observeObjectives(any())).thenReturn(
+        whenever(blazeObjectivesDao.observeObjectives(any())).thenReturn(
             flowOf(
                 List(4) {
                     BlazeCampaignObjectiveEntity(
                         id = it.toString(),
                         title = "Title $it",
                         description = "Description $it",
+                        suitableForDescription = "Suitable for description $it",
                         locale = "en"
                     )
                 }