@@ -32,10 +32,12 @@ import net.mullvad.mullvadvpn.lib.model.Provider
32
32
import net.mullvad.mullvadvpn.lib.model.Providers
33
33
import net.mullvad.mullvadvpn.lib.model.RelayItem
34
34
import net.mullvad.mullvadvpn.lib.model.RelayItemId
35
+ import net.mullvad.mullvadvpn.lib.model.Settings
35
36
import net.mullvad.mullvadvpn.relaylist.descendants
36
37
import net.mullvad.mullvadvpn.repository.CustomListsRepository
37
38
import net.mullvad.mullvadvpn.repository.RelayListFilterRepository
38
39
import net.mullvad.mullvadvpn.repository.RelayListRepository
40
+ import net.mullvad.mullvadvpn.repository.SettingsRepository
39
41
import net.mullvad.mullvadvpn.usecase.AvailableProvidersUseCase
40
42
import net.mullvad.mullvadvpn.usecase.FilteredRelayListUseCase
41
43
import net.mullvad.mullvadvpn.usecase.customlists.CustomListActionUseCase
@@ -58,6 +60,9 @@ class SelectLocationViewModelTest {
58
60
private val mockCustomListsRepository: CustomListsRepository = mockk()
59
61
private val mockCustomListsRelayItemUseCase: CustomListsRelayItemUseCase = mockk()
60
62
63
+ private val mockSettingsRepository: SettingsRepository = mockk()
64
+ private val settingsFlow = MutableStateFlow (mockk<Settings >(relaxed = true ))
65
+
61
66
private lateinit var viewModel: SelectLocationViewModel
62
67
63
68
private val allProviders = MutableStateFlow <List <Provider >>(emptyList())
@@ -79,6 +84,7 @@ class SelectLocationViewModelTest {
79
84
every { mockFilteredRelayListUseCase() } returns filteredRelayList
80
85
every { mockFilteredCustomListRelayItemsUseCase() } returns filteredCustomRelayListItems
81
86
every { mockCustomListsRelayItemUseCase() } returns customListsRelayItem
87
+ every { mockSettingsRepository.settingsUpdates } returns settingsFlow
82
88
83
89
mockkStatic(RELAY_LIST_EXTENSIONS )
84
90
mockkStatic(RELAY_ITEM_EXTENSIONS )
@@ -93,6 +99,7 @@ class SelectLocationViewModelTest {
93
99
relayListRepository = mockRelayListRepository,
94
100
customListsRepository = mockCustomListsRepository,
95
101
customListsRelayItemUseCase = mockCustomListsRelayItemUseCase,
102
+ settingsRepository = mockSettingsRepository,
96
103
)
97
104
}
98
105
0 commit comments