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

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

+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);