|
| 1 | +package io.newm.sharedfeatures.devmenu |
| 2 | + |
| 3 | +import com.slack.circuit.test.test |
| 4 | +import com.varabyte.truthish.assertThat |
| 5 | +import io.newm.shared.commonPublic.featureflags.FeatureFlags |
| 6 | +import io.newm.sharedfeatures.fakes.FakeFeatureFlagService |
| 7 | +import io.newm.sharedfeatures.fakes.FakeNavigator |
| 8 | +import io.newm.sharedfeatures.fakes.FakeNewmSharedBuildConfig |
| 9 | +import io.newm.sharedfeatures.screens.FeatureFlagsListScreen.UiEvent |
| 10 | +import io.newm.sharedfeatures.screens.FeatureFlagsListScreen.UiState |
| 11 | +import kotlinx.coroutines.test.runTest |
| 12 | +import kotlin.test.BeforeTest |
| 13 | +import kotlin.test.Test |
| 14 | + |
| 15 | +class FeatureFlagsListPresenterTest { |
| 16 | + private lateinit var navigator: FakeNavigator |
| 17 | + private lateinit var featureFlagService: FakeFeatureFlagService |
| 18 | + private lateinit var buildConfig: FakeNewmSharedBuildConfig |
| 19 | + |
| 20 | + @BeforeTest |
| 21 | + fun setup() { |
| 22 | + navigator = FakeNavigator() |
| 23 | + featureFlagService = FakeFeatureFlagService() |
| 24 | + buildConfig = FakeNewmSharedBuildConfig() |
| 25 | + |
| 26 | + // Setup default flags |
| 27 | + featureFlagService.setAvailableFlags(FeatureFlags.ALL_FLAGS) |
| 28 | + FeatureFlags.ALL_FLAGS.forEach { featureFlagService.setFlag(it.key, it.defaultValue) } |
| 29 | + } |
| 30 | + |
| 31 | + @Test |
| 32 | + fun `initial state loads flags`() = |
| 33 | + runTest { |
| 34 | + val presenter = FeatureFlagsListPresenter(navigator, featureFlagService, buildConfig) |
| 35 | + |
| 36 | + presenter.test { |
| 37 | + assertThat(awaitItem()).isInstanceOf<UiState.Loading>() |
| 38 | + |
| 39 | + val initialState = awaitItem() |
| 40 | + assertThat(initialState).isInstanceOf<UiState.Content>() |
| 41 | + |
| 42 | + val contentState = initialState as UiState.Content |
| 43 | + assertThat(contentState.flags).isNotEmpty() |
| 44 | + assertThat(contentState.environmentInfo.environment) |
| 45 | + .isEqualTo("Production") // Default in fake is !isStagingMode |
| 46 | + } |
| 47 | + } |
| 48 | + |
| 49 | + @Test |
| 50 | + fun `toggling flag updates local override`() = |
| 51 | + runTest { |
| 52 | + val presenter = FeatureFlagsListPresenter(navigator, featureFlagService, buildConfig) |
| 53 | + |
| 54 | + presenter.test { |
| 55 | + awaitItem() // Loading |
| 56 | + val initialState = awaitItem() as UiState.Content |
| 57 | + val firstFlag = initialState.flags.first() |
| 58 | + |
| 59 | + initialState.onEvent( |
| 60 | + UiEvent.OnFlagToggled(firstFlag.featureFlag.key, !firstFlag.effectiveValue), |
| 61 | + ) |
| 62 | + |
| 63 | + // Verify override was set in service |
| 64 | + val override = featureFlagService.getLocalOverride(firstFlag.featureFlag.key) |
| 65 | + assertThat(override).isEqualTo(!firstFlag.effectiveValue) |
| 66 | + } |
| 67 | + } |
| 68 | + |
| 69 | + @Test |
| 70 | + fun `resetting flag removes local override`() = |
| 71 | + runTest { |
| 72 | + val presenter = FeatureFlagsListPresenter(navigator, featureFlagService, buildConfig) |
| 73 | + |
| 74 | + val flag = FeatureFlags.ALL_FLAGS.first() |
| 75 | + featureFlagService.setLocalOverride(flag.key, !flag.defaultValue) |
| 76 | + |
| 77 | + presenter.test { |
| 78 | + awaitItem() // Loading |
| 79 | + val initialState = awaitItem() as UiState.Content |
| 80 | + val flagItem = initialState.flags.first { it.featureFlag.key == flag.key } |
| 81 | + assertThat(flagItem.isOverridden).isTrue() |
| 82 | + |
| 83 | + initialState.onEvent(UiEvent.OnResetFlag(flag.key)) |
| 84 | + |
| 85 | + assertThat(featureFlagService.getLocalOverride(flag.key)).isNull() |
| 86 | + } |
| 87 | + } |
| 88 | + |
| 89 | + @Test |
| 90 | + fun `resetting all flags removes all overrides`() = |
| 91 | + runTest { |
| 92 | + val presenter = FeatureFlagsListPresenter(navigator, featureFlagService, buildConfig) |
| 93 | + |
| 94 | + featureFlagService.setLocalOverride(FeatureFlags.ALL_FLAGS[0].key, true) |
| 95 | + featureFlagService.setLocalOverride(FeatureFlags.ALL_FLAGS[1].key, false) |
| 96 | + |
| 97 | + presenter.test { |
| 98 | + awaitItem() // Loading |
| 99 | + val initialState = awaitItem() as UiState.Content |
| 100 | + assertThat(initialState.flags.any { it.isOverridden }).isTrue() |
| 101 | + |
| 102 | + initialState.onEvent(UiEvent.OnResetAllFlags) |
| 103 | + |
| 104 | + assertThat(featureFlagService.localOverrides.value).isEmpty() |
| 105 | + } |
| 106 | + } |
| 107 | + |
| 108 | + @Test |
| 109 | + fun `refreshing reloads flags`() = |
| 110 | + runTest { |
| 111 | + val presenter = FeatureFlagsListPresenter(navigator, featureFlagService, buildConfig) |
| 112 | + |
| 113 | + presenter.test { |
| 114 | + awaitItem() // Loading |
| 115 | + val initialState = awaitItem() as UiState.Content |
| 116 | + |
| 117 | + // Change a value in the service to verify refresh picks it up |
| 118 | + val flag = FeatureFlags.ALL_FLAGS.first() |
| 119 | + featureFlagService.setFlag(flag.key, !flag.defaultValue) |
| 120 | + |
| 121 | + initialState.onEvent(UiEvent.OnRefresh) |
| 122 | + |
| 123 | + // Expect the state to settle |
| 124 | + val finalState = expectMostRecentItem() as UiState.Content |
| 125 | + assertThat(finalState.isRefreshing).isFalse() |
| 126 | + |
| 127 | + val updatedFlag = finalState.flags.first { it.featureFlag.key == flag.key } |
| 128 | + assertThat(updatedFlag.remoteValue).isEqualTo(!flag.defaultValue) |
| 129 | + } |
| 130 | + } |
| 131 | + |
| 132 | + @Test |
| 133 | + fun `back navigates back`() = |
| 134 | + runTest { |
| 135 | + val presenter = FeatureFlagsListPresenter(navigator, featureFlagService, buildConfig) |
| 136 | + |
| 137 | + presenter.test { |
| 138 | + awaitItem() // Loading |
| 139 | + val initialState = awaitItem() as UiState.Content |
| 140 | + |
| 141 | + initialState.onEvent(UiEvent.OnBack) |
| 142 | + |
| 143 | + assertThat(navigator.popHistory).isNotEmpty() |
| 144 | + } |
| 145 | + } |
| 146 | +} |
0 commit comments