Add missing updates to view state (duckduckgo#410)

CDRussell · web-flow · commit a25a5139fcc9 · 2018-12-17T14:17:26.000Z
diff --git a/app/src/main/java/com/duckduckgo/app/settings/SettingsViewModel.kt b/app/src/main/java/com/duckduckgo/app/settings/SettingsViewModel.kt
@@ -98,6 +98,7 @@ class SettingsViewModel @Inject constructor(
     fun onLightThemeToggled(enabled: Boolean) {
         Timber.i("User toggled light theme, is now enabled: $enabled")
         settingsDataStore.theme = if (enabled) DuckDuckGoTheme.LIGHT else DuckDuckGoTheme.DARK
+        viewState.value = currentViewState().copy(lightThemeEnabled = enabled)
         command.value = Command.UpdateTheme
 
         val pixelName = if (enabled) SETTINGS_THEME_TOGGLED_LIGHT else SETTINGS_THEME_TOGGLED_DARK
@@ -107,6 +108,8 @@ class SettingsViewModel @Inject constructor(
     fun onAutocompleteSettingChanged(enabled: Boolean) {
         Timber.i("User changed autocomplete setting, is now enabled: $enabled")
         settingsDataStore.autoCompleteSuggestionsEnabled = enabled
+
+        viewState.value = currentViewState().copy(autoCompleteSuggestionsEnabled = enabled)
     }
 
     private fun obtainVersion(variantKey: String): String {
