@@ -24,6 +24,7 @@ import androidx.room.Room
2424import androidx.test.platform.app.InstrumentationRegistry
2525import com.duckduckgo.app.browser.DuckDuckGoUrlDetectorImpl
2626import com.duckduckgo.app.browser.R
27+ import com.duckduckgo.app.browser.senseofprotection.SenseOfProtectionExperiment
2728import com.duckduckgo.app.cta.db.DismissedCtaDao
2829import com.duckduckgo.app.cta.model.CtaId
2930import com.duckduckgo.app.cta.model.DismissedCta
@@ -70,6 +71,7 @@ import org.junit.Assert.*
7071import org.junit.Before
7172import org.junit.Rule
7273import org.junit.Test
74+ import org.mockito.Mockito
7375import org.mockito.kotlin.*
7476
7577@FlowPreview
@@ -117,6 +119,8 @@ class CtaViewModelTest {
117119
118120 private val mockUserBrowserProperties: UserBrowserProperties = mock()
119121
122+ private val mockSenseOfProtectionExperiment: SenseOfProtectionExperiment = Mockito .mock()
123+
120124 private val requiredDaxOnboardingCtas: List <CtaId > = listOf (
121125 CtaId .DAX_INTRO ,
122126 CtaId .DAX_DIALOG_SERP ,
@@ -170,6 +174,7 @@ class CtaViewModelTest {
170174 duckPlayer = mockDuckPlayer,
171175 brokenSitePrompt = mockBrokenSitePrompt,
172176 userBrowserProperties = mockUserBrowserProperties,
177+ senseOfProtectionExperiment = mockSenseOfProtectionExperiment,
173178 )
174179 }
175180
0 commit comments