Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit 13d681f9 authored by Milton Wu's avatar Milton Wu
Browse files

FingerprintEnrollmentActivity UI test

Add test cases in FingerprintEnrollmentActivity UI tests for
FindSensor fragments.

Bug: 260957816
Test: atest FingerprintEnrollmentActivityTest
Change-Id: I1a256f09a1a5ebe43d6d563e62397f0dd2a1e4cb
parent d2b71ce4
Loading
Loading
Loading
Loading
+296 −29
Original line number Diff line number Diff line
@@ -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() {
        launchFingerprintEnrollActivity(true);
        Assert.assertNotNull(mDevice.wait(Until.hasObject(
                By.text("Choose your backup screen lock method")), IDLE_TIMEOUT));
    public void testLaunchChooseLock() {
        launchFingerprintEnrollActivity(false, null);
        assertThat(mDevice.wait(Until.hasObject(By.text("Choose your backup screen lock method")),
                IDLE_TIMEOUT)).isTrue();
    }

    @Test
    public void launchWithCredential() {
        LockScreenUtil.setLockscreen(LockScreenUtil.LockscreenType.PIN, "1234", true);
        launchFingerprintEnrollActivity(true);
    private void verifyIntroPage() {
        mDevice.waitForIdle();
        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(
                By.text("I agree")), IDLE_TIMEOUT));
        Assert.assertNotNull(mDevice.wait(Until.hasObject(
                By.text("No thanks")), IDLE_TIMEOUT));
        assertThat(mDevice.wait(Until.hasObject(By.text("No thanks")), IDLE_TIMEOUT)).isTrue();
        assertThat(mDevice.wait(Until.hasObject(By.text("I agree")), IDLE_TIMEOUT)).isTrue();
    }

    @Test
    public void testLaunchWithGkPwHandle_withUdfps_clickStart() {
        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();
        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);

        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();
        lottie.click();

        // Enrolling page
        assertThat(mDevice.wait(Until.hasObject(By.text("Touch & hold the fingerprint sensor")),
                IDLE_TIMEOUT)).isTrue();
    }

    @Test
    public void launchCheckPin() {
        LockScreenUtil.setLockscreen(LockScreenUtil.LockscreenType.PIN, "1234", true);
        launchFingerprintEnrollActivity(false);
        Assert.assertNotNull(mDevice.wait(Until.hasObject(
                By.text("Enter your device PIN to continue")), IDLE_TIMEOUT));
        LockScreenUtil.resetLockscreen("1234");
    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) {
            intent.putExtra(EXTRA_KEY_CHALLENGE_TOKEN, mToken);
        if (gkPwHandle != null) {
            intent.putExtra(EXTRA_KEY_GK_PW_HANDLE, gkPwHandle);
        }
        intent.setFlags(Intent.FLAG_ACTIVITY_CLEAR_TOP | Intent.FLAG_ACTIVITY_NEW_TASK);
        mContext.startActivity(intent);