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:
@@ -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
|
||||||
|
@@ -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();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
Reference in New Issue
Block a user