Use CDS display white balance getter and setter

Bug: 123930917
Test: make ROBOTEST_FILTER="(DisplayWhite*)" RunSettingsRoboTests -j48
Change-Id: Iafc6ff9a61b69baf03efddd52687093c55d35178
(cherry picked from commit ebacc33852)
This commit is contained in:
Christine Franks
2019-03-29 10:14:55 -07:00
parent b8808e35a6
commit 84e67e1f72
2 changed files with 11 additions and 23 deletions

View File

@@ -41,15 +41,12 @@ public class DisplayWhiteBalancePreferenceController extends TogglePreferenceCon
@Override @Override
public boolean isChecked() { public boolean isChecked() {
return Secure.getIntForUser(mContext.getContentResolver(), return getColorDisplayManager().isDisplayWhiteBalanceEnabled();
Secure.DISPLAY_WHITE_BALANCE_ENABLED, 0, UserHandle.USER_CURRENT) == 1;
} }
@Override @Override
public boolean setChecked(boolean isChecked) { public boolean setChecked(boolean isChecked) {
Secure.putIntForUser(mContext.getContentResolver(), Secure.DISPLAY_WHITE_BALANCE_ENABLED, return getColorDisplayManager().setDisplayWhiteBalanceEnabled(isChecked);
isChecked ? 1 : 0, UserHandle.USER_CURRENT);
return true;
} }
@VisibleForTesting @VisibleForTesting

View File

@@ -5,10 +5,7 @@ import static org.mockito.Mockito.spy;
import static org.mockito.Mockito.when; import static org.mockito.Mockito.when;
import static com.google.common.truth.Truth.assertThat; import static com.google.common.truth.Truth.assertThat;
import android.content.Context;
import android.hardware.display.ColorDisplayManager; import android.hardware.display.ColorDisplayManager;
import android.provider.Settings;
import android.provider.Settings.Secure;
import com.android.settings.testutils.shadow.SettingsShadowResources; import com.android.settings.testutils.shadow.SettingsShadowResources;
import org.junit.After; import org.junit.After;
import org.junit.Before; import org.junit.Before;
@@ -26,7 +23,6 @@ import org.robolectric.annotation.Config;
}) })
public class DisplayWhiteBalancePreferenceControllerTest { public class DisplayWhiteBalancePreferenceControllerTest {
private Context mContext;
private DisplayWhiteBalancePreferenceController mController; private DisplayWhiteBalancePreferenceController mController;
@Mock @Mock
@@ -40,8 +36,7 @@ public class DisplayWhiteBalancePreferenceControllerTest {
@Before @Before
public void setUp() { public void setUp() {
MockitoAnnotations.initMocks(this); MockitoAnnotations.initMocks(this);
mContext = RuntimeEnvironment.application; mController = spy(new DisplayWhiteBalancePreferenceController(RuntimeEnvironment.application,
mController = spy(new DisplayWhiteBalancePreferenceController(mContext,
"display_white_balance")); "display_white_balance"));
doReturn(mColorDisplayManager).when(mController).getColorDisplayManager(); doReturn(mColorDisplayManager).when(mController).getColorDisplayManager();
} }
@@ -77,30 +72,26 @@ public class DisplayWhiteBalancePreferenceControllerTest {
} }
@Test @Test
public void setChecked_true() { public void setChecked_true_setSuccessfully() {
mController.setChecked(true); when(mColorDisplayManager.setDisplayWhiteBalanceEnabled(true)).thenReturn(true);
assertThat(Settings.Secure assertThat(mController.setChecked(true)).isTrue();
.getInt(mContext.getContentResolver(), Secure.DISPLAY_WHITE_BALANCE_ENABLED, 0) == 1)
.isTrue();
} }
@Test @Test
public void setChecked_false() { public void setChecked_false_setSuccessfully() {
mController.setChecked(false); when(mColorDisplayManager.setDisplayWhiteBalanceEnabled(false)).thenReturn(true);
assertThat(Settings.Secure assertThat(mController.setChecked(false)).isTrue();
.getInt(mContext.getContentResolver(), Secure.DISPLAY_WHITE_BALANCE_ENABLED, 0) == 1)
.isFalse();
} }
@Test @Test
public void isChecked_true() { public void isChecked_true() {
Settings.Secure.putInt(mContext.getContentResolver(), Secure.DISPLAY_WHITE_BALANCE_ENABLED, 1); when(mColorDisplayManager.isDisplayWhiteBalanceEnabled()).thenReturn(true);
assertThat(mController.isChecked()).isTrue(); assertThat(mController.isChecked()).isTrue();
} }
@Test @Test
public void isChecked_false() { public void isChecked_false() {
Settings.Secure.putInt(mContext.getContentResolver(), Secure.DISPLAY_WHITE_BALANCE_ENABLED, 0); when(mColorDisplayManager.isDisplayWhiteBalanceEnabled()).thenReturn(false);
assertThat(mController.isChecked()).isFalse(); assertThat(mController.isChecked()).isFalse();
} }
} }