Fix SelectedSectionManagerTest

wordpress-mobile · Mar 2, 2024 · e604bec · e604bec
1 parent 82a97eb
commit e604bec
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/.../src/test/java/org/wordpress/android/ui/stats/refresh/utils/SelectedSectionManagerTest.kt b/.../src/test/java/org/wordpress/android/ui/stats/refresh/utils/SelectedSectionManagerTest.kt
@@ -13,9 +13,13 @@ import org.mockito.kotlin.whenever
 import org.wordpress.android.BaseUnitTest
 import org.wordpress.android.ui.stats.refresh.lists.StatsListViewModel.StatsSection
 import org.wordpress.android.ui.stats.refresh.lists.StatsListViewModel.StatsSection.MONTHS
+import org.wordpress.android.util.config.StatsTrafficTabFeatureConfig
 
 @ExperimentalCoroutinesApi
 class SelectedSectionManagerTest : BaseUnitTest() {
+    @Mock
+    private lateinit var trafficTabFeatureConfig: StatsTrafficTabFeatureConfig
+
     @Mock
     lateinit var sharedPreferences: SharedPreferences
 
@@ -25,7 +29,7 @@ class SelectedSectionManagerTest : BaseUnitTest() {
 
     @Before
     fun setUp() {
-        selectedSectionManager = SelectedSectionManager(sharedPreferences)
+        selectedSectionManager = SelectedSectionManager(sharedPreferences, trafficTabFeatureConfig)
         whenever(sharedPreferences.edit()).thenReturn(sharedPreferencesEditor)
     }