Merge "Update footer text whenever the ScreenPinningSettings is shown" into main

src/com/android/settings/security/ScreenPinningSettings.java

@@ -240,9 +240,9 @@ public class ScreenPinningSettings extends SettingsPreferenceFragment
             mUseScreenLock.setChecked(isScreenLockUsed());
             mUseScreenLock.setTitle(getCurrentSecurityTitle(mLockPatternUtils));
         } else {
-            mFooterPreference.setSummary(getAppPinningContent());
             mUseScreenLock.setEnabled(false);
         }
+        mFooterPreference.setSummary(getAppPinningContent());
     }
 
     private boolean isGuestModeSupported() {

tests/robotests/src/com/android/settings/security/ScreenPinningSettingsTest.java

@@ -18,34 +18,48 @@ package com.android.settings.security;
 
 import static com.google.common.truth.Truth.assertThat;
 
+import static org.robolectric.Shadows.shadowOf;
+
 import android.app.admin.DevicePolicyManager;
 import android.content.Context;
+import android.content.Intent;
+import android.icu.text.MessageFormat;
 import android.os.UserHandle;
+import android.os.UserManager;
+import android.provider.Settings;
 
+import androidx.fragment.app.Fragment;
 import androidx.test.core.app.ApplicationProvider;
 
 import com.android.settings.R;
+import com.android.settings.SettingsActivity;
 import com.android.settings.testutils.shadow.ShadowLockPatternUtils;
 import com.android.settingslib.search.SearchIndexableRaw;
+import com.android.settingslib.widget.FooterPreference;
 
 import org.junit.After;
 import org.junit.Before;
 import org.junit.Test;
 import org.junit.runner.RunWith;
 import org.robolectric.RobolectricTestRunner;
+import org.robolectric.android.controller.ActivityController;
 import org.robolectric.annotation.Config;
+import org.robolectric.shadows.ShadowUserManager;
 
 import java.util.List;
+import java.util.function.Consumer;
 
 @RunWith(RobolectricTestRunner.class)
 @Config(shadows = ShadowLockPatternUtils.class)
 public class ScreenPinningSettingsTest {
-
+    private static final String KEY_FOOTER = "screen_pinning_settings_screen_footer";
     private Context mContext;
+    private UserManager mUserManager;
 
     @Before
     public void setUp() {
         mContext = ApplicationProvider.getApplicationContext();
+        mUserManager = mContext.getSystemService(UserManager.class);
     }
 
     @After
@@ -110,4 +124,97 @@ public class ScreenPinningSettingsTest {
         assertThat(indexRaws.get(0).title).isEqualTo(
                 mContext.getString(R.string.screen_pinning_unlock_none));
     }
+
+    @Test
+    public void onCreate_lockToAppEnabled_guestModeSupported_verifyFooterText() {
+        setupLockToAppState(/* enabled= */ true);
+        setupGuestModeState(/* supported= */ true);
+
+        launchFragmentAndRunTest(fragment -> {
+            FooterPreference footer = fragment.findPreference(KEY_FOOTER);
+
+            assertThat(footer.getSummary())
+                    .isEqualTo(getExpectedFooterText(/* guestModeSupported= */ true));
+        });
+    }
+
+    @Test
+    public void onCreate_lockToAppEnabled_guestModeNotSupported_verifyFooterText() {
+        setupLockToAppState(/* enabled= */ true);
+        setupGuestModeState(/* supported= */ false);
+
+        launchFragmentAndRunTest(fragment -> {
+            FooterPreference footer = fragment.findPreference(KEY_FOOTER);
+
+            assertThat(footer.getSummary())
+                    .isEqualTo(getExpectedFooterText(/* guestModeSupported= */ false));
+        });
+    }
+
+    @Test
+    public void onCreate_lockToAppDisabled_guestModeSupported_verifyFooterText() {
+        setupLockToAppState(/* enabled= */ false);
+        setupGuestModeState(/* supported= */ true);
+
+        launchFragmentAndRunTest(fragment -> {
+            FooterPreference footer = fragment.findPreference(KEY_FOOTER);
+
+            assertThat(footer.getSummary())
+                    .isEqualTo(getExpectedFooterText(/* guestModeSupported= */ true));
+        });
+    }
+
+    @Test
+    public void onCreate_lockToAppDisabled_guestModeNotSupported_verifyFooterText() {
+        setupLockToAppState(/* enabled= */ false);
+        setupGuestModeState(/* supported= */ false);
+
+        launchFragmentAndRunTest(fragment -> {
+            FooterPreference footer = fragment.findPreference(KEY_FOOTER);
+
+            assertThat(footer.getSummary())
+                    .isEqualTo(getExpectedFooterText(/* guestModeSupported= */ false));
+        });
+    }
+
+    private CharSequence getExpectedFooterText(boolean guestModeSupported) {
+        final int stringResource = guestModeSupported
+                ? R.string.screen_pinning_guest_user_description
+                : R.string.screen_pinning_description;
+        return MessageFormat.format(mContext.getString(stringResource), 1, 2, 3);
+    }
+
+    private void setupLockToAppState(boolean enabled) {
+        Settings.System.putInt(mContext.getContentResolver(), Settings.System.LOCK_TO_APP_ENABLED,
+                enabled ? 1 : 0);
+    }
+
+    private void setupGuestModeState(boolean supported) {
+        ShadowUserManager shadowUserManager = shadowOf(mUserManager);
+        shadowUserManager.setSupportsMultipleUsers(supported);
+        shadowUserManager.setUserRestriction(
+                UserHandle.of(UserHandle.myUserId()), UserManager.DISALLOW_USER_SWITCH, !supported);
+    }
+
+    private void launchFragmentAndRunTest(Consumer<ScreenPinningSettings> test) {
+        Intent intent = new Intent();
+        intent.putExtra(SettingsActivity.EXTRA_SHOW_FRAGMENT,
+                SecurityAdvancedSettings.class.getName());
+        // ScreenPinningSettings is tightly coupled with the SettingsActivity
+        // In order to successfully launch the ScreenPinningSettings, have to use an indirect route
+        // to launch the SecurityAdvancedSetting first, then replace it with ScreenPinningSettings.
+        try (ActivityController<SettingsActivity> controller =
+                     ActivityController.of(new SettingsActivity(), intent)) {
+            controller.create().start().resume();
+
+            controller.get().getSupportFragmentManager().beginTransaction().replace(
+                    R.id.main_content, ScreenPinningSettings.class, null).commitNow();
+            Fragment fragment = controller.get().getSupportFragmentManager()
+                    .findFragmentById(R.id.main_content);
+            assertThat(fragment).isNotNull();
+            assertThat(fragment).isInstanceOf(ScreenPinningSettings.class);
+
+            test.accept((ScreenPinningSettings) fragment);
+        }
+    }
 }