FingerprintEnrollmentActivity UI test

Add test cases in FingerprintEnrollmentActivity UI tests for
FindSensor fragments.

Bug: 260957816
Test: atest FingerprintEnrollmentActivityTest
Change-Id: I1a256f09a1a5ebe43d6d563e62397f0dd2a1e4cb

tests/uitests/src/com/android/settings/biometrics2/ui/view/FingerprintEnrollmentActivityTest.java

@@ -16,8 +16,18 @@
 
 package com.android.settings.biometrics2.ui.view;
 
+import static com.google.common.truth.Truth.assertThat;
+
+import static org.junit.Assume.assumeFalse;
+import static org.junit.Assume.assumeTrue;
+
 import android.content.Context;
 import android.content.Intent;
+import android.content.pm.PackageManager;
+import android.hardware.fingerprint.FingerprintManager;
+import android.hardware.fingerprint.FingerprintSensorPropertiesInternal;
+import android.hardware.fingerprint.IFingerprintAuthenticatorsRegisteredCallback;
+import android.os.UserHandle;
 import android.support.test.uiautomator.By;
 import android.support.test.uiautomator.UiDevice;
 import android.support.test.uiautomator.UiObject2;
@@ -26,94 +36,351 @@ import android.support.test.uiautomator.Until;
 import androidx.test.InstrumentationRegistry;
 import androidx.test.runner.AndroidJUnit4;
 
+import com.android.internal.widget.LockPatternChecker;
+import com.android.internal.widget.LockPatternUtils;
+import com.android.internal.widget.LockscreenCredential;
 import com.android.settings.biometrics2.utils.LockScreenUtil;
 
 import org.junit.After;
-import org.junit.Assert;
 import org.junit.Before;
 import org.junit.Test;
 import org.junit.runner.RunWith;
 
+import java.util.List;
+
 @RunWith(AndroidJUnit4.class)
 public class FingerprintEnrollmentActivityTest {
 
     private static final String SETTINGS_PACKAGE_NAME = "com.android.settings";
     private static final String ACTIVITY_CLASS_NAME =
             "com.android.settings.biometrics2.ui.view.FingerprintEnrollmentActivity";
+    public static final String EXTRA_IS_SETUP_FLOW = "isSetupFlow";
     private static final String EXTRA_FROM_SETTINGS_SUMMARY = "from_settings_summary";
     private static final String EXTRA_PAGE_TRANSITION_TYPE = "page_transition_type";
-    private static final String EXTRA_KEY_CHALLENGE_TOKEN = "hw_auth_token";
+    private static final String EXTRA_KEY_GK_PW_HANDLE = "gk_pw_handle";
+    private static final String TEST_PIN = "1234";
 
     private UiDevice mDevice;
     private byte[] mToken = new byte[]{};
     private Context mContext;
+    private boolean mFingerprintPropCallbackLaunched;
+    private boolean mCanAssumeUdfps;
+    private boolean mCanAssumeSfps;
 
     private static final int IDLE_TIMEOUT = 10000;
 
     @Before
-    public void setUp() {
+    public void setUp() throws InterruptedException {
         mDevice = UiDevice.getInstance(InstrumentationRegistry.getInstrumentation());
+
         mContext = InstrumentationRegistry.getContext();
+
+        // Stop every test if it is not a fingerprint device
+        assumeTrue(mContext.getPackageManager().hasSystemFeature(
+                PackageManager.FEATURE_FINGERPRINT));
+
+        final FingerprintManager fingerprintManager = mContext.getSystemService(
+                FingerprintManager.class);
+        mFingerprintPropCallbackLaunched = false;
+        fingerprintManager.addAuthenticatorsRegisteredCallback(
+                new IFingerprintAuthenticatorsRegisteredCallback.Stub() {
+                    @Override
+                    public void onAllAuthenticatorsRegistered(
+                            List<FingerprintSensorPropertiesInternal> list) {
+                        mFingerprintPropCallbackLaunched = true;
+
+                        assertThat(list).isNotNull();
+                        assertThat(list).isNotEmpty();
+                        final FingerprintSensorPropertiesInternal prop = list.get(0);
+                        mCanAssumeUdfps = prop.isAnyUdfpsType();
+                        mCanAssumeSfps = prop.isAnySidefpsType();
+                    }
+                });
+
+        for (long i = 0; i < IDLE_TIMEOUT && !mFingerprintPropCallbackLaunched; i += 100L) {
+            Thread.sleep(100L);
+        }
+        assertThat(mFingerprintPropCallbackLaunched).isTrue();
+
         mDevice.pressHome();
     }
 
     @Test
-    public void launchWithoutCredential() {
+    public void testLaunchChooseLock() {
-        launchFingerprintEnrollActivity(true);
+        launchFingerprintEnrollActivity(false, null);
-        Assert.assertNotNull(mDevice.wait(Until.hasObject(
+        assertThat(mDevice.wait(Until.hasObject(By.text("Choose your backup screen lock method")),
-                By.text("Choose your backup screen lock method")), IDLE_TIMEOUT));
+                IDLE_TIMEOUT)).isTrue();
     }
 
-    @Test
+    private void verifyIntroPage() {
-    public void launchWithCredential() {
+        mDevice.waitForIdle();
-        LockScreenUtil.setLockscreen(LockScreenUtil.LockscreenType.PIN, "1234", true);
-        launchFingerprintEnrollActivity(true);
         for (long i = 0; i < IDLE_TIMEOUT; i += 100L) {
-            if (mDevice.wait(Until.hasObject(By.text("More")), 50L) != null) {
+            if (mDevice.wait(Until.hasObject(By.text("More")), 50L)) {
                 break;
-            } else if (mDevice.wait(Until.hasObject(By.text("I agree")), 50L) != null) {
+            } else if (mDevice.wait(Until.hasObject(By.text("I agree")), 50L)) {
                 break;
             }
         }
 
-        //click more btn at most twice and the introduction should stay in the last page
+        // Click more btn at most twice and the introduction should stay in the last page
         UiObject2 moreBtn;
         for (int i = 0; i < 2 && (moreBtn = mDevice.findObject(By.text("More"))) != null; ++i) {
             moreBtn.click();
+            mDevice.waitForIdle();
             mDevice.wait(Until.hasObject(By.text("More")), IDLE_TIMEOUT);
         }
 
-        Assert.assertNotNull(mDevice.wait(Until.hasObject(
+        assertThat(mDevice.wait(Until.hasObject(By.text("No thanks")), IDLE_TIMEOUT)).isTrue();
-                By.text("I agree")), IDLE_TIMEOUT));
+        assertThat(mDevice.wait(Until.hasObject(By.text("I agree")), IDLE_TIMEOUT)).isTrue();
-        Assert.assertNotNull(mDevice.wait(Until.hasObject(
-                By.text("No thanks")), IDLE_TIMEOUT));
-
-        LockScreenUtil.resetLockscreen("1234");
     }
 
     @Test
-    public void launchCheckPin() {
+    public void testLaunchWithGkPwHandle_withUdfps_clickStart() {
-        LockScreenUtil.setLockscreen(LockScreenUtil.LockscreenType.PIN, "1234", true);
+        assumeTrue(mCanAssumeUdfps);
-        launchFingerprintEnrollActivity(false);
+
-        Assert.assertNotNull(mDevice.wait(Until.hasObject(
+        LockScreenUtil.setLockscreen(LockScreenUtil.LockscreenType.PIN, TEST_PIN, true);
-                By.text("Enter your device PIN to continue")), IDLE_TIMEOUT));
+
-        LockScreenUtil.resetLockscreen("1234");
+        launchFingerprintEnrollActivityWithGkPwHandle(false);
+
+        // Intro page
+        verifyIntroPage();
+        final UiObject2 agreeBtn = mDevice.findObject(By.text("I agree"));
+        assertThat(agreeBtn).isNotNull();
+        agreeBtn.click();
+
+        // FindUdfps page
+        assertThat(mDevice.wait(Until.hasObject(By.text("Do it later")), IDLE_TIMEOUT)).isTrue();
+        final UiObject2 lottie = mDevice.findObject(By.res(SETTINGS_PACKAGE_NAME,
+                "illustration_lottie"));
+        assertThat(lottie).isNotNull();
+        assertThat(lottie.isClickable()).isTrue();
+        final UiObject2 startBtn = mDevice.findObject(By.text("Start"));
+        assertThat(startBtn.isClickable()).isTrue();
+        startBtn.click();
+
+        // Enrolling page
+        assertThat(mDevice.wait(Until.hasObject(By.text("Touch & hold the fingerprint sensor")),
+                IDLE_TIMEOUT)).isTrue();
+    }
+
+    @Test
+    public void testLaunchWithGkPwHandle_withUdfps_clickLottie() {
+        assumeTrue(mCanAssumeUdfps);
+
+        LockScreenUtil.setLockscreen(LockScreenUtil.LockscreenType.PIN, TEST_PIN, true);
+
+        launchFingerprintEnrollActivityWithGkPwHandle(false);
+
+        // Intro page
+        verifyIntroPage();
+        final UiObject2 agreeBtn = mDevice.findObject(By.text("I agree"));
+        assertThat(agreeBtn).isNotNull();
+        agreeBtn.click();
+
+        // FindUdfps page
+        assertThat(mDevice.wait(Until.hasObject(By.text("Do it later")), IDLE_TIMEOUT)).isTrue();
+        final UiObject2 lottie = mDevice.findObject(By.res(SETTINGS_PACKAGE_NAME,
+                "illustration_lottie"));
+        assertThat(lottie).isNotNull();
+        assertThat(lottie.isClickable()).isTrue();
+        final UiObject2 startBtn = mDevice.findObject(By.text("Start"));
+        assertThat(startBtn.isClickable()).isTrue();
+        lottie.click();
+
+        // Enrolling page
+        assertThat(mDevice.wait(Until.hasObject(By.text("Touch & hold the fingerprint sensor")),
+                IDLE_TIMEOUT)).isTrue();
+    }
+
+    @Test
+    public void testLaunchWithGkPwHandle_withSfps() {
+        assumeTrue(mCanAssumeSfps);
+
+        LockScreenUtil.setLockscreen(LockScreenUtil.LockscreenType.PIN, TEST_PIN, true);
+
+        launchFingerprintEnrollActivityWithGkPwHandle(false);
+
+        // Intro page
+        verifyIntroPage();
+        final UiObject2 agreeBtn = mDevice.findObject(By.text("I agree"));
+        assertThat(agreeBtn).isNotNull();
+        agreeBtn.click();
+
+        // FindSfps page
+        assertThat(mDevice.wait(Until.hasObject(By.text("Do it later")), IDLE_TIMEOUT)).isTrue();
+        final UiObject2 lottie = mDevice.findObject(By.res(SETTINGS_PACKAGE_NAME,
+                "illustration_lottie"));
+        assertThat(lottie).isNotNull();
+
+        // We don't have view which can be clicked to run to next page, stop at here.
+    }
+
+    @Test
+    public void testLaunchWithGkPwHandle_withRfps() {
+        assumeFalse(mCanAssumeUdfps || mCanAssumeSfps);
+
+        LockScreenUtil.setLockscreen(LockScreenUtil.LockscreenType.PIN, TEST_PIN, true);
+
+        launchFingerprintEnrollActivityWithGkPwHandle(false);
+
+        // Intro page
+        verifyIntroPage();
+        final UiObject2 agreeBtn = mDevice.findObject(By.text("I agree"));
+        assertThat(agreeBtn).isNotNull();
+        agreeBtn.click();
+
+        // FindRfps page
+        assertThat(mDevice.wait(Until.hasObject(By.text("Do it later")), IDLE_TIMEOUT)).isTrue();
+        final UiObject2 lottie = mDevice.findObject(By.res(SETTINGS_PACKAGE_NAME,
+                "illustration_lottie"));
+        if (lottie == null) {
+            // FindSfps page shall have an animation view if no lottie view
+            assertThat(mDevice.findObject(By.res(SETTINGS_PACKAGE_NAME,
+                    "fingerprint_sensor_location_animation"))).isNotNull();
+        }
+    }
+
+    @Test
+    public void testLaunchWithGkPwHandle_clickNoThanksInIntroPage() {
+        LockScreenUtil.setLockscreen(LockScreenUtil.LockscreenType.PIN, TEST_PIN, true);
+
+        launchFingerprintEnrollActivityWithGkPwHandle(false);
+
+        // Intro page
+        verifyIntroPage();
+        final UiObject2 noThanksBtn = mDevice.findObject(By.text("No thanks"));
+        assertThat(noThanksBtn).isNotNull();
+        noThanksBtn.click();
+
+        // Back to home
+        mDevice.waitForWindowUpdate("com.android.settings", IDLE_TIMEOUT);
+        assertThat(mDevice.findObject(By.text("No thanks"))).isNull();
+    }
+
+    @Test
+    public void testLaunchWithGkPwHandle_clickSkipInFindSensor() {
+        LockScreenUtil.setLockscreen(LockScreenUtil.LockscreenType.PIN, TEST_PIN, true);
+
+        launchFingerprintEnrollActivityWithGkPwHandle(false);
+
+        // Intro page
+        verifyIntroPage();
+        final UiObject2 agreeBtn = mDevice.findObject(By.text("I agree"));
+        assertThat(agreeBtn).isNotNull();
+        agreeBtn.click();
+
+        // FindSensor page
+        assertThat(mDevice.wait(Until.hasObject(By.text("Do it later")), IDLE_TIMEOUT)).isTrue();
+        final UiObject2 doItLaterBtn = mDevice.findObject(By.text("Do it later"));
+        assertThat(doItLaterBtn).isNotNull();
+        assertThat(doItLaterBtn.isClickable()).isTrue();
+        doItLaterBtn.click();
+
+        // Back to home
+        mDevice.waitForWindowUpdate("com.android.settings", IDLE_TIMEOUT);
+        assertThat(mDevice.findObject(By.text("Do it later"))).isNull();
+    }
+
+    @Test
+    public void testLaunchWithGkPwHandle_clickSkipAnywayInFindFpsDialog_whenIsSuw() {
+        LockScreenUtil.setLockscreen(LockScreenUtil.LockscreenType.PIN, TEST_PIN, true);
+
+        launchFingerprintEnrollActivityWithGkPwHandle(true);
+
+        // Intro page
+        verifyIntroPage();
+        final UiObject2 agreeBtn = mDevice.findObject(By.text("I agree"));
+        assertThat(agreeBtn).isNotNull();
+        agreeBtn.click();
+
+        // FindSensor page
+        assertThat(mDevice.wait(Until.hasObject(By.text("Do it later")), IDLE_TIMEOUT)).isTrue();
+        final UiObject2 doItLaterBtn = mDevice.findObject(By.text("Do it later"));
+        assertThat(doItLaterBtn).isNotNull();
+        assertThat(doItLaterBtn.isClickable()).isTrue();
+        doItLaterBtn.click();
+
+        // SkipSetupFindFpsDialog
+        assertThat(mDevice.wait(Until.hasObject(By.text("Skip fingerprint?")),
+                IDLE_TIMEOUT)).isTrue();
+        final UiObject2 skipAnywayBtn = mDevice.findObject(By.text("Skip anyway"));
+        assertThat(skipAnywayBtn).isNotNull();
+        assertThat(skipAnywayBtn.isClickable()).isTrue();
+        skipAnywayBtn.click();
+
+        // Back to home
+        mDevice.waitForWindowUpdate("com.android.settings", IDLE_TIMEOUT);
+        assertThat(mDevice.findObject(By.text("Skip anyway"))).isNull();
+        assertThat(mDevice.findObject(By.text("Do it later"))).isNull();
+    }
+
+    @Test
+    public void testLaunchWithGkPwHandle_clickGoBackInFindFpsDialog_whenIsSuw() {
+        LockScreenUtil.setLockscreen(LockScreenUtil.LockscreenType.PIN, TEST_PIN, true);
+
+        launchFingerprintEnrollActivityWithGkPwHandle(true);
+
+        // Intro page
+        verifyIntroPage();
+        final UiObject2 agreeBtn = mDevice.findObject(By.text("I agree"));
+        assertThat(agreeBtn).isNotNull();
+        agreeBtn.click();
+
+        // FindSensor page
+        assertThat(mDevice.wait(Until.hasObject(By.text("Do it later")), IDLE_TIMEOUT)).isTrue();
+        final UiObject2 doItLaterBtn = mDevice.findObject(By.text("Do it later"));
+        assertThat(doItLaterBtn).isNotNull();
+        assertThat(doItLaterBtn.isClickable()).isTrue();
+        doItLaterBtn.click();
+
+        // SkipSetupFindFpsDialog
+        assertThat(mDevice.wait(Until.hasObject(By.text("Skip fingerprint?")), IDLE_TIMEOUT))
+                .isTrue();
+        final UiObject2 goBackBtn = mDevice.findObject(By.text("Go back"));
+        assertThat(goBackBtn).isNotNull();
+        assertThat(goBackBtn.isClickable()).isTrue();
+        goBackBtn.click();
+
+        // FindSensor page again
+        assertThat(mDevice.wait(Until.hasObject(By.text("Do it later")), IDLE_TIMEOUT)).isTrue();
+    }
+
+    @Test
+    public void testLaunchCheckPin() {
+        LockScreenUtil.setLockscreen(LockScreenUtil.LockscreenType.PIN, TEST_PIN, true);
+        launchFingerprintEnrollActivity(false, null);
+        assertThat(mDevice.wait(Until.hasObject(By.text("Enter your device PIN to continue")),
+                IDLE_TIMEOUT)).isTrue();
     }
 
     @After
     public void tearDown() throws Exception {
-        LockScreenUtil.resetLockscreen("1234");
+        LockScreenUtil.resetLockscreen(TEST_PIN);
         mDevice.pressHome();
     }
 
-    private void launchFingerprintEnrollActivity(boolean hasToken) {
+    private void launchFingerprintEnrollActivityWithGkPwHandle(boolean isSuw) {
+        LockPatternUtils lockPatternUtils = new LockPatternUtils(mContext);
+        final LockscreenCredential lockscreenCredential = LockscreenCredential.createPin(TEST_PIN);
+        final int userId = UserHandle.myUserId();
+        final LockPatternChecker.OnVerifyCallback onVerifyCallback = (response, timeoutMs) -> {
+            launchFingerprintEnrollActivity(isSuw, response.getGatekeeperPasswordHandle());
+        };
+        LockPatternChecker.verifyCredential(lockPatternUtils, lockscreenCredential,
+                userId, LockPatternUtils.VERIFY_FLAG_REQUEST_GK_PW_HANDLE, onVerifyCallback);
+    }
+
+    private void launchFingerprintEnrollActivity(boolean isSuw, Long gkPwHandle) {
         Intent intent = new Intent();
         intent.setClassName(SETTINGS_PACKAGE_NAME, ACTIVITY_CLASS_NAME);
+        if (isSuw) {
+            intent.putExtra(EXTRA_IS_SETUP_FLOW, true);
+        }
         intent.putExtra(EXTRA_FROM_SETTINGS_SUMMARY, true);
         intent.putExtra(EXTRA_PAGE_TRANSITION_TYPE, 1);
         intent.putExtra(Intent.EXTRA_USER_ID, mContext.getUserId());
-        if (hasToken) {
+        if (gkPwHandle != null) {
-            intent.putExtra(EXTRA_KEY_CHALLENGE_TOKEN, mToken);
+            intent.putExtra(EXTRA_KEY_GK_PW_HANDLE, gkPwHandle);
         }
         intent.setFlags(Intent.FLAG_ACTIVITY_CLEAR_TOP | Intent.FLAG_ACTIVITY_NEW_TASK);
         mContext.startActivity(intent);