Merge "Redesign the update logic of Allow Background Usage Page." into main
This commit is contained in:
Treehugger Robot
committed by
Android (Google) Code Review
Android (Google) Code Review
commit
a868c602f1
@@ -17,10 +17,10 @@
|
||||
package com.android.settings.fuelgauge;
|
||||
|
||||
import static com.android.settings.SettingsActivity.EXTRA_SHOW_FRAGMENT_ARGUMENTS;
|
||||
import static com.android.settings.fuelgauge.BatteryOptimizeHistoricalLogEntry.Action;
|
||||
|
||||
import static com.google.common.truth.Truth.assertThat;
|
||||
|
||||
import static org.mockito.ArgumentMatchers.any;
|
||||
import static org.mockito.ArgumentMatchers.anyInt;
|
||||
import static org.mockito.ArgumentMatchers.anyString;
|
||||
import static org.mockito.ArgumentMatchers.nullable;
|
||||
@@ -49,13 +49,13 @@ import androidx.test.core.app.ApplicationProvider;
|
||||
|
||||
import com.android.settings.R;
|
||||
import com.android.settings.SettingsActivity;
|
||||
import com.android.settings.fuelgauge.BatteryOptimizeHistoricalLogEntry.Action;
|
||||
import com.android.settings.fuelgauge.batteryusage.BatteryDiffEntry;
|
||||
import com.android.settings.fuelgauge.batteryusage.BatteryEntry;
|
||||
import com.android.settings.fuelgauge.batteryusage.ConvertUtils;
|
||||
import com.android.settings.testutils.FakeFeatureFactory;
|
||||
import com.android.settings.testutils.shadow.ShadowEntityHeaderController;
|
||||
import com.android.settings.widget.EntityHeaderController;
|
||||
import com.android.settingslib.PrimarySwitchPreference;
|
||||
import com.android.settingslib.applications.AppUtils;
|
||||
import com.android.settingslib.applications.ApplicationsState;
|
||||
import com.android.settingslib.applications.instantapps.InstantAppDataProvider;
|
||||
@@ -87,12 +87,12 @@ import java.util.concurrent.TimeUnit;
|
||||
})
|
||||
public class AdvancedPowerUsageDetailTest {
|
||||
|
||||
@Rule
|
||||
public final MockitoRule mMockitoRule = MockitoJUnit.rule();
|
||||
@Rule public final MockitoRule mMockitoRule = MockitoJUnit.rule();
|
||||
|
||||
private static final String APP_LABEL = "app label";
|
||||
private static final String SUMMARY = "summary";
|
||||
private static final String[] PACKAGE_NAME = {"com.android.app"};
|
||||
private static final String PACKAGE_NAME = "com.android.app";
|
||||
private static final String INITIATING_PACKAGE_NAME = "com.android.vending";
|
||||
private static final String USAGE_PERCENT = "16%";
|
||||
private static final int ICON_ID = 123;
|
||||
private static final int UID = 1;
|
||||
@@ -100,39 +100,43 @@ public class AdvancedPowerUsageDetailTest {
|
||||
private static final long FOREGROUND_SERVICE_TIME_MS = 123;
|
||||
private static final long BACKGROUND_TIME_MS = 100;
|
||||
private static final long SCREEN_ON_TIME_MS = 321;
|
||||
private static final String KEY_ALLOW_BACKGROUND_USAGE = "allow_background_usage";
|
||||
|
||||
@Mock(answer = Answers.RETURNS_DEEP_STUBS)
|
||||
private FragmentActivity mActivity;
|
||||
|
||||
@Mock private EntityHeaderController mEntityHeaderController;
|
||||
@Mock private LayoutPreference mHeaderPreference;
|
||||
@Mock private ApplicationsState mState;
|
||||
@Mock private ApplicationsState.AppEntry mAppEntry;
|
||||
@Mock private BatteryEntry mBatteryEntry;
|
||||
@Mock private PackageManager mPackageManager;
|
||||
@Mock private InstallSourceInfo mInstallSourceInfo;
|
||||
@Mock private LayoutPreference mLayoutPreference;
|
||||
@Mock private AppOpsManager mAppOpsManager;
|
||||
@Mock private LoaderManager mLoaderManager;
|
||||
@Mock private BatteryOptimizeUtils mBatteryOptimizeUtils;
|
||||
|
||||
private int mTestMode;
|
||||
private Context mContext;
|
||||
private PrimarySwitchPreference mAllowBackgroundUsagePreference;
|
||||
private AdvancedPowerUsageDetail mFragment;
|
||||
private SettingsActivity mTestActivity;
|
||||
private FakeFeatureFactory mFeatureFactory;
|
||||
private MetricsFeatureProvider mMetricsFeatureProvider;
|
||||
private BatteryDiffEntry mBatteryDiffEntry;
|
||||
private Bundle mBundle;
|
||||
private BatteryOptimizeUtils mBatteryOptimizeUtils;
|
||||
|
||||
@Before
|
||||
public void setUp() {
|
||||
public void setUp() throws Exception {
|
||||
mContext = spy(ApplicationProvider.getApplicationContext());
|
||||
when(mContext.getPackageName()).thenReturn("foo");
|
||||
when(mContext.getPackageName()).thenReturn(PACKAGE_NAME);
|
||||
when(mContext.getPackageManager()).thenReturn(mPackageManager);
|
||||
when(mPackageManager.getInstallSourceInfo(anyString())).thenReturn(mInstallSourceInfo);
|
||||
mFeatureFactory = FakeFeatureFactory.setupForTest();
|
||||
mMetricsFeatureProvider = mFeatureFactory.metricsFeatureProvider;
|
||||
|
||||
prepareTestBatteryOptimizationUtils();
|
||||
mFragment = spy(new AdvancedPowerUsageDetail());
|
||||
mFragment.mBatteryOptimizeUtils = mBatteryOptimizeUtils;
|
||||
doReturn(mLayoutPreference).when(mFragment).findPreference(any());
|
||||
mBundle = spy(new Bundle());
|
||||
doReturn(mContext).when(mFragment).getContext();
|
||||
doReturn(mActivity).when(mFragment).getActivity();
|
||||
@@ -195,7 +199,6 @@ public class AdvancedPowerUsageDetailTest {
|
||||
when(mBatteryDiffEntry.getAppLabel()).thenReturn(APP_LABEL);
|
||||
when(mBatteryDiffEntry.getAppIconId()).thenReturn(ICON_ID);
|
||||
|
||||
mFragment.mHeaderPreference = mHeaderPreference;
|
||||
mFragment.mState = mState;
|
||||
mFragment.mBatteryOptimizeUtils = mBatteryOptimizeUtils;
|
||||
mFragment.mLogStringBuilder = new StringBuilder();
|
||||
@@ -219,10 +222,6 @@ public class AdvancedPowerUsageDetailTest {
|
||||
.startActivityAsUser(captor.capture(), nullable(UserHandle.class));
|
||||
doAnswer(callable).when(mActivity).startActivity(captor.capture());
|
||||
doAnswer(callable).when(mContext).startActivity(captor.capture());
|
||||
|
||||
mAllowBackgroundUsagePreference = new PrimarySwitchPreference(mContext);
|
||||
mAllowBackgroundUsagePreference.setKey(KEY_ALLOW_BACKGROUND_USAGE);
|
||||
mFragment.mAllowBackgroundUsagePreference = mAllowBackgroundUsagePreference;
|
||||
}
|
||||
|
||||
@After
|
||||
@@ -320,17 +319,15 @@ public class AdvancedPowerUsageDetailTest {
|
||||
.isEqualTo(SCREEN_ON_TIME_MS);
|
||||
assertThat(mBundle.getString(AdvancedPowerUsageDetail.EXTRA_POWER_USAGE_PERCENT))
|
||||
.isEqualTo(USAGE_PERCENT);
|
||||
assertThat(mBundle.getString(AdvancedPowerUsageDetail.EXTRA_SLOT_TIME))
|
||||
.isEqualTo(null);
|
||||
assertThat(mBundle.getString(AdvancedPowerUsageDetail.EXTRA_SLOT_TIME)).isNull();
|
||||
}
|
||||
|
||||
|
||||
@Test
|
||||
public void startBatteryDetailPage_noBatteryUsage_hasBasicData() {
|
||||
final ArgumentCaptor<Intent> captor = ArgumentCaptor.forClass(Intent.class);
|
||||
|
||||
AdvancedPowerUsageDetail.startBatteryDetailPage(
|
||||
mActivity, mFragment, PACKAGE_NAME[0], UserHandle.OWNER);
|
||||
mActivity, mFragment, PACKAGE_NAME, UserHandle.OWNER);
|
||||
|
||||
verify(mActivity).startActivity(captor.capture());
|
||||
|
||||
@@ -338,7 +335,7 @@ public class AdvancedPowerUsageDetailTest {
|
||||
captor.getValue()
|
||||
.getBundleExtra(EXTRA_SHOW_FRAGMENT_ARGUMENTS)
|
||||
.getString(AdvancedPowerUsageDetail.EXTRA_PACKAGE_NAME))
|
||||
.isEqualTo(PACKAGE_NAME[0]);
|
||||
.isEqualTo(PACKAGE_NAME);
|
||||
|
||||
assertThat(
|
||||
captor.getValue()
|
||||
@@ -351,62 +348,21 @@ public class AdvancedPowerUsageDetailTest {
|
||||
public void startBatteryDetailPage_batteryEntryNotExisted_extractUidFromPackageName()
|
||||
throws PackageManager.NameNotFoundException {
|
||||
mBundle.clear();
|
||||
doReturn(UID).when(mPackageManager).getPackageUid(PACKAGE_NAME[0], 0 /* no flag */);
|
||||
doReturn(UID).when(mPackageManager).getPackageUid(PACKAGE_NAME, 0 /* no flag */);
|
||||
|
||||
AdvancedPowerUsageDetail.startBatteryDetailPage(
|
||||
mActivity, mFragment, PACKAGE_NAME[0], UserHandle.OWNER);
|
||||
mActivity, mFragment, PACKAGE_NAME, UserHandle.OWNER);
|
||||
|
||||
assertThat(mBundle.getInt(AdvancedPowerUsageDetail.EXTRA_UID)).isEqualTo(UID);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void initFooter_isValidPackageName_hasCorrectString() {
|
||||
when(mBatteryOptimizeUtils.isDisabledForOptimizeModeOnly()).thenReturn(true);
|
||||
public void onPause_optimizationModeIsChanged_logPreference() throws Exception {
|
||||
mFragment.mOptimizationMode = BatteryOptimizeUtils.MODE_RESTRICTED;
|
||||
when(mBatteryOptimizeUtils.getPackageName()).thenReturn(PACKAGE_NAME);
|
||||
when(mInstallSourceInfo.getInitiatingPackageName()).thenReturn(INITIATING_PACKAGE_NAME);
|
||||
|
||||
mFragment.initFooter();
|
||||
|
||||
assertThat(mAllowBackgroundUsagePreference.getSummary().toString())
|
||||
.isEqualTo("This app requires optimized battery usage.");
|
||||
}
|
||||
|
||||
@Test
|
||||
public void initFooter_isSystemOrDefaultApp_hasCorrectString() {
|
||||
when(mBatteryOptimizeUtils.isDisabledForOptimizeModeOnly()).thenReturn(false);
|
||||
when(mBatteryOptimizeUtils.isSystemOrDefaultApp()).thenReturn(true);
|
||||
|
||||
mFragment.initFooter();
|
||||
|
||||
assertThat(mAllowBackgroundUsagePreference.getSummary().toString())
|
||||
.isEqualTo("This app requires unrestricted battery usage.");
|
||||
}
|
||||
|
||||
@Test
|
||||
public void initFooter_hasCorrectString() {
|
||||
when(mBatteryOptimizeUtils.isDisabledForOptimizeModeOnly()).thenReturn(false);
|
||||
when(mBatteryOptimizeUtils.isSystemOrDefaultApp()).thenReturn(false);
|
||||
|
||||
mFragment.initFooter();
|
||||
|
||||
assertThat(mAllowBackgroundUsagePreference.getSummary().toString())
|
||||
.isEqualTo("Enable for real-time updates, disable to save battery");
|
||||
}
|
||||
|
||||
@Test
|
||||
public void onPause_optimizationModeChanged_logPreference()
|
||||
throws PackageManager.NameNotFoundException, InterruptedException {
|
||||
final String packageName = "testPackageName";
|
||||
final int restrictedMode = BatteryOptimizeUtils.MODE_RESTRICTED;
|
||||
final int optimizedMode = BatteryOptimizeUtils.MODE_OPTIMIZED;
|
||||
mFragment.mOptimizationMode = restrictedMode;
|
||||
when(mBatteryOptimizeUtils.getAppOptimizationMode()).thenReturn(restrictedMode);
|
||||
when(mBatteryOptimizeUtils.getPackageName()).thenReturn(packageName);
|
||||
when(mContext.getPackageManager()).thenReturn(mPackageManager);
|
||||
when(mPackageManager.getInstallSourceInfo(anyString())).thenReturn(mInstallSourceInfo);
|
||||
when(mInstallSourceInfo.getInitiatingPackageName()).thenReturn("com.android.vending");
|
||||
|
||||
mFragment.onPreferenceChange(mAllowBackgroundUsagePreference, true);
|
||||
verify(mBatteryOptimizeUtils).setAppUsageState(optimizedMode, Action.APPLY);
|
||||
when(mBatteryOptimizeUtils.getAppOptimizationMode()).thenReturn(optimizedMode);
|
||||
mBatteryOptimizeUtils.setAppUsageState(BatteryOptimizeUtils.MODE_OPTIMIZED, Action.APPLY);
|
||||
mFragment.onPause();
|
||||
|
||||
TimeUnit.SECONDS.sleep(1);
|
||||
@@ -415,27 +371,18 @@ public class AdvancedPowerUsageDetailTest {
|
||||
SettingsEnums.LEAVE_APP_BATTERY_USAGE,
|
||||
SettingsEnums.ACTION_APP_BATTERY_USAGE_ALLOW_BACKGROUND,
|
||||
SettingsEnums.FUELGAUGE_POWER_USAGE_DETAIL,
|
||||
packageName,
|
||||
PACKAGE_NAME,
|
||||
/* consumed battery */ 0);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void onPause_optimizationModeIsNotChanged_notInvokeLogging()
|
||||
throws PackageManager.NameNotFoundException, InterruptedException {
|
||||
final int restrictedMode = BatteryOptimizeUtils.MODE_RESTRICTED;
|
||||
final int optimizedMode = BatteryOptimizeUtils.MODE_OPTIMIZED;
|
||||
mFragment.mOptimizationMode = restrictedMode;
|
||||
when(mBatteryOptimizeUtils.getAppOptimizationMode()).thenReturn(restrictedMode);
|
||||
when(mContext.getPackageManager()).thenReturn(mPackageManager);
|
||||
when(mPackageManager.getInstallSourceInfo(anyString())).thenReturn(mInstallSourceInfo);
|
||||
when(mInstallSourceInfo.getInitiatingPackageName()).thenReturn("com.android.vending");
|
||||
public void onPause_optimizationModeIsNotChanged_notInvokeLogging() throws Exception {
|
||||
mFragment.mOptimizationMode = BatteryOptimizeUtils.MODE_RESTRICTED;
|
||||
when(mBatteryOptimizeUtils.getPackageName()).thenReturn(PACKAGE_NAME);
|
||||
when(mInstallSourceInfo.getInitiatingPackageName()).thenReturn(INITIATING_PACKAGE_NAME);
|
||||
|
||||
mFragment.onPreferenceChange(mAllowBackgroundUsagePreference, true);
|
||||
verify(mBatteryOptimizeUtils).setAppUsageState(optimizedMode, Action.APPLY);
|
||||
when(mBatteryOptimizeUtils.getAppOptimizationMode()).thenReturn(optimizedMode);
|
||||
mFragment.onPreferenceChange(mAllowBackgroundUsagePreference, false);
|
||||
verify(mBatteryOptimizeUtils).setAppUsageState(restrictedMode, Action.APPLY);
|
||||
when(mBatteryOptimizeUtils.getAppOptimizationMode()).thenReturn(restrictedMode);
|
||||
mBatteryOptimizeUtils.setAppUsageState(BatteryOptimizeUtils.MODE_OPTIMIZED, Action.APPLY);
|
||||
mBatteryOptimizeUtils.setAppUsageState(BatteryOptimizeUtils.MODE_RESTRICTED, Action.APPLY);
|
||||
mFragment.onPause();
|
||||
|
||||
TimeUnit.SECONDS.sleep(1);
|
||||
@@ -446,4 +393,16 @@ public class AdvancedPowerUsageDetailTest {
|
||||
public void shouldSkipForInitialSUW_returnTrue() {
|
||||
assertThat(mFragment.shouldSkipForInitialSUW()).isTrue();
|
||||
}
|
||||
|
||||
private void prepareTestBatteryOptimizationUtils() {
|
||||
mBatteryOptimizeUtils = spy(new BatteryOptimizeUtils(mContext, UID, PACKAGE_NAME));
|
||||
Answer<Void> setTestMode =
|
||||
invocation -> {
|
||||
mTestMode = invocation.getArgument(0);
|
||||
return null;
|
||||
};
|
||||
doAnswer(setTestMode).when(mBatteryOptimizeUtils).setAppUsageState(anyInt(), any());
|
||||
Answer<Integer> getTestMode = invocation -> mTestMode;
|
||||
doAnswer(getTestMode).when(mBatteryOptimizeUtils).getAppOptimizationMode();
|
||||
}
|
||||
}
|
||||
|
||||
@@ -1,150 +0,0 @@
|
||||
/*
|
||||
* Copyright (C) 2023 The Android Open Source Project
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package com.android.settings.fuelgauge;
|
||||
|
||||
import static com.google.common.truth.Truth.assertThat;
|
||||
|
||||
import static org.mockito.Mockito.doReturn;
|
||||
import static org.mockito.Mockito.spy;
|
||||
import static org.mockito.Mockito.when;
|
||||
|
||||
import android.content.Context;
|
||||
import android.content.pm.PackageManager;
|
||||
|
||||
import com.android.settingslib.widget.MainSwitchPreference;
|
||||
|
||||
import org.junit.Before;
|
||||
import org.junit.Test;
|
||||
import org.junit.runner.RunWith;
|
||||
import org.mockito.Mock;
|
||||
import org.mockito.MockitoAnnotations;
|
||||
import org.robolectric.RobolectricTestRunner;
|
||||
import org.robolectric.RuntimeEnvironment;
|
||||
|
||||
@RunWith(RobolectricTestRunner.class)
|
||||
public class AllowBackgroundPreferenceControllerTest {
|
||||
private static final int UID = 12345;
|
||||
private static final String PACKAGE_NAME = "com.android.app";
|
||||
|
||||
private AllowBackgroundPreferenceController mController;
|
||||
private MainSwitchPreference mMainSwitchPreference;
|
||||
private BatteryOptimizeUtils mBatteryOptimizeUtils;
|
||||
|
||||
@Mock private PackageManager mMockPackageManager;
|
||||
|
||||
@Before
|
||||
public void setUp() throws Exception {
|
||||
MockitoAnnotations.initMocks(this);
|
||||
|
||||
Context context = spy(RuntimeEnvironment.application);
|
||||
BatteryUtils.getInstance(context).reset();
|
||||
doReturn(UID)
|
||||
.when(mMockPackageManager)
|
||||
.getPackageUid(PACKAGE_NAME, PackageManager.GET_META_DATA);
|
||||
|
||||
mController = new AllowBackgroundPreferenceController(context, UID, PACKAGE_NAME);
|
||||
mMainSwitchPreference = new MainSwitchPreference(RuntimeEnvironment.application);
|
||||
mBatteryOptimizeUtils = spy(new BatteryOptimizeUtils(context, UID, PACKAGE_NAME));
|
||||
mController.mBatteryOptimizeUtils = mBatteryOptimizeUtils;
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testUpdateState_isValidPackage_prefEnabled() {
|
||||
when(mBatteryOptimizeUtils.isDisabledForOptimizeModeOnly()).thenReturn(false);
|
||||
when(mBatteryOptimizeUtils.isSystemOrDefaultApp()).thenReturn(false);
|
||||
|
||||
mController.updateState(mMainSwitchPreference);
|
||||
|
||||
assertThat(mBatteryOptimizeUtils.isOptimizeModeMutable()).isTrue();
|
||||
assertThat(mMainSwitchPreference.isEnabled()).isTrue();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testUpdateState_invalidPackage_prefDisabled() {
|
||||
when(mBatteryOptimizeUtils.isDisabledForOptimizeModeOnly()).thenReturn(true);
|
||||
when(mBatteryOptimizeUtils.isSystemOrDefaultApp()).thenReturn(false);
|
||||
|
||||
mController.updateState(mMainSwitchPreference);
|
||||
|
||||
assertThat(mBatteryOptimizeUtils.isOptimizeModeMutable()).isFalse();
|
||||
assertThat(mMainSwitchPreference.isEnabled()).isFalse();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testUpdateState_isSystemOrDefaultAppAndRestrictedStates_prefChecked() {
|
||||
when(mBatteryOptimizeUtils.isDisabledForOptimizeModeOnly()).thenReturn(false);
|
||||
when(mBatteryOptimizeUtils.isSystemOrDefaultApp()).thenReturn(true);
|
||||
when(mBatteryOptimizeUtils.getAppOptimizationMode())
|
||||
.thenReturn(BatteryOptimizeUtils.MODE_RESTRICTED);
|
||||
|
||||
mController.updateState(mMainSwitchPreference);
|
||||
|
||||
assertThat(mMainSwitchPreference.isEnabled()).isFalse();
|
||||
assertThat(mMainSwitchPreference.isChecked()).isFalse();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testUpdateState_isSystemOrDefaultApp_prefUnchecked() {
|
||||
when(mBatteryOptimizeUtils.isDisabledForOptimizeModeOnly()).thenReturn(false);
|
||||
when(mBatteryOptimizeUtils.isSystemOrDefaultApp()).thenReturn(true);
|
||||
when(mBatteryOptimizeUtils.getAppOptimizationMode())
|
||||
.thenReturn(BatteryOptimizeUtils.MODE_OPTIMIZED);
|
||||
|
||||
mController.updateState(mMainSwitchPreference);
|
||||
|
||||
assertThat(mMainSwitchPreference.isEnabled()).isFalse();
|
||||
assertThat(mMainSwitchPreference.isChecked()).isTrue();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testUpdateState_isRestrictedStates_prefChecked() {
|
||||
when(mBatteryOptimizeUtils.isOptimizeModeMutable()).thenReturn(true);
|
||||
when(mBatteryOptimizeUtils.getAppOptimizationMode())
|
||||
.thenReturn(BatteryOptimizeUtils.MODE_RESTRICTED);
|
||||
|
||||
mController.updateState(mMainSwitchPreference);
|
||||
|
||||
assertThat(mMainSwitchPreference.isEnabled()).isTrue();
|
||||
assertThat(mMainSwitchPreference.isChecked()).isFalse();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testUpdateState_prefUnchecked() {
|
||||
when(mBatteryOptimizeUtils.isOptimizeModeMutable()).thenReturn(true);
|
||||
when(mBatteryOptimizeUtils.getAppOptimizationMode())
|
||||
.thenReturn(BatteryOptimizeUtils.MODE_OPTIMIZED);
|
||||
|
||||
mController.updateState(mMainSwitchPreference);
|
||||
|
||||
assertThat(mMainSwitchPreference.isEnabled()).isTrue();
|
||||
assertThat(mMainSwitchPreference.isChecked()).isTrue();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testHandlePreferenceTreeClick_samePrefKey_verifyAction() {
|
||||
mMainSwitchPreference.setKey(
|
||||
AllowBackgroundPreferenceController.KEY_ALLOW_BACKGROUND_USAGE);
|
||||
mController.handlePreferenceTreeClick(mMainSwitchPreference);
|
||||
|
||||
assertThat(mController.handlePreferenceTreeClick(mMainSwitchPreference)).isTrue();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testHandlePreferenceTreeClick_incorrectPrefKey_noAction() {
|
||||
assertThat(mController.handlePreferenceTreeClick(mMainSwitchPreference)).isFalse();
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,202 @@
|
||||
/*
|
||||
* Copyright (C) 2024 The Android Open Source Project
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package com.android.settings.fuelgauge;
|
||||
|
||||
import static com.google.common.truth.Truth.assertThat;
|
||||
|
||||
import static org.mockito.ArgumentMatchers.any;
|
||||
import static org.mockito.ArgumentMatchers.anyInt;
|
||||
import static org.mockito.Mockito.doAnswer;
|
||||
import static org.mockito.Mockito.doReturn;
|
||||
import static org.mockito.Mockito.never;
|
||||
import static org.mockito.Mockito.spy;
|
||||
import static org.mockito.Mockito.verify;
|
||||
|
||||
import android.content.Context;
|
||||
|
||||
import androidx.test.core.app.ApplicationProvider;
|
||||
|
||||
import com.android.settings.R;
|
||||
import com.android.settings.dashboard.DashboardFragment;
|
||||
import com.android.settings.fuelgauge.BatteryOptimizeHistoricalLogEntry.Action;
|
||||
import com.android.settingslib.PrimarySwitchPreference;
|
||||
|
||||
import org.junit.Before;
|
||||
import org.junit.Test;
|
||||
import org.junit.runner.RunWith;
|
||||
import org.mockito.Mock;
|
||||
import org.mockito.MockitoAnnotations;
|
||||
import org.mockito.stubbing.Answer;
|
||||
import org.robolectric.RobolectricTestRunner;
|
||||
|
||||
@RunWith(RobolectricTestRunner.class)
|
||||
public class BackgroundUsageAllowabilityPreferenceControllerTest {
|
||||
private static final int UID = 12345;
|
||||
private static final String PACKAGE_NAME = "com.android.app";
|
||||
|
||||
private int mTestMode;
|
||||
private Context mContext;
|
||||
private BackgroundUsageAllowabilityPreferenceController mBackgroundUsageController;
|
||||
private BatteryOptimizeUtils mBatteryOptimizeUtils;
|
||||
|
||||
@Mock DashboardFragment mDashboardFragment;
|
||||
@Mock PrimarySwitchPreference mBackgroundUsageAllowabilityPreference;
|
||||
|
||||
@Before
|
||||
public void setUp() throws Exception {
|
||||
MockitoAnnotations.initMocks(this);
|
||||
|
||||
mContext = spy(ApplicationProvider.getApplicationContext());
|
||||
prepareTestBatteryOptimizationUtils();
|
||||
mBackgroundUsageController =
|
||||
spy(
|
||||
new BackgroundUsageAllowabilityPreferenceController(
|
||||
mContext,
|
||||
mDashboardFragment,
|
||||
/* preferenceKey= */ "test",
|
||||
mBatteryOptimizeUtils));
|
||||
mBackgroundUsageController.mBackgroundUsageAllowabilityPreference =
|
||||
mBackgroundUsageAllowabilityPreference;
|
||||
}
|
||||
|
||||
@Test
|
||||
public void initPreferences_immutableOptimized_setExpectedContent() {
|
||||
doReturn(false).when(mBatteryOptimizeUtils).isOptimizeModeMutable();
|
||||
doReturn(true).when(mBatteryOptimizeUtils).isDisabledForOptimizeModeOnly();
|
||||
|
||||
mBackgroundUsageController.initPreferences();
|
||||
|
||||
verify(mBackgroundUsageAllowabilityPreference).setEnabled(false);
|
||||
verify(mBackgroundUsageAllowabilityPreference).setSwitchEnabled(false);
|
||||
verify(mBackgroundUsageAllowabilityPreference)
|
||||
.setSummary(
|
||||
mContext.getString(
|
||||
R.string.manager_battery_usage_footer_limited,
|
||||
mContext.getString(R.string.manager_battery_usage_optimized_only)));
|
||||
verify(mBackgroundUsageAllowabilityPreference, never())
|
||||
.setOnPreferenceChangeListener(any());
|
||||
verify(mBackgroundUsageAllowabilityPreference, never()).setOnPreferenceClickListener(any());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void initPreferences_immutableUnrestricted_setExpectedContent() {
|
||||
doReturn(false).when(mBatteryOptimizeUtils).isOptimizeModeMutable();
|
||||
doReturn(false).when(mBatteryOptimizeUtils).isDisabledForOptimizeModeOnly();
|
||||
doReturn(true).when(mBatteryOptimizeUtils).isSystemOrDefaultApp();
|
||||
|
||||
mBackgroundUsageController.initPreferences();
|
||||
|
||||
verify(mBackgroundUsageAllowabilityPreference).setEnabled(false);
|
||||
verify(mBackgroundUsageAllowabilityPreference).setSwitchEnabled(false);
|
||||
verify(mBackgroundUsageAllowabilityPreference)
|
||||
.setSummary(
|
||||
mContext.getString(
|
||||
R.string.manager_battery_usage_footer_limited,
|
||||
mContext.getString(
|
||||
R.string.manager_battery_usage_unrestricted_only)));
|
||||
verify(mBackgroundUsageAllowabilityPreference, never())
|
||||
.setOnPreferenceChangeListener(any());
|
||||
verify(mBackgroundUsageAllowabilityPreference, never()).setOnPreferenceClickListener(any());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void initPreferences_mutableMode_setExpectedContent() {
|
||||
doReturn(true).when(mBatteryOptimizeUtils).isOptimizeModeMutable();
|
||||
doReturn(false).when(mBatteryOptimizeUtils).isDisabledForOptimizeModeOnly();
|
||||
doReturn(false).when(mBatteryOptimizeUtils).isSystemOrDefaultApp();
|
||||
|
||||
mBackgroundUsageController.initPreferences();
|
||||
|
||||
verify(mBackgroundUsageAllowabilityPreference).setEnabled(true);
|
||||
verify(mBackgroundUsageAllowabilityPreference).setSwitchEnabled(true);
|
||||
verify(mBackgroundUsageAllowabilityPreference)
|
||||
.setSummary(
|
||||
mContext.getString(
|
||||
R.string.manager_battery_usage_allow_background_usage_summary));
|
||||
verify(mBackgroundUsageAllowabilityPreference).setOnPreferenceChangeListener(any());
|
||||
verify(mBackgroundUsageAllowabilityPreference).setOnPreferenceClickListener(any());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void updatePreferences_setIntoUnrestrictedMode_setExpectedPrefStatus() {
|
||||
mTestMode = BatteryOptimizeUtils.MODE_UNRESTRICTED;
|
||||
|
||||
mBackgroundUsageController.updatePreferences(mTestMode);
|
||||
|
||||
verifyPreferences(mTestMode);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void updatePreferences_setIntoOptimizedMode_setExpectedPrefStatus() {
|
||||
mTestMode = BatteryOptimizeUtils.MODE_OPTIMIZED;
|
||||
|
||||
mBackgroundUsageController.updatePreferences(mTestMode);
|
||||
|
||||
verifyPreferences(mTestMode);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void updatePreferences_setIntoRestrictedMode_setExpectedPrefStatus() {
|
||||
mTestMode = BatteryOptimizeUtils.MODE_RESTRICTED;
|
||||
|
||||
mBackgroundUsageController.updatePreferences(mTestMode);
|
||||
|
||||
verifyPreferences(mTestMode);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void handleBatteryOptimizeModeUpdated_modeChange_setExpectedPrefStatus() {
|
||||
mTestMode = BatteryOptimizeUtils.MODE_RESTRICTED;
|
||||
|
||||
mBackgroundUsageController.handleBatteryOptimizeModeUpdated(
|
||||
BatteryOptimizeUtils.MODE_OPTIMIZED);
|
||||
|
||||
verify(mBatteryOptimizeUtils)
|
||||
.setAppUsageState(BatteryOptimizeUtils.MODE_OPTIMIZED, Action.APPLY);
|
||||
assertThat(mTestMode).isEqualTo(BatteryOptimizeUtils.MODE_OPTIMIZED);
|
||||
verifyPreferences(mTestMode);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void handleBatteryOptimizeModeUpdated_modeNotChange_setExpectedPrefStatus() {
|
||||
mTestMode = BatteryOptimizeUtils.MODE_RESTRICTED;
|
||||
|
||||
mBackgroundUsageController.handleBatteryOptimizeModeUpdated(
|
||||
BatteryOptimizeUtils.MODE_RESTRICTED);
|
||||
|
||||
verify(mBatteryOptimizeUtils, never()).setAppUsageState(anyInt(), any());
|
||||
assertThat(mTestMode).isEqualTo(BatteryOptimizeUtils.MODE_RESTRICTED);
|
||||
verify(mBackgroundUsageController, never()).updatePreferences(mTestMode);
|
||||
}
|
||||
|
||||
private void prepareTestBatteryOptimizationUtils() {
|
||||
mBatteryOptimizeUtils = spy(new BatteryOptimizeUtils(mContext, UID, PACKAGE_NAME));
|
||||
Answer<Void> setTestMode =
|
||||
invocation -> {
|
||||
mTestMode = invocation.getArgument(0);
|
||||
return null;
|
||||
};
|
||||
doAnswer(setTestMode).when(mBatteryOptimizeUtils).setAppUsageState(anyInt(), any());
|
||||
Answer<Integer> getTestMode = invocation -> mTestMode;
|
||||
doAnswer(getTestMode).when(mBatteryOptimizeUtils).getAppOptimizationMode();
|
||||
}
|
||||
|
||||
private void verifyPreferences(int mode) {
|
||||
boolean isAllowBackground = mode != BatteryOptimizeUtils.MODE_RESTRICTED;
|
||||
verify(mBackgroundUsageAllowabilityPreference).setChecked(isAllowBackground);
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,179 @@
|
||||
/*
|
||||
* Copyright (C) 2024 The Android Open Source Project
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package com.android.settings.fuelgauge;
|
||||
|
||||
import static com.google.common.truth.Truth.assertThat;
|
||||
|
||||
import static org.mockito.ArgumentMatchers.any;
|
||||
import static org.mockito.ArgumentMatchers.anyInt;
|
||||
import static org.mockito.Mockito.doAnswer;
|
||||
import static org.mockito.Mockito.doReturn;
|
||||
import static org.mockito.Mockito.never;
|
||||
import static org.mockito.Mockito.spy;
|
||||
import static org.mockito.Mockito.verify;
|
||||
|
||||
import android.content.Context;
|
||||
|
||||
import androidx.test.core.app.ApplicationProvider;
|
||||
|
||||
import com.android.settingslib.widget.MainSwitchPreference;
|
||||
import com.android.settingslib.widget.SelectorWithWidgetPreference;
|
||||
|
||||
import org.junit.Before;
|
||||
import org.junit.Test;
|
||||
import org.junit.runner.RunWith;
|
||||
import org.mockito.Mock;
|
||||
import org.mockito.MockitoAnnotations;
|
||||
import org.mockito.stubbing.Answer;
|
||||
import org.robolectric.RobolectricTestRunner;
|
||||
|
||||
@RunWith(RobolectricTestRunner.class)
|
||||
public class BatteryOptimizationModePreferenceControllerTest {
|
||||
private static final int UID = 12345;
|
||||
private static final String PACKAGE_NAME = "com.android.app";
|
||||
|
||||
private int mTestMode;
|
||||
private Context mContext;
|
||||
private BatteryOptimizationModePreferenceController mBackgroundUsageController;
|
||||
private BatteryOptimizeUtils mBatteryOptimizeUtils;
|
||||
|
||||
@Mock MainSwitchPreference mBackgroundUsageAllowabilityPreference;
|
||||
@Mock SelectorWithWidgetPreference mOptimizedPreference;
|
||||
@Mock SelectorWithWidgetPreference mUnrestrictedPreference;
|
||||
|
||||
@Before
|
||||
public void setUp() throws Exception {
|
||||
MockitoAnnotations.initMocks(this);
|
||||
|
||||
mContext = spy(ApplicationProvider.getApplicationContext());
|
||||
prepareTestBatteryOptimizationUtils();
|
||||
mBackgroundUsageController =
|
||||
spy(
|
||||
new BatteryOptimizationModePreferenceController(
|
||||
mContext, "test", mBatteryOptimizeUtils));
|
||||
mBackgroundUsageController.mBackgroundUsageAllowabilityPreference =
|
||||
mBackgroundUsageAllowabilityPreference;
|
||||
mBackgroundUsageController.mOptimizedPreference = mOptimizedPreference;
|
||||
mBackgroundUsageController.mUnrestrictedPreference = mUnrestrictedPreference;
|
||||
}
|
||||
|
||||
@Test
|
||||
public void initPreferences_mutableMode_setEnabled() {
|
||||
doReturn(true).when(mBatteryOptimizeUtils).isOptimizeModeMutable();
|
||||
|
||||
mBackgroundUsageController.initPreferences();
|
||||
|
||||
verify(mBackgroundUsageAllowabilityPreference).setEnabled(true);
|
||||
verify(mOptimizedPreference).setEnabled(true);
|
||||
verify(mUnrestrictedPreference).setEnabled(true);
|
||||
verify(mBackgroundUsageAllowabilityPreference, never()).setOnPreferenceClickListener(any());
|
||||
verify(mBackgroundUsageAllowabilityPreference).setOnPreferenceChangeListener(any());
|
||||
verify(mOptimizedPreference).setOnPreferenceClickListener(any());
|
||||
verify(mUnrestrictedPreference).setOnPreferenceClickListener(any());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void initPreferences_immutableMode_setDisabledAndSkipSetListeners() {
|
||||
doReturn(false).when(mBatteryOptimizeUtils).isOptimizeModeMutable();
|
||||
|
||||
mBackgroundUsageController.initPreferences();
|
||||
|
||||
verify(mBackgroundUsageAllowabilityPreference).setEnabled(false);
|
||||
verify(mOptimizedPreference).setEnabled(false);
|
||||
verify(mUnrestrictedPreference).setEnabled(false);
|
||||
verify(mBackgroundUsageAllowabilityPreference, never()).setOnPreferenceClickListener(any());
|
||||
verify(mBackgroundUsageAllowabilityPreference, never())
|
||||
.setOnPreferenceChangeListener(any());
|
||||
verify(mOptimizedPreference, never()).setOnPreferenceClickListener(any());
|
||||
verify(mUnrestrictedPreference, never()).setOnPreferenceClickListener(any());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void updatePreferences_setIntoUnrestrictedMode_setExpectedPrefStatus() {
|
||||
mTestMode = BatteryOptimizeUtils.MODE_UNRESTRICTED;
|
||||
|
||||
mBackgroundUsageController.updatePreferences(mTestMode);
|
||||
|
||||
verifyPreferences(mTestMode);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void updatePreferences_setIntoOptimizedMode_setExpectedPrefStatus() {
|
||||
mTestMode = BatteryOptimizeUtils.MODE_OPTIMIZED;
|
||||
|
||||
mBackgroundUsageController.updatePreferences(mTestMode);
|
||||
|
||||
verifyPreferences(mTestMode);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void updatePreferences_setIntoRestrictedMode_setExpectedPrefStatus() {
|
||||
mTestMode = BatteryOptimizeUtils.MODE_RESTRICTED;
|
||||
|
||||
mBackgroundUsageController.updatePreferences(mTestMode);
|
||||
|
||||
verifyPreferences(mTestMode);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void handleBatteryOptimizeModeUpdated_modeChange_setExpectedPrefStatus() {
|
||||
mTestMode = BatteryOptimizeUtils.MODE_RESTRICTED;
|
||||
|
||||
mBackgroundUsageController.handleBatteryOptimizeModeUpdated(
|
||||
BatteryOptimizeUtils.MODE_OPTIMIZED);
|
||||
|
||||
verify(mBatteryOptimizeUtils)
|
||||
.setAppUsageState(
|
||||
BatteryOptimizeUtils.MODE_OPTIMIZED,
|
||||
BatteryOptimizeHistoricalLogEntry.Action.APPLY);
|
||||
assertThat(mTestMode).isEqualTo(BatteryOptimizeUtils.MODE_OPTIMIZED);
|
||||
verifyPreferences(mBatteryOptimizeUtils.getAppOptimizationMode());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void handleBatteryOptimizeModeUpdated_modeNotChange_setExpectedPrefStatus() {
|
||||
mTestMode = BatteryOptimizeUtils.MODE_RESTRICTED;
|
||||
|
||||
mBackgroundUsageController.handleBatteryOptimizeModeUpdated(
|
||||
BatteryOptimizeUtils.MODE_RESTRICTED);
|
||||
|
||||
verify(mBatteryOptimizeUtils, never()).setAppUsageState(anyInt(), any());
|
||||
assertThat(mTestMode).isEqualTo(BatteryOptimizeUtils.MODE_RESTRICTED);
|
||||
verify(mBackgroundUsageController, never()).updatePreferences(anyInt());
|
||||
}
|
||||
|
||||
private void prepareTestBatteryOptimizationUtils() {
|
||||
mBatteryOptimizeUtils = spy(new BatteryOptimizeUtils(mContext, UID, PACKAGE_NAME));
|
||||
Answer<Void> setTestMode =
|
||||
invocation -> {
|
||||
mTestMode = invocation.getArgument(0);
|
||||
return null;
|
||||
};
|
||||
doAnswer(setTestMode).when(mBatteryOptimizeUtils).setAppUsageState(anyInt(), any());
|
||||
Answer<Integer> getTestMode = invocation -> mTestMode;
|
||||
doAnswer(getTestMode).when(mBatteryOptimizeUtils).getAppOptimizationMode();
|
||||
}
|
||||
|
||||
private void verifyPreferences(int mode) {
|
||||
boolean isAllowBackground = mode != BatteryOptimizeUtils.MODE_RESTRICTED;
|
||||
verify(mBackgroundUsageAllowabilityPreference).setChecked(isAllowBackground);
|
||||
verify(mOptimizedPreference).setEnabled(isAllowBackground);
|
||||
verify(mUnrestrictedPreference).setEnabled(isAllowBackground);
|
||||
verify(mOptimizedPreference).setChecked(mode == BatteryOptimizeUtils.MODE_OPTIMIZED);
|
||||
verify(mUnrestrictedPreference).setChecked(mode == BatteryOptimizeUtils.MODE_UNRESTRICTED);
|
||||
}
|
||||
}
|
||||
@@ -1,130 +0,0 @@
|
||||
/*
|
||||
* Copyright (C) 2021 The Android Open Source Project
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package com.android.settings.fuelgauge;
|
||||
|
||||
import static com.google.common.truth.Truth.assertThat;
|
||||
|
||||
import static org.mockito.Mockito.doReturn;
|
||||
import static org.mockito.Mockito.spy;
|
||||
import static org.mockito.Mockito.when;
|
||||
|
||||
import android.content.Context;
|
||||
import android.content.pm.PackageManager;
|
||||
|
||||
import com.android.settingslib.widget.SelectorWithWidgetPreference;
|
||||
|
||||
import org.junit.Before;
|
||||
import org.junit.Test;
|
||||
import org.junit.runner.RunWith;
|
||||
import org.mockito.Mock;
|
||||
import org.mockito.MockitoAnnotations;
|
||||
import org.robolectric.RobolectricTestRunner;
|
||||
import org.robolectric.RuntimeEnvironment;
|
||||
|
||||
@RunWith(RobolectricTestRunner.class)
|
||||
public class OptimizedPreferenceControllerTest {
|
||||
private static final int UID = 12345;
|
||||
private static final String PACKAGE_NAME = "com.android.app";
|
||||
|
||||
private OptimizedPreferenceController mController;
|
||||
private SelectorWithWidgetPreference mPreference;
|
||||
private BatteryOptimizeUtils mBatteryOptimizeUtils;
|
||||
|
||||
@Mock PackageManager mMockPackageManager;
|
||||
|
||||
@Before
|
||||
public void setUp() throws Exception {
|
||||
MockitoAnnotations.initMocks(this);
|
||||
|
||||
Context context = spy(RuntimeEnvironment.application);
|
||||
BatteryUtils.getInstance(context).reset();
|
||||
doReturn(UID)
|
||||
.when(mMockPackageManager)
|
||||
.getPackageUid(PACKAGE_NAME, PackageManager.GET_META_DATA);
|
||||
|
||||
mController = new OptimizedPreferenceController(context, UID, PACKAGE_NAME);
|
||||
mPreference = new SelectorWithWidgetPreference(RuntimeEnvironment.application);
|
||||
mBatteryOptimizeUtils = spy(new BatteryOptimizeUtils(context, UID, PACKAGE_NAME));
|
||||
mController.mBatteryOptimizeUtils = mBatteryOptimizeUtils;
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testUpdateState_invalidPackage_prefEnabled() {
|
||||
when(mBatteryOptimizeUtils.isDisabledForOptimizeModeOnly()).thenReturn(true);
|
||||
|
||||
mController.updateState(mPreference);
|
||||
|
||||
assertThat(mPreference.isEnabled()).isFalse();
|
||||
assertThat(mPreference.isChecked()).isTrue();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testUpdateState_isSystemOrDefaultAppAndOptimizeStates_prefChecked() {
|
||||
when(mBatteryOptimizeUtils.isDisabledForOptimizeModeOnly()).thenReturn(false);
|
||||
when(mBatteryOptimizeUtils.isSystemOrDefaultApp()).thenReturn(true);
|
||||
when(mBatteryOptimizeUtils.getAppOptimizationMode())
|
||||
.thenReturn(BatteryOptimizeUtils.MODE_OPTIMIZED);
|
||||
|
||||
mController.updateState(mPreference);
|
||||
|
||||
assertThat(mPreference.isChecked()).isTrue();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testUpdateState_isSystemOrDefaultApp_prefUnchecked() {
|
||||
when(mBatteryOptimizeUtils.isDisabledForOptimizeModeOnly()).thenReturn(false);
|
||||
when(mBatteryOptimizeUtils.isSystemOrDefaultApp()).thenReturn(true);
|
||||
|
||||
mController.updateState(mPreference);
|
||||
|
||||
assertThat(mPreference.isChecked()).isFalse();
|
||||
assertThat(mPreference.isEnabled()).isFalse();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testUpdateState_isOptimizedStates_prefChecked() {
|
||||
when(mBatteryOptimizeUtils.isDisabledForOptimizeModeOnly()).thenReturn(false);
|
||||
when(mBatteryOptimizeUtils.getAppOptimizationMode())
|
||||
.thenReturn(BatteryOptimizeUtils.MODE_OPTIMIZED);
|
||||
|
||||
mController.updateState(mPreference);
|
||||
|
||||
assertThat(mPreference.isChecked()).isTrue();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testUpdateState_prefUnchecked() {
|
||||
when(mBatteryOptimizeUtils.isDisabledForOptimizeModeOnly()).thenReturn(false);
|
||||
|
||||
mController.updateState(mPreference);
|
||||
|
||||
assertThat(mPreference.isChecked()).isFalse();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testHandlePreferenceTreeClick_samePrefKey_verifyAction() {
|
||||
mPreference.setKey(mController.KEY_OPTIMIZED_PREF);
|
||||
mController.handlePreferenceTreeClick(mPreference);
|
||||
|
||||
assertThat(mController.handlePreferenceTreeClick(mPreference)).isTrue();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testHandlePreferenceTreeClick_incorrectPrefKey_noAction() {
|
||||
assertThat(mController.handlePreferenceTreeClick(mPreference)).isFalse();
|
||||
}
|
||||
}
|
||||
@@ -17,10 +17,10 @@
|
||||
package com.android.settings.fuelgauge;
|
||||
|
||||
import static com.android.settings.SettingsActivity.EXTRA_SHOW_FRAGMENT_ARGUMENTS;
|
||||
import static com.android.settings.fuelgauge.BatteryOptimizeHistoricalLogEntry.Action;
|
||||
|
||||
import static com.google.common.truth.Truth.assertThat;
|
||||
|
||||
import static org.mockito.ArgumentMatchers.any;
|
||||
import static org.mockito.ArgumentMatchers.anyInt;
|
||||
import static org.mockito.ArgumentMatchers.anyString;
|
||||
import static org.mockito.ArgumentMatchers.nullable;
|
||||
@@ -42,16 +42,17 @@ import android.content.pm.PackageManager;
|
||||
import android.graphics.drawable.Drawable;
|
||||
import android.os.Bundle;
|
||||
import android.os.UserHandle;
|
||||
import android.widget.CompoundButton;
|
||||
|
||||
import androidx.fragment.app.FragmentActivity;
|
||||
import androidx.loader.app.LoaderManager;
|
||||
import androidx.test.core.app.ApplicationProvider;
|
||||
|
||||
import com.android.settings.R;
|
||||
import com.android.settings.SettingsActivity;
|
||||
import com.android.settings.fuelgauge.batteryusage.BatteryEntry;
|
||||
import com.android.settings.testutils.FakeFeatureFactory;
|
||||
import com.android.settings.testutils.shadow.ShadowEntityHeaderController;
|
||||
import com.android.settings.testutils.shadow.ShadowHelpUtils;
|
||||
import com.android.settings.widget.EntityHeaderController;
|
||||
import com.android.settingslib.applications.AppUtils;
|
||||
import com.android.settingslib.applications.ApplicationsState;
|
||||
@@ -59,8 +60,6 @@ import com.android.settingslib.applications.instantapps.InstantAppDataProvider;
|
||||
import com.android.settingslib.core.instrumentation.MetricsFeatureProvider;
|
||||
import com.android.settingslib.widget.FooterPreference;
|
||||
import com.android.settingslib.widget.LayoutPreference;
|
||||
import com.android.settingslib.widget.MainSwitchPreference;
|
||||
import com.android.settingslib.widget.SelectorWithWidgetPreference;
|
||||
|
||||
import org.junit.After;
|
||||
import org.junit.Before;
|
||||
@@ -83,36 +82,33 @@ import java.util.concurrent.TimeUnit;
|
||||
@Config(
|
||||
shadows = {
|
||||
ShadowEntityHeaderController.class,
|
||||
ShadowHelpUtils.class,
|
||||
com.android.settings.testutils.shadow.ShadowFragment.class,
|
||||
})
|
||||
public class PowerBackgroundUsageDetailTest {
|
||||
|
||||
@Rule
|
||||
public final MockitoRule mMockitoRule = MockitoJUnit.rule();
|
||||
@Rule public final MockitoRule mMockitoRule = MockitoJUnit.rule();
|
||||
|
||||
private static final String APP_LABEL = "app label";
|
||||
private static final String SUMMARY = "summary";
|
||||
private static final int ICON_ID = 123;
|
||||
private static final int UID = 1;
|
||||
private static final String KEY_PREF_UNRESTRICTED = "unrestricted_preference";
|
||||
private static final String KEY_PREF_OPTIMIZED = "optimized_preference";
|
||||
private static final String KEY_ALLOW_BACKGROUND_USAGE = "allow_background_usage";
|
||||
private static final String PACKAGE_NAME = "com.android.app";
|
||||
private static final String KEY_PREF_HEADER = "header_view";
|
||||
private static final String KEY_FOOTER_PREFERENCE = "app_usage_footer_preference";
|
||||
private static final String INITIATING_PACKAGE_NAME = "com.android.vending";
|
||||
|
||||
private int mTestMode;
|
||||
private Context mContext;
|
||||
private PowerBackgroundUsageDetail mFragment;
|
||||
private FooterPreference mFooterPreference;
|
||||
private MainSwitchPreference mMainSwitchPreference;
|
||||
private MetricsFeatureProvider mMetricsFeatureProvider;
|
||||
private SelectorWithWidgetPreference mOptimizePreference;
|
||||
private SelectorWithWidgetPreference mUnrestrictedPreference;
|
||||
private SettingsActivity mTestActivity;
|
||||
private BatteryOptimizeUtils mBatteryOptimizeUtils;
|
||||
|
||||
@Mock(answer = Answers.RETURNS_DEEP_STUBS)
|
||||
private FragmentActivity mActivity;
|
||||
|
||||
@Mock private EntityHeaderController mEntityHeaderController;
|
||||
@Mock private BatteryOptimizeUtils mBatteryOptimizeUtils;
|
||||
@Mock private LayoutPreference mHeaderPreference;
|
||||
@Mock private ApplicationsState mState;
|
||||
@Mock private Bundle mBundle;
|
||||
@Mock private LoaderManager mLoaderManager;
|
||||
@@ -120,21 +116,26 @@ public class PowerBackgroundUsageDetailTest {
|
||||
@Mock private BatteryEntry mBatteryEntry;
|
||||
@Mock private PackageManager mPackageManager;
|
||||
@Mock private AppOpsManager mAppOpsManager;
|
||||
@Mock private CompoundButton mMockSwitch;
|
||||
@Mock private InstallSourceInfo mInstallSourceInfo;
|
||||
@Mock private LayoutPreference mLayoutPreference;
|
||||
@Mock private FooterPreference mFooterPreference;
|
||||
|
||||
@Before
|
||||
public void setUp() throws Exception {
|
||||
mContext = spy(ApplicationProvider.getApplicationContext());
|
||||
when(mContext.getPackageName()).thenReturn("foo");
|
||||
when(mContext.getPackageName()).thenReturn(PACKAGE_NAME);
|
||||
when(mContext.getPackageManager()).thenReturn(mPackageManager);
|
||||
when(mPackageManager.getInstallSourceInfo(anyString())).thenReturn(mInstallSourceInfo);
|
||||
|
||||
final FakeFeatureFactory fakeFeatureFactory = FakeFeatureFactory.setupForTest();
|
||||
mMetricsFeatureProvider = fakeFeatureFactory.metricsFeatureProvider;
|
||||
|
||||
prepareTestBatteryOptimizationUtils();
|
||||
mFragment = spy(new PowerBackgroundUsageDetail());
|
||||
mFragment.mBatteryOptimizeUtils = mBatteryOptimizeUtils;
|
||||
mFragment.mLogStringBuilder = new StringBuilder();
|
||||
doReturn(mLayoutPreference).when(mFragment).findPreference(KEY_PREF_HEADER);
|
||||
doReturn(mFooterPreference).when(mFragment).findPreference(KEY_FOOTER_PREFERENCE);
|
||||
doReturn(mContext).when(mFragment).getContext();
|
||||
doReturn(mActivity).when(mFragment).getActivity();
|
||||
doReturn(SUMMARY).when(mFragment).getString(anyInt());
|
||||
@@ -169,9 +170,7 @@ public class PowerBackgroundUsageDetailTest {
|
||||
when(mBatteryEntry.getLabel()).thenReturn(APP_LABEL);
|
||||
mBatteryEntry.mIconId = ICON_ID;
|
||||
|
||||
mFragment.mHeaderPreference = mHeaderPreference;
|
||||
mFragment.mState = mState;
|
||||
mFragment.mBatteryOptimizeUtils = mBatteryOptimizeUtils;
|
||||
mAppEntry.info = mock(ApplicationInfo.class);
|
||||
|
||||
mTestActivity = spy(new SettingsActivity());
|
||||
@@ -191,23 +190,12 @@ public class PowerBackgroundUsageDetailTest {
|
||||
.when(mActivity)
|
||||
.startActivityAsUser(captor.capture(), nullable(UserHandle.class));
|
||||
doAnswer(callable).when(mActivity).startActivity(captor.capture());
|
||||
|
||||
mFooterPreference = spy(new FooterPreference(mContext));
|
||||
mMainSwitchPreference = spy(new MainSwitchPreference(mContext));
|
||||
mMainSwitchPreference.setKey(KEY_ALLOW_BACKGROUND_USAGE);
|
||||
mOptimizePreference = spy(new SelectorWithWidgetPreference(mContext));
|
||||
mOptimizePreference.setKey(KEY_PREF_OPTIMIZED);
|
||||
mUnrestrictedPreference = spy(new SelectorWithWidgetPreference(mContext));
|
||||
mUnrestrictedPreference.setKey(KEY_PREF_UNRESTRICTED);
|
||||
mFragment.mFooterPreference = mFooterPreference;
|
||||
mFragment.mMainSwitchPreference = mMainSwitchPreference;
|
||||
mFragment.mOptimizePreference = mOptimizePreference;
|
||||
mFragment.mUnrestrictedPreference = mUnrestrictedPreference;
|
||||
}
|
||||
|
||||
@After
|
||||
public void reset() {
|
||||
ShadowEntityHeaderController.reset();
|
||||
ShadowHelpUtils.reset();
|
||||
}
|
||||
|
||||
@Test
|
||||
@@ -258,91 +246,64 @@ public class PowerBackgroundUsageDetailTest {
|
||||
}
|
||||
|
||||
@Test
|
||||
public void initFooter_hasCorrectString() {
|
||||
when(mBatteryOptimizeUtils.isDisabledForOptimizeModeOnly()).thenReturn(false);
|
||||
when(mBatteryOptimizeUtils.isSystemOrDefaultApp()).thenReturn(false);
|
||||
|
||||
public void initFooter_setExpectedFooterContent() {
|
||||
mFragment.initFooter();
|
||||
|
||||
assertThat(mFooterPreference.getTitle().toString())
|
||||
.isEqualTo("Changing how an app uses your battery can affect its performance.");
|
||||
verify(mFooterPreference)
|
||||
.setTitle(mContext.getString(R.string.manager_battery_usage_footer));
|
||||
verify(mFooterPreference).setLearnMoreAction(any());
|
||||
verify(mFooterPreference)
|
||||
.setLearnMoreText(mContext.getString(R.string.manager_battery_usage_link_a11y));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void onSwitchChanged_fromUnrestrictedModeSetDisabled_becomeRestrictedMode() {
|
||||
final int restrictedMode = BatteryOptimizeUtils.MODE_RESTRICTED;
|
||||
final int optimizedMode = BatteryOptimizeUtils.MODE_OPTIMIZED;
|
||||
mFragment.mOptimizationMode = optimizedMode;
|
||||
public void onPause_optimizationModeIsChanged_logPreference() throws Exception {
|
||||
mFragment.mOptimizationMode = BatteryOptimizeUtils.MODE_OPTIMIZED;
|
||||
when(mBatteryOptimizeUtils.getPackageName()).thenReturn(PACKAGE_NAME);
|
||||
when(mInstallSourceInfo.getInitiatingPackageName()).thenReturn(INITIATING_PACKAGE_NAME);
|
||||
|
||||
mFragment.onCheckedChanged(mMockSwitch, /* isChecked= */ false);
|
||||
|
||||
verify(mOptimizePreference).setEnabled(false);
|
||||
verify(mUnrestrictedPreference).setEnabled(false);
|
||||
verify(mFragment).onRadioButtonClicked(null);
|
||||
verify(mMainSwitchPreference).setChecked(false);
|
||||
assertThat(mFragment.getSelectedPreference()).isEqualTo(restrictedMode);
|
||||
verify(mBatteryOptimizeUtils).setAppUsageState(restrictedMode, Action.APPLY);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void onSwitchChanged_fromRestrictedModeSetEnabled_becomeOptimizedMode() {
|
||||
final int restrictedMode = BatteryOptimizeUtils.MODE_RESTRICTED;
|
||||
final int optimizedMode = BatteryOptimizeUtils.MODE_OPTIMIZED;
|
||||
mFragment.mOptimizationMode = restrictedMode;
|
||||
|
||||
mFragment.onCheckedChanged(mMockSwitch, /* isChecked= */ true);
|
||||
|
||||
verify(mOptimizePreference).setEnabled(true);
|
||||
verify(mUnrestrictedPreference).setEnabled(true);
|
||||
verify(mFragment).onRadioButtonClicked(mOptimizePreference);
|
||||
verify(mMainSwitchPreference).setChecked(true);
|
||||
verify(mOptimizePreference).setChecked(true);
|
||||
assertThat(mFragment.getSelectedPreference()).isEqualTo(optimizedMode);
|
||||
verify(mBatteryOptimizeUtils).setAppUsageState(optimizedMode, Action.APPLY);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void onPause_optimizationModeChanged_logPreference() throws Exception {
|
||||
final String packageName = "testPackageName";
|
||||
final int restrictedMode = BatteryOptimizeUtils.MODE_RESTRICTED;
|
||||
final int optimizedMode = BatteryOptimizeUtils.MODE_OPTIMIZED;
|
||||
mFragment.mOptimizationMode = restrictedMode;
|
||||
when(mBatteryOptimizeUtils.getPackageName()).thenReturn(packageName);
|
||||
when(mInstallSourceInfo.getInitiatingPackageName()).thenReturn("com.android.vending");
|
||||
|
||||
mFragment.onCheckedChanged(mMockSwitch, /* isChecked= */ true);
|
||||
verify(mBatteryOptimizeUtils).setAppUsageState(optimizedMode, Action.APPLY);
|
||||
when(mBatteryOptimizeUtils.getAppOptimizationMode()).thenReturn(optimizedMode);
|
||||
mTestMode = BatteryOptimizeUtils.MODE_UNRESTRICTED;
|
||||
assertThat(mBatteryOptimizeUtils.getAppOptimizationMode())
|
||||
.isEqualTo(BatteryOptimizeUtils.MODE_UNRESTRICTED);
|
||||
mFragment.onPause();
|
||||
|
||||
TimeUnit.SECONDS.sleep(1);
|
||||
verify(mMetricsFeatureProvider)
|
||||
.action(
|
||||
SettingsEnums.LEAVE_POWER_USAGE_MANAGE_BACKGROUND,
|
||||
SettingsEnums.ACTION_APP_BATTERY_USAGE_OPTIMIZED,
|
||||
SettingsEnums.ACTION_APP_BATTERY_USAGE_UNRESTRICTED,
|
||||
SettingsEnums.FUELGAUGE_POWER_USAGE_MANAGE_BACKGROUND,
|
||||
packageName,
|
||||
PACKAGE_NAME,
|
||||
/* consumed battery */ 0);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void onPause_optimizationModeIsNotChanged_notInvokeLogging() throws Exception {
|
||||
final String packageName = "testPackageName";
|
||||
final int restrictedMode = BatteryOptimizeUtils.MODE_RESTRICTED;
|
||||
final int optimizedMode = BatteryOptimizeUtils.MODE_OPTIMIZED;
|
||||
mFragment.mOptimizationMode = restrictedMode;
|
||||
when(mBatteryOptimizeUtils.getPackageName()).thenReturn(packageName);
|
||||
when(mInstallSourceInfo.getInitiatingPackageName()).thenReturn("com.android.vending");
|
||||
mFragment.mOptimizationMode = BatteryOptimizeUtils.MODE_OPTIMIZED;
|
||||
when(mBatteryOptimizeUtils.getPackageName()).thenReturn(PACKAGE_NAME);
|
||||
when(mInstallSourceInfo.getInitiatingPackageName()).thenReturn(INITIATING_PACKAGE_NAME);
|
||||
|
||||
mFragment.onCheckedChanged(mMockSwitch, /* isChecked= */ true);
|
||||
verify(mBatteryOptimizeUtils).setAppUsageState(optimizedMode, Action.APPLY);
|
||||
when(mBatteryOptimizeUtils.getAppOptimizationMode()).thenReturn(optimizedMode);
|
||||
mFragment.onCheckedChanged(mMockSwitch, /* isChecked= */ false);
|
||||
verify(mBatteryOptimizeUtils).setAppUsageState(restrictedMode, Action.APPLY);
|
||||
when(mBatteryOptimizeUtils.getAppOptimizationMode()).thenReturn(restrictedMode);
|
||||
mTestMode = BatteryOptimizeUtils.MODE_UNRESTRICTED;
|
||||
assertThat(mBatteryOptimizeUtils.getAppOptimizationMode())
|
||||
.isEqualTo(BatteryOptimizeUtils.MODE_UNRESTRICTED);
|
||||
mTestMode = BatteryOptimizeUtils.MODE_OPTIMIZED;
|
||||
assertThat(mBatteryOptimizeUtils.getAppOptimizationMode())
|
||||
.isEqualTo(BatteryOptimizeUtils.MODE_OPTIMIZED);
|
||||
mFragment.onPause();
|
||||
|
||||
TimeUnit.SECONDS.sleep(1);
|
||||
verifyNoInteractions(mMetricsFeatureProvider);
|
||||
}
|
||||
|
||||
private void prepareTestBatteryOptimizationUtils() {
|
||||
mBatteryOptimizeUtils = spy(new BatteryOptimizeUtils(mContext, UID, PACKAGE_NAME));
|
||||
Answer<Void> setTestMode =
|
||||
invocation -> {
|
||||
mTestMode = invocation.getArgument(0);
|
||||
return null;
|
||||
};
|
||||
doAnswer(setTestMode).when(mBatteryOptimizeUtils).setAppUsageState(anyInt(), any());
|
||||
Answer<Integer> getTestMode = invocation -> mTestMode;
|
||||
doAnswer(getTestMode).when(mBatteryOptimizeUtils).getAppOptimizationMode();
|
||||
}
|
||||
}
|
||||
|
||||
@@ -1,148 +0,0 @@
|
||||
/*
|
||||
* Copyright (C) 2021 The Android Open Source Project
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package com.android.settings.fuelgauge;
|
||||
|
||||
import static com.google.common.truth.Truth.assertThat;
|
||||
|
||||
import static org.mockito.Mockito.doReturn;
|
||||
import static org.mockito.Mockito.spy;
|
||||
import static org.mockito.Mockito.when;
|
||||
|
||||
import android.content.Context;
|
||||
import android.content.pm.PackageManager;
|
||||
|
||||
import com.android.settingslib.widget.SelectorWithWidgetPreference;
|
||||
|
||||
import org.junit.Before;
|
||||
import org.junit.Test;
|
||||
import org.junit.runner.RunWith;
|
||||
import org.mockito.Mock;
|
||||
import org.mockito.MockitoAnnotations;
|
||||
import org.robolectric.RobolectricTestRunner;
|
||||
import org.robolectric.RuntimeEnvironment;
|
||||
|
||||
@RunWith(RobolectricTestRunner.class)
|
||||
public class UnrestrictedPreferenceControllerTest {
|
||||
private static final int UID = 12345;
|
||||
private static final String PACKAGE_NAME = "com.android.app";
|
||||
|
||||
private UnrestrictedPreferenceController mController;
|
||||
private SelectorWithWidgetPreference mPreference;
|
||||
private BatteryOptimizeUtils mBatteryOptimizeUtils;
|
||||
|
||||
@Mock PackageManager mMockPackageManager;
|
||||
|
||||
@Before
|
||||
public void setUp() throws Exception {
|
||||
MockitoAnnotations.initMocks(this);
|
||||
|
||||
Context context = spy(RuntimeEnvironment.application);
|
||||
BatteryUtils.getInstance(context).reset();
|
||||
doReturn(UID)
|
||||
.when(mMockPackageManager)
|
||||
.getPackageUid(PACKAGE_NAME, PackageManager.GET_META_DATA);
|
||||
|
||||
mController = new UnrestrictedPreferenceController(context, UID, PACKAGE_NAME);
|
||||
mPreference = new SelectorWithWidgetPreference(RuntimeEnvironment.application);
|
||||
mBatteryOptimizeUtils = spy(new BatteryOptimizeUtils(context, UID, PACKAGE_NAME));
|
||||
mController.mBatteryOptimizeUtils = mBatteryOptimizeUtils;
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testUpdateState_isValidPackage_prefEnabled() {
|
||||
when(mBatteryOptimizeUtils.isDisabledForOptimizeModeOnly()).thenReturn(false);
|
||||
when(mBatteryOptimizeUtils.isSystemOrDefaultApp()).thenReturn(false);
|
||||
|
||||
mController.updateState(mPreference);
|
||||
|
||||
assertThat(mBatteryOptimizeUtils.isOptimizeModeMutable()).isTrue();
|
||||
assertThat(mPreference.isEnabled()).isTrue();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testUpdateState_invalidPackage_prefDisabled() {
|
||||
when(mBatteryOptimizeUtils.isDisabledForOptimizeModeOnly()).thenReturn(true);
|
||||
when(mBatteryOptimizeUtils.isSystemOrDefaultApp()).thenReturn(false);
|
||||
|
||||
mController.updateState(mPreference);
|
||||
|
||||
assertThat(mBatteryOptimizeUtils.isOptimizeModeMutable()).isFalse();
|
||||
assertThat(mPreference.isEnabled()).isFalse();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testUpdateState_isSystemOrDefaultAppAndUnrestrictedStates_prefChecked() {
|
||||
when(mBatteryOptimizeUtils.isDisabledForOptimizeModeOnly()).thenReturn(false);
|
||||
when(mBatteryOptimizeUtils.isSystemOrDefaultApp()).thenReturn(true);
|
||||
when(mBatteryOptimizeUtils.getAppOptimizationMode())
|
||||
.thenReturn(BatteryOptimizeUtils.MODE_UNRESTRICTED);
|
||||
|
||||
mController.updateState(mPreference);
|
||||
|
||||
assertThat(mPreference.isChecked()).isTrue();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testUpdateState_isSystemOrDefaultApp_prefUnchecked() {
|
||||
when(mBatteryOptimizeUtils.isDisabledForOptimizeModeOnly()).thenReturn(false);
|
||||
when(mBatteryOptimizeUtils.isSystemOrDefaultApp()).thenReturn(true);
|
||||
when(mBatteryOptimizeUtils.getAppOptimizationMode())
|
||||
.thenReturn(BatteryOptimizeUtils.MODE_OPTIMIZED);
|
||||
|
||||
mController.updateState(mPreference);
|
||||
|
||||
assertThat(mPreference.isEnabled()).isFalse();
|
||||
assertThat(mPreference.isChecked()).isFalse();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testUpdateState_isUnrestrictedStates_prefChecked() {
|
||||
when(mBatteryOptimizeUtils.isOptimizeModeMutable()).thenReturn(true);
|
||||
when(mBatteryOptimizeUtils.getAppOptimizationMode())
|
||||
.thenReturn(BatteryOptimizeUtils.MODE_UNRESTRICTED);
|
||||
|
||||
mController.updateState(mPreference);
|
||||
|
||||
assertThat(mPreference.isEnabled()).isTrue();
|
||||
assertThat(mPreference.isChecked()).isTrue();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testUpdateState_prefUnchecked() {
|
||||
when(mBatteryOptimizeUtils.isOptimizeModeMutable()).thenReturn(true);
|
||||
when(mBatteryOptimizeUtils.getAppOptimizationMode())
|
||||
.thenReturn(BatteryOptimizeUtils.MODE_OPTIMIZED);
|
||||
|
||||
mController.updateState(mPreference);
|
||||
|
||||
assertThat(mPreference.isEnabled()).isTrue();
|
||||
assertThat(mPreference.isChecked()).isFalse();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testHandlePreferenceTreeClick_samePrefKey_verifyAction() {
|
||||
mPreference.setKey(mController.KEY_UNRESTRICTED_PREF);
|
||||
mController.handlePreferenceTreeClick(mPreference);
|
||||
|
||||
assertThat(mController.handlePreferenceTreeClick(mPreference)).isTrue();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testHandlePreferenceTreeClick_incorrectPrefKey_noAction() {
|
||||
assertThat(mController.handlePreferenceTreeClick(mPreference)).isFalse();
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user