diff --git a/AndroidManifest.xml b/AndroidManifest.xml index 66e4442b355..9582126cb3f 100644 --- a/AndroidManifest.xml +++ b/AndroidManifest.xml @@ -114,6 +114,7 @@ + intentCaptor = ArgumentCaptor.forClass(Intent.class); mActivity.handleSafetyCenterRedirection(); @@ -94,12 +89,7 @@ public class PrivacyDashboardActivityTest { @Test public void onCreate_whenSafetyCenterDisabled_doesntRedirectToSafetyCenter() { - DeviceConfig.setProperty( - DeviceConfig.NAMESPACE_PRIVACY, - SafetyCenterStatus.SAFETY_CENTER_IS_ENABLED, - /* value = */ Boolean.toString(false), - /* makeDefault = */ false); - + when(mSafetyCenterStatusHolder.isEnabled(any(Context.class))).thenReturn(false); mActivity.handleSafetyCenterRedirection(); verify(mActivity, times(0)).startActivity(any()); diff --git a/tests/unit/src/com/android/settings/privacy/TopLevelPrivacyEntryPreferenceControllerTest.java b/tests/unit/src/com/android/settings/privacy/TopLevelPrivacyEntryPreferenceControllerTest.java index 570df739488..495700a10dd 100644 --- a/tests/unit/src/com/android/settings/privacy/TopLevelPrivacyEntryPreferenceControllerTest.java +++ b/tests/unit/src/com/android/settings/privacy/TopLevelPrivacyEntryPreferenceControllerTest.java @@ -18,20 +18,22 @@ package com.android.settings.privacy; import static com.google.common.truth.Truth.assertThat; -import android.content.Context; -import android.provider.DeviceConfig; -import android.provider.Settings; +import static org.mockito.ArgumentMatchers.any; +import static org.mockito.Mockito.when; +import android.content.Context; + +import androidx.test.core.app.ApplicationProvider; import androidx.test.ext.junit.runners.AndroidJUnit4; -import com.android.settings.safetycenter.SafetyCenterStatus; +import com.android.settings.safetycenter.SafetyCenterStatusHolder; import com.android.settings.security.TopLevelSecurityEntryPreferenceController; -import org.junit.After; import org.junit.Before; import org.junit.Test; import org.junit.runner.RunWith; import org.mockito.Mock; +import org.mockito.MockitoAnnotations; @RunWith(AndroidJUnit4.class) public class TopLevelPrivacyEntryPreferenceControllerTest { @@ -41,30 +43,21 @@ public class TopLevelPrivacyEntryPreferenceControllerTest { private TopLevelPrivacyEntryPreferenceController mTopLevelPrivacyEntryPreferenceController; @Mock - private Context mContext; + private SafetyCenterStatusHolder mSafetyCenterStatusHolder; @Before public void setUp() { - DeviceConfig.resetToDefaults(Settings.RESET_MODE_PACKAGE_DEFAULTS, - DeviceConfig.NAMESPACE_PRIVACY); + MockitoAnnotations.initMocks(this); + SafetyCenterStatusHolder.sInstance = mSafetyCenterStatusHolder; mTopLevelPrivacyEntryPreferenceController = - new TopLevelPrivacyEntryPreferenceController(mContext, PREFERENCE_KEY); - } - - @After - public void tearDown() { - DeviceConfig.resetToDefaults(Settings.RESET_MODE_PACKAGE_DEFAULTS, - DeviceConfig.NAMESPACE_PRIVACY); + new TopLevelPrivacyEntryPreferenceController( + ApplicationProvider.getApplicationContext(), PREFERENCE_KEY); } @Test public void getAvailabilityStatus_whenSafetyCenterEnabled_returnsUnavailable() { - DeviceConfig.setProperty( - DeviceConfig.NAMESPACE_PRIVACY, - SafetyCenterStatus.SAFETY_CENTER_IS_ENABLED, - /* value = */ Boolean.toString(true), - /* makeDefault = */ false); + when(mSafetyCenterStatusHolder.isEnabled(any(Context.class))).thenReturn(true); assertThat(mTopLevelPrivacyEntryPreferenceController.getAvailabilityStatus()) .isEqualTo(TopLevelSecurityEntryPreferenceController.CONDITIONALLY_UNAVAILABLE); @@ -72,11 +65,7 @@ public class TopLevelPrivacyEntryPreferenceControllerTest { @Test public void getAvailabilityStatus_whenSafetyCenterDisabled_returnsAvailable() { - DeviceConfig.setProperty( - DeviceConfig.NAMESPACE_PRIVACY, - SafetyCenterStatus.SAFETY_CENTER_IS_ENABLED, - /* value = */ Boolean.toString(false), - /* makeDefault = */ false); + when(mSafetyCenterStatusHolder.isEnabled(any(Context.class))).thenReturn(false); assertThat(mTopLevelPrivacyEntryPreferenceController.getAvailabilityStatus()) .isEqualTo(TopLevelSecurityEntryPreferenceController.AVAILABLE); diff --git a/tests/unit/src/com/android/settings/safetycenter/SafetyCenterStatusHolderTest.java b/tests/unit/src/com/android/settings/safetycenter/SafetyCenterStatusHolderTest.java new file mode 100644 index 00000000000..2a31eb8e14a --- /dev/null +++ b/tests/unit/src/com/android/settings/safetycenter/SafetyCenterStatusHolderTest.java @@ -0,0 +1,56 @@ +/* + * Copyright (C) 2022 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.safetycenter; + +import static com.google.common.truth.Truth.assertThat; + +import static org.mockito.Mockito.when; + +import android.content.Context; +import android.safetycenter.SafetyCenterManager; + +import androidx.test.ext.junit.runners.AndroidJUnit4; + +import org.junit.Before; +import org.junit.Test; +import org.junit.runner.RunWith; +import org.mockito.Mock; +import org.mockito.MockitoAnnotations; + +@RunWith(AndroidJUnit4.class) +public class SafetyCenterStatusHolderTest { + + @Mock + private Context mContext; + + @Before + public void setUp() { + MockitoAnnotations.initMocks(this); + } + + @Test + public void isEnabled_whenContextNull_returnsFalse() { + assertThat(SafetyCenterStatusHolder.get().isEnabled(null)).isFalse(); + } + + @Test + public void isEnabled_whenSystemServiceNull_returnsFalse() { + when(mContext.getSystemService(SafetyCenterManager.class)).thenReturn(null); + + assertThat(SafetyCenterStatusHolder.get().isEnabled(mContext)).isFalse(); + } +} diff --git a/tests/unit/src/com/android/settings/safetycenter/SafetyCenterStatusTest.java b/tests/unit/src/com/android/settings/safetycenter/SafetyCenterStatusTest.java deleted file mode 100644 index f36fbe1f552..00000000000 --- a/tests/unit/src/com/android/settings/safetycenter/SafetyCenterStatusTest.java +++ /dev/null @@ -1,67 +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.safetycenter; - -import static com.google.common.truth.Truth.assertThat; - -import android.provider.DeviceConfig; -import android.provider.Settings; - -import androidx.test.ext.junit.runners.AndroidJUnit4; - -import org.junit.After; -import org.junit.Before; -import org.junit.Test; -import org.junit.runner.RunWith; - -@RunWith(AndroidJUnit4.class) -public class SafetyCenterStatusTest { - - @Before - public void setUp() { - DeviceConfig.resetToDefaults(Settings.RESET_MODE_PACKAGE_DEFAULTS, - DeviceConfig.NAMESPACE_PRIVACY); - } - - @After - public void tearDown() { - DeviceConfig.resetToDefaults(Settings.RESET_MODE_PACKAGE_DEFAULTS, - DeviceConfig.NAMESPACE_PRIVACY); - } - - @Test - public void isEnabled_whenFlagTrue_returnsTrue() { - DeviceConfig.setProperty( - DeviceConfig.NAMESPACE_PRIVACY, - SafetyCenterStatus.SAFETY_CENTER_IS_ENABLED, - /* value = */ Boolean.toString(true), - /* makeDefault = */ false); - - assertThat(SafetyCenterStatus.isEnabled()).isTrue(); - } - - @Test - public void isEnabled_whenFlagFalse_returnsFalse() { - DeviceConfig.setProperty( - DeviceConfig.NAMESPACE_PRIVACY, - SafetyCenterStatus.SAFETY_CENTER_IS_ENABLED, - /* value = */ Boolean.toString(false), - /* makeDefault = */ false); - - assertThat(SafetyCenterStatus.isEnabled()).isFalse(); - } -} diff --git a/tests/unit/src/com/android/settings/safetycenter/TopLevelSafetyCenterEntryPreferenceControllerTest.java b/tests/unit/src/com/android/settings/safetycenter/TopLevelSafetyCenterEntryPreferenceControllerTest.java index 907fb99d00c..e9c451e4c31 100644 --- a/tests/unit/src/com/android/settings/safetycenter/TopLevelSafetyCenterEntryPreferenceControllerTest.java +++ b/tests/unit/src/com/android/settings/safetycenter/TopLevelSafetyCenterEntryPreferenceControllerTest.java @@ -22,18 +22,16 @@ import static org.mockito.ArgumentMatchers.any; import static org.mockito.Mockito.doNothing; import static org.mockito.Mockito.doThrow; import static org.mockito.Mockito.verify; +import static org.mockito.Mockito.when; import android.content.ActivityNotFoundException; import android.content.Context; import android.content.Intent; -import android.provider.DeviceConfig; -import android.provider.Settings; import androidx.preference.Preference; import androidx.test.core.app.ApplicationProvider; import androidx.test.ext.junit.runners.AndroidJUnit4; -import org.junit.After; import org.junit.Before; import org.junit.Test; import org.junit.runner.RunWith; @@ -50,12 +48,16 @@ public class TopLevelSafetyCenterEntryPreferenceControllerTest { mTopLevelSafetyCenterEntryPreferenceController; private Preference mPreference; + @Mock + private SafetyCenterStatusHolder mSafetyCenterStatusHolder; + @Mock private Context mContext; @Before public void setUp() { MockitoAnnotations.initMocks(this); + SafetyCenterStatusHolder.sInstance = mSafetyCenterStatusHolder; mPreference = new Preference(ApplicationProvider.getApplicationContext()); mPreference.setKey(PREFERENCE_KEY); @@ -63,14 +65,6 @@ public class TopLevelSafetyCenterEntryPreferenceControllerTest { doNothing().when(mContext).startActivity(any(Intent.class)); mTopLevelSafetyCenterEntryPreferenceController = new TopLevelSafetyCenterEntryPreferenceController(mContext, PREFERENCE_KEY); - DeviceConfig.resetToDefaults(Settings.RESET_MODE_PACKAGE_DEFAULTS, - DeviceConfig.NAMESPACE_PRIVACY); - } - - @After - public void tearDown() { - DeviceConfig.resetToDefaults(Settings.RESET_MODE_PACKAGE_DEFAULTS, - DeviceConfig.NAMESPACE_PRIVACY); } @Test @@ -110,11 +104,7 @@ public class TopLevelSafetyCenterEntryPreferenceControllerTest { @Test public void getAvailabilityStatus_whenSafetyCenterDisabled_returnsUnavailable() { - DeviceConfig.setProperty( - DeviceConfig.NAMESPACE_PRIVACY, - SafetyCenterStatus.SAFETY_CENTER_IS_ENABLED, - /* value = */ Boolean.toString(false), - /* makeDefault = */ false); + when(mSafetyCenterStatusHolder.isEnabled(any(Context.class))).thenReturn(false); assertThat(mTopLevelSafetyCenterEntryPreferenceController.getAvailabilityStatus()) .isEqualTo(TopLevelSafetyCenterEntryPreferenceController.CONDITIONALLY_UNAVAILABLE); @@ -122,11 +112,7 @@ public class TopLevelSafetyCenterEntryPreferenceControllerTest { @Test public void getAvailabilityStatus_whenSafetyCenterEnabled_returnsAvailable() { - DeviceConfig.setProperty( - DeviceConfig.NAMESPACE_PRIVACY, - SafetyCenterStatus.SAFETY_CENTER_IS_ENABLED, - /* value = */ Boolean.toString(true), - /* makeDefault = */ false); + when(mSafetyCenterStatusHolder.isEnabled(any(Context.class))).thenReturn(true); assertThat(mTopLevelSafetyCenterEntryPreferenceController.getAvailabilityStatus()) .isEqualTo(TopLevelSafetyCenterEntryPreferenceController.AVAILABLE); diff --git a/tests/unit/src/com/android/settings/security/SecurityAdvancedSettingsTest.java b/tests/unit/src/com/android/settings/security/SecurityAdvancedSettingsTest.java index e18b2583f58..ac91abba547 100644 --- a/tests/unit/src/com/android/settings/security/SecurityAdvancedSettingsTest.java +++ b/tests/unit/src/com/android/settings/security/SecurityAdvancedSettingsTest.java @@ -18,17 +18,18 @@ package com.android.settings.security; import static com.google.common.truth.Truth.assertThat; +import static org.mockito.ArgumentMatchers.any; +import static org.mockito.Mockito.spy; import static org.mockito.Mockito.when; import android.content.Context; import android.os.Looper; -import android.provider.DeviceConfig; import androidx.test.annotation.UiThreadTest; import androidx.test.core.app.ApplicationProvider; import androidx.test.ext.junit.runners.AndroidJUnit4; -import com.android.settings.safetycenter.SafetyCenterStatus; +import com.android.settings.safetycenter.SafetyCenterStatusHolder; import com.android.settings.testutils.FakeFeatureFactory; import com.android.settings.testutils.ResourcesUtils; import com.android.settingslib.drawer.CategoryKey; @@ -36,6 +37,8 @@ import com.android.settingslib.drawer.CategoryKey; import org.junit.Before; import org.junit.Test; import org.junit.runner.RunWith; +import org.mockito.Mock; +import org.mockito.MockitoAnnotations; @RunWith(AndroidJUnit4.class) public class SecurityAdvancedSettingsTest { @@ -47,16 +50,22 @@ public class SecurityAdvancedSettingsTest { private Context mContext; private SecurityAdvancedSettings mSecurityAdvancedSettings; + @Mock + private SafetyCenterStatusHolder mSafetyCenterStatusHolder; + @Before @UiThreadTest public void setUp() { + MockitoAnnotations.initMocks(this); if (Looper.myLooper() == null) { Looper.prepare(); } + SafetyCenterStatusHolder.sInstance = mSafetyCenterStatusHolder; mContext = ApplicationProvider.getApplicationContext(); - mSecurityAdvancedSettings = new SecurityAdvancedSettings(); + mSecurityAdvancedSettings = spy(new SecurityAdvancedSettings()); + when(mSecurityAdvancedSettings.getContext()).thenReturn(mContext); } @Test @@ -67,7 +76,7 @@ public class SecurityAdvancedSettingsTest { @Test public void getCategoryKey_whenSafetyCenterIsEnabled_returnsSecurity() { - setSafetyCenterEnabled(true); + when(mSafetyCenterStatusHolder.isEnabled(any())).thenReturn(true); assertThat(mSecurityAdvancedSettings.getCategoryKey()) .isEqualTo(CategoryKey.CATEGORY_SECURITY_ADVANCED_SETTINGS); @@ -75,7 +84,7 @@ public class SecurityAdvancedSettingsTest { @Test public void getCategoryKey_whenAlternativeFragmentPresented_returnsAlternative() { - setSafetyCenterEnabled(false); + when(mSafetyCenterStatusHolder.isEnabled(any(Context.class))).thenReturn(false); setupAlternativeFragment(true, ALTERNATIVE_CATEGORY_KEY); assertThat(mSecurityAdvancedSettings.getCategoryKey()) @@ -84,7 +93,7 @@ public class SecurityAdvancedSettingsTest { @Test public void getCategoryKey_whenNoAlternativeFragmentPresented_returnsLegacy() { - setSafetyCenterEnabled(false); + when(mSafetyCenterStatusHolder.isEnabled(any(Context.class))).thenReturn(false); setupAlternativeFragment(false, null); assertThat(mSecurityAdvancedSettings.getCategoryKey()) @@ -95,14 +104,6 @@ public class SecurityAdvancedSettingsTest { return ResourcesUtils.getResourcesId(mContext, "xml", resName); } - private void setSafetyCenterEnabled(boolean isEnabled) { - DeviceConfig.setProperty( - DeviceConfig.NAMESPACE_PRIVACY, - SafetyCenterStatus.SAFETY_CENTER_IS_ENABLED, - /* value = */ Boolean.toString(isEnabled), - /* makeDefault = */ false); - } - private void setupAlternativeFragment(boolean hasAlternativeFragment, String alternativeCategoryKey) { final FakeFeatureFactory fakeFeatureFactory = FakeFeatureFactory.setupForTest(); diff --git a/tests/unit/src/com/android/settings/security/SecurityDashboardActivityTest.java b/tests/unit/src/com/android/settings/security/SecurityDashboardActivityTest.java index 040661693cf..33486638bd9 100644 --- a/tests/unit/src/com/android/settings/security/SecurityDashboardActivityTest.java +++ b/tests/unit/src/com/android/settings/security/SecurityDashboardActivityTest.java @@ -25,22 +25,22 @@ import static org.mockito.Mockito.times; import static org.mockito.Mockito.verify; import static org.mockito.Mockito.when; +import android.content.Context; import android.content.Intent; -import android.provider.DeviceConfig; import androidx.test.ext.junit.runners.AndroidJUnit4; import androidx.test.platform.app.InstrumentationRegistry; import com.android.settings.Settings; import com.android.settings.SettingsActivity; -import com.android.settings.safetycenter.SafetyCenterStatus; +import com.android.settings.safetycenter.SafetyCenterStatusHolder; import com.android.settings.testutils.FakeFeatureFactory; -import org.junit.After; import org.junit.Before; import org.junit.Test; import org.junit.runner.RunWith; import org.mockito.ArgumentCaptor; +import org.mockito.Mock; import org.mockito.MockitoAnnotations; @RunWith(AndroidJUnit4.class) @@ -52,13 +52,16 @@ public class SecurityDashboardActivityTest { private Settings.SecurityDashboardActivity mActivity; private Intent mDefaultIntent; + @Mock + private SafetyCenterStatusHolder mSafetyCenterStatusHolder; + @Before public void setUp() { MockitoAnnotations.initMocks(this); FakeFeatureFactory mFeatureFactory = FakeFeatureFactory.setupForTest(); mSecuritySettingsFeatureProvider = mFeatureFactory.getSecuritySettingsFeatureProvider(); - DeviceConfig.resetToDefaults(android.provider.Settings.RESET_MODE_PACKAGE_DEFAULTS, - DeviceConfig.NAMESPACE_PRIVACY); + SafetyCenterStatusHolder.sInstance = mSafetyCenterStatusHolder; + mDefaultIntent = new Intent(); mDefaultIntent.setAction(android.provider.Settings.ACTION_SECURITY_SETTINGS); mDefaultIntent.setClass(InstrumentationRegistry.getInstrumentation().getTargetContext(), @@ -79,12 +82,6 @@ public class SecurityDashboardActivityTest { doNothing().when(mActivity).startActivity(any(Intent.class)); } - @After - public void tearDown() { - DeviceConfig.resetToDefaults(android.provider.Settings.RESET_MODE_PACKAGE_DEFAULTS, - DeviceConfig.NAMESPACE_PRIVACY); - } - @Test public void noAvailableAlternativeFragmentAvailable_defaultFragmentSet() { when(mSecuritySettingsFeatureProvider.hasAlternativeSecuritySettingsFragment()) @@ -125,11 +122,7 @@ public class SecurityDashboardActivityTest { @Test public void onCreate_whenSafetyCenterEnabled_redirectsToSafetyCenter() { - DeviceConfig.setProperty( - DeviceConfig.NAMESPACE_PRIVACY, - SafetyCenterStatus.SAFETY_CENTER_IS_ENABLED, - /* value = */ Boolean.toString(true), - /* makeDefault = */ false); + when(mSafetyCenterStatusHolder.isEnabled(any(Context.class))).thenReturn(true); final ArgumentCaptor intentCaptor = ArgumentCaptor.forClass(Intent.class); mActivity.handleSafetyCenterRedirection(); @@ -140,11 +133,7 @@ public class SecurityDashboardActivityTest { @Test public void onCreate_whenSafetyCenterDisabled_doesntRedirectToSafetyCenter() { - DeviceConfig.setProperty( - DeviceConfig.NAMESPACE_PRIVACY, - SafetyCenterStatus.SAFETY_CENTER_IS_ENABLED, - /* value = */ Boolean.toString(false), - /* makeDefault = */ false); + when(mSafetyCenterStatusHolder.isEnabled(any(Context.class))).thenReturn(false); mActivity.handleSafetyCenterRedirection(); diff --git a/tests/unit/src/com/android/settings/security/TopLevelSecurityEntryPreferenceControllerTest.java b/tests/unit/src/com/android/settings/security/TopLevelSecurityEntryPreferenceControllerTest.java index 7e83ca7dcf4..81ebda6f4ac 100644 --- a/tests/unit/src/com/android/settings/security/TopLevelSecurityEntryPreferenceControllerTest.java +++ b/tests/unit/src/com/android/settings/security/TopLevelSecurityEntryPreferenceControllerTest.java @@ -25,18 +25,15 @@ import static org.mockito.Mockito.when; import android.content.Context; import android.content.Intent; -import android.provider.DeviceConfig; -import android.provider.Settings; import androidx.preference.Preference; import androidx.test.core.app.ApplicationProvider; import androidx.test.ext.junit.runners.AndroidJUnit4; import com.android.settings.SettingsActivity; -import com.android.settings.safetycenter.SafetyCenterStatus; +import com.android.settings.safetycenter.SafetyCenterStatusHolder; import com.android.settings.testutils.FakeFeatureFactory; -import org.junit.After; import org.junit.Before; import org.junit.Test; import org.junit.runner.RunWith; @@ -57,15 +54,15 @@ public class TopLevelSecurityEntryPreferenceControllerTest { @Mock private Context mContext; + @Mock + private SafetyCenterStatusHolder mSafetyCenterStatusHolder; @Before public void setUp() { MockitoAnnotations.initMocks(this); mFeatureFactory = FakeFeatureFactory.setupForTest(); mSecuritySettingsFeatureProvider = mFeatureFactory.getSecuritySettingsFeatureProvider(); - - DeviceConfig.resetToDefaults(Settings.RESET_MODE_PACKAGE_DEFAULTS, - DeviceConfig.NAMESPACE_PRIVACY); + SafetyCenterStatusHolder.sInstance = mSafetyCenterStatusHolder; mPreference = new Preference(ApplicationProvider.getApplicationContext()); mPreference.setKey(PREFERENCE_KEY); @@ -75,12 +72,6 @@ public class TopLevelSecurityEntryPreferenceControllerTest { new TopLevelSecurityEntryPreferenceController(mContext, PREFERENCE_KEY); } - @After - public void tearDown() { - DeviceConfig.resetToDefaults(Settings.RESET_MODE_PACKAGE_DEFAULTS, - DeviceConfig.NAMESPACE_PRIVACY); - } - @Test public void handlePreferenceTreeClick_forDifferentPreferenceKey_isNotHandled() { Preference preference = new Preference(ApplicationProvider.getApplicationContext()); @@ -137,11 +128,7 @@ public class TopLevelSecurityEntryPreferenceControllerTest { @Test public void getAvailabilityStatus_whenSafetyCenterEnabled_returnsUnavailable() { - DeviceConfig.setProperty( - DeviceConfig.NAMESPACE_PRIVACY, - SafetyCenterStatus.SAFETY_CENTER_IS_ENABLED, - /* value = */ Boolean.toString(true), - /* makeDefault = */ false); + when(mSafetyCenterStatusHolder.isEnabled(any(Context.class))).thenReturn(true); assertThat(mTopLevelSecurityEntryPreferenceController.getAvailabilityStatus()) .isEqualTo(TopLevelSecurityEntryPreferenceController.CONDITIONALLY_UNAVAILABLE); @@ -149,11 +136,7 @@ public class TopLevelSecurityEntryPreferenceControllerTest { @Test public void getAvailabilityStatus_whenSafetyCenterDisabled_returnsAvailable() { - DeviceConfig.setProperty( - DeviceConfig.NAMESPACE_PRIVACY, - SafetyCenterStatus.SAFETY_CENTER_IS_ENABLED, - /* value = */ Boolean.toString(false), - /* makeDefault = */ false); + when(mSafetyCenterStatusHolder.isEnabled(any(Context.class))).thenReturn(false); assertThat(mTopLevelSecurityEntryPreferenceController.getAvailabilityStatus()) .isEqualTo(TopLevelSecurityEntryPreferenceController.AVAILABLE);