Merge "FingerprintEnrollmentActivity UI test" (0bebfe26) · Commits · e / os / android_packages_apps_Settings

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