quickstep/tests/src/com/android/quickstep/TaplTestsQuickstep.java - platform/packages/apps/Launcher3 - Git at Google

 /*
  * Copyright (C) 2018 The Android Open Source Project
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
  * You may obtain a copy of the License at
  *
  *      http://www.apache.org/licenses/LICENSE-2.0
  *
  * Unless required by applicable law or agreed to in writing, software
  * distributed under the License is distributed on an "AS IS" BASIS,
  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */

 package com.android.quickstep;

 import static com.android.launcher3.ui.TaplTestsLauncher3.getAppPackageName;

 import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.assertNotNull;
 import static org.junit.Assert.assertTrue;

 import android.content.Intent;
 import android.util.Log;

 import androidx.test.filters.LargeTest;
 import androidx.test.runner.AndroidJUnit4;
 import androidx.test.uiautomator.By;
 import androidx.test.uiautomator.Until;

 import com.android.launcher3.Launcher;
 import com.android.launcher3.LauncherState;
 import com.android.launcher3.tapl.Background;
 import com.android.launcher3.tapl.LauncherInstrumentation.NavigationModel;
 import com.android.launcher3.tapl.Overview;
 import com.android.launcher3.tapl.OverviewActions;
 import com.android.launcher3.tapl.OverviewTask;
 import com.android.launcher3.testing.TestProtocol;
 import com.android.launcher3.ui.TaplTestsLauncher3;
 import com.android.quickstep.NavigationModeSwitchRule.NavigationModeSwitch;
 import com.android.quickstep.views.RecentsView;

 import org.junit.After;
 import org.junit.Before;
 import org.junit.Test;
 import org.junit.runner.RunWith;

 @LargeTest
 @RunWith(AndroidJUnit4.class)
 public class TaplTestsQuickstep extends AbstractQuickStepTest {
     @Before
     public void setUp() throws Exception {
         super.setUp();
         TaplTestsLauncher3.initialize(this);
         executeOnLauncher(launcher -> {
             RecentsView recentsView = launcher.getOverviewPanel();
             recentsView.getPagedViewOrientedState().forceAllowRotationForTesting(true);
         });
     }

     @After
     public void tearDown() {
         executeOnLauncher(launcher -> {
             RecentsView recentsView = launcher.getOverviewPanel();
             recentsView.getPagedViewOrientedState().forceAllowRotationForTesting(false);
         });
     }

     public static void startTestApps() throws Exception {
         startAppFast(getAppPackageName());
         startAppFast(resolveSystemApp(Intent.CATEGORY_APP_CALCULATOR));
         startTestActivity(2);
     }

     private void startTestAppsWithCheck() throws Exception {
         startTestApps();
         executeOnLauncher(launcher -> assertTrue(
                 "Launcher activity is the top activity; expecting another activity to be the top "
                         + "one",
                 isInBackground(launcher)));
     }

     @Test
     @NavigationModeSwitch
     @PortraitLandscape
     public void testWorkspaceSwitchToAllApps() {
         assertNotNull("switchToAllApps() returned null",
                 mLauncher.getWorkspace().switchToAllApps());
         assertTrue("Launcher internal state is not All Apps",
                 isInState(() -> LauncherState.ALL_APPS));
     }

     @Test
     @PortraitLandscape
     public void testOverview() throws Exception {
         startTestAppsWithCheck();
         // mLauncher.pressHome() also tests an important case of pressing home while in background.
         Overview overview = mLauncher.pressHome().switchToOverview();
         assertTrue("Launcher internal state didn't switch to Overview",
                 isInState(() -> LauncherState.OVERVIEW));
         executeOnLauncher(
                 launcher -> assertTrue("Don't have at least 3 tasks", getTaskCount(launcher) >= 3));

         // Test flinging forward and backward.
         executeOnLauncher(launcher -> assertEquals("Current task in Overview is not 0",
                 0, getCurrentOverviewPage(launcher)));

         overview.flingForward();
         assertTrue("Launcher internal state is not Overview",
                 isInState(() -> LauncherState.OVERVIEW));
         final Integer currentTaskAfterFlingForward = getFromLauncher(
                 launcher -> getCurrentOverviewPage(launcher));
         executeOnLauncher(launcher -> assertTrue("Current task in Overview is still 0",
                 currentTaskAfterFlingForward > 0));

         overview.flingBackward();
         assertTrue("Launcher internal state is not Overview",
                 isInState(() -> LauncherState.OVERVIEW));
         executeOnLauncher(launcher -> assertTrue("Flinging back in Overview did nothing",
                 getCurrentOverviewPage(launcher) < currentTaskAfterFlingForward));

         // Test opening a task.
         OverviewTask task = mLauncher.pressHome().switchToOverview().getCurrentTask();
         assertNotNull("overview.getCurrentTask() returned null (1)", task);
         assertNotNull("OverviewTask.open returned null", task.open());
         assertTrue("Test activity didn't open from Overview", mDevice.wait(Until.hasObject(
                 By.pkg(getAppPackageName()).text("TestActivity2")),
                 DEFAULT_UI_TIMEOUT));
         executeOnLauncher(launcher -> assertTrue(
                 "Launcher activity is the top activity; expecting another activity to be the top "
                         + "one",
                 isInBackground(launcher)));

         // Test dismissing a task.
         overview = mLauncher.pressHome().switchToOverview();
         assertTrue("Launcher internal state didn't switch to Overview",
                 isInState(() -> LauncherState.OVERVIEW));
         final Integer numTasks = getFromLauncher(launcher -> getTaskCount(launcher));
         task = overview.getCurrentTask();
         assertNotNull("overview.getCurrentTask() returned null (2)", task);
         task.dismiss();
         executeOnLauncher(
                 launcher -> assertEquals("Dismissing a task didn't remove 1 task from Overview",
                         numTasks - 1, getTaskCount(launcher)));

         // Test UIDevice.pressHome, once we are in AllApps.
         mDevice.pressHome();
         waitForState("Launcher internal state didn't switch to Home", () -> LauncherState.NORMAL);

         // Test dismissing all tasks.
         mLauncher.getWorkspace().switchToOverview().dismissAllTasks();
         waitForState("Launcher internal state didn't switch to Home", () -> LauncherState.NORMAL);
         executeOnLauncher(
                 launcher -> assertEquals("Still have tasks after dismissing all",
                         0, getTaskCount(launcher)));
     }

     /**
      * Smoke test for action buttons: Presses all the buttons and makes sure no crashes occur.
      */
     @Test
     @NavigationModeSwitch
     @PortraitLandscape
     public void testOverviewActions() throws Exception {
         // Experimenting for b/165029151:
         final Overview overview = mLauncher.pressHome().switchToOverview();
         if (overview.hasTasks()) overview.dismissAllTasks();
         mLauncher.pressHome();
         //

         startTestAppsWithCheck();
         OverviewActions actionsView =
                 mLauncher.pressHome().switchToOverview().getOverviewActions();
         actionsView.clickAndDismissScreenshot();
         actionsView.clickAndDismissShare();
     }

     private int getCurrentOverviewPage(Launcher launcher) {
         return launcher.<RecentsView>getOverviewPanel().getCurrentPage();
     }

     private int getTaskCount(Launcher launcher) {
         return launcher.<RecentsView>getOverviewPanel().getTaskViewCount();
     }

     @Test
     @NavigationModeSwitch
     @PortraitLandscape
     public void testSwitchToOverview() throws Exception {
         assertNotNull("Workspace.switchToOverview() returned null",
                 mLauncher.pressHome().switchToOverview());
         assertTrue("Launcher internal state didn't switch to Overview",
                 isInState(() -> LauncherState.OVERVIEW));
     }

     @Test
     @NavigationModeSwitch
     @PortraitLandscape
     public void testBackground() throws Exception {
         startAppFast(resolveSystemApp(Intent.CATEGORY_APP_CALCULATOR));
         final Background background = getAndAssertBackground();

         assertNotNull("Background.switchToOverview() returned null", background.switchToOverview());
         assertTrue("Launcher internal state didn't switch to Overview",
                 isInState(() -> LauncherState.OVERVIEW));
     }

     private Background getAndAssertBackground() {
         final Background background = mLauncher.getBackground();
         assertNotNull("Launcher.getBackground() returned null", background);
         executeOnLauncher(launcher -> assertTrue(
                 "Launcher activity is the top activity; expecting another activity to be the top "
                         + "one",
                 isInBackground(launcher)));
         return background;
     }

     @Test
     @PortraitLandscape
     public void testAllAppsFromHome() throws Exception {
         // Test opening all apps
         assertNotNull("switchToAllApps() returned null",
                 mLauncher.getWorkspace().switchToAllApps());

         TaplTestsLauncher3.runAllAppsTest(this, mLauncher.getAllApps());

         // Testing pressHome.
         assertTrue("Launcher internal state is not All Apps",
                 isInState(() -> LauncherState.ALL_APPS));
         assertNotNull("pressHome returned null", mLauncher.pressHome());
         assertTrue("Launcher internal state is not Home",
                 isInState(() -> LauncherState.NORMAL));
         assertNotNull("getHome returned null", mLauncher.getWorkspace());
     }

     @Test
     @NavigationModeSwitch
     @PortraitLandscape
     public void testQuickSwitchFromApp() throws Exception {
         startTestActivity(2);
         startTestActivity(3);
         startTestActivity(4);

         Background background = getAndAssertBackground();
         background.quickSwitchToPreviousApp();
         assertTrue("The first app we should have quick switched to is not running",
                 isTestActivityRunning(3));

         background = getAndAssertBackground();
         background.quickSwitchToPreviousApp();
         if (mLauncher.getNavigationModel() == NavigationModel.THREE_BUTTON) {
             // 3-button mode toggles between 2 apps, rather than going back further.
             assertTrue("Second quick switch should have returned to the first app.",
                     isTestActivityRunning(4));
         } else {
             assertTrue("The second app we should have quick switched to is not running",
                     isTestActivityRunning(2));
         }
         background = getAndAssertBackground();
         background.quickSwitchToPreviousAppSwipeLeft();
         assertTrue("The 2nd app we should have quick switched to is not running",
                 isTestActivityRunning(3));
         getAndAssertBackground();
     }

     private boolean isTestActivityRunning(int activityNumber) {
         return mDevice.wait(Until.hasObject(By.pkg(getAppPackageName())
                         .text("TestActivity" + activityNumber)),
                 DEFAULT_UI_TIMEOUT);
     }

     @Test
     @NavigationModeSwitch
     @PortraitLandscape
     public void testQuickSwitchFromHome() throws Exception {
         startTestActivity(2);
         mLauncher.pressHome().quickSwitchToPreviousApp();
         assertTrue("The most recent task is not running after quick switching from home",
                 isTestActivityRunning(2));
         getAndAssertBackground();
     }
 }
	/*
	* Copyright (C) 2018 The Android Open Source Project
	*
	* Licensed under the Apache License, Version 2.0 (the "License");
	* you may not use this file except in compliance with the License.
	* You may obtain a copy of the License at
	*
	* http://www.apache.org/licenses/LICENSE-2.0
	*
	* Unless required by applicable law or agreed to in writing, software
	* distributed under the License is distributed on an "AS IS" BASIS,
	* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
	* See the License for the specific language governing permissions and
	* limitations under the License.
	*/

	package com.android.quickstep;

	import static com.android.launcher3.ui.TaplTestsLauncher3.getAppPackageName;

	import static org.junit.Assert.assertEquals;
	import static org.junit.Assert.assertNotNull;
	import static org.junit.Assert.assertTrue;

	import android.content.Intent;
	import android.util.Log;

	import androidx.test.filters.LargeTest;
	import androidx.test.runner.AndroidJUnit4;
	import androidx.test.uiautomator.By;
	import androidx.test.uiautomator.Until;

	import com.android.launcher3.Launcher;
	import com.android.launcher3.LauncherState;
	import com.android.launcher3.tapl.Background;
	import com.android.launcher3.tapl.LauncherInstrumentation.NavigationModel;
	import com.android.launcher3.tapl.Overview;
	import com.android.launcher3.tapl.OverviewActions;
	import com.android.launcher3.tapl.OverviewTask;
	import com.android.launcher3.testing.TestProtocol;
	import com.android.launcher3.ui.TaplTestsLauncher3;
	import com.android.quickstep.NavigationModeSwitchRule.NavigationModeSwitch;
	import com.android.quickstep.views.RecentsView;

	import org.junit.After;
	import org.junit.Before;
	import org.junit.Test;
	import org.junit.runner.RunWith;

	@LargeTest
	@RunWith(AndroidJUnit4.class)
	public class TaplTestsQuickstep extends AbstractQuickStepTest {
	@Before
	public void setUp() throws Exception {
	super.setUp();
	TaplTestsLauncher3.initialize(this);
	executeOnLauncher(launcher -> {
	RecentsView recentsView = launcher.getOverviewPanel();
	recentsView.getPagedViewOrientedState().forceAllowRotationForTesting(true);
	});
	}

	@After
	public void tearDown() {
	executeOnLauncher(launcher -> {
	RecentsView recentsView = launcher.getOverviewPanel();
	recentsView.getPagedViewOrientedState().forceAllowRotationForTesting(false);
	});
	}

	public static void startTestApps() throws Exception {
	startAppFast(getAppPackageName());
	startAppFast(resolveSystemApp(Intent.CATEGORY_APP_CALCULATOR));
	startTestActivity(2);
	}

	private void startTestAppsWithCheck() throws Exception {
	startTestApps();
	executeOnLauncher(launcher -> assertTrue(
	"Launcher activity is the top activity; expecting another activity to be the top "
	+ "one",
	isInBackground(launcher)));
	}

	@Test
	@NavigationModeSwitch
	@PortraitLandscape
	public void testWorkspaceSwitchToAllApps() {
	assertNotNull("switchToAllApps() returned null",
	mLauncher.getWorkspace().switchToAllApps());
	assertTrue("Launcher internal state is not All Apps",
	isInState(() -> LauncherState.ALL_APPS));
	}

	@Test
	@PortraitLandscape
	public void testOverview() throws Exception {
	startTestAppsWithCheck();
	// mLauncher.pressHome() also tests an important case of pressing home while in background.
	Overview overview = mLauncher.pressHome().switchToOverview();
	assertTrue("Launcher internal state didn't switch to Overview",
	isInState(() -> LauncherState.OVERVIEW));
	executeOnLauncher(
	launcher -> assertTrue("Don't have at least 3 tasks", getTaskCount(launcher) >= 3));

	// Test flinging forward and backward.
	executeOnLauncher(launcher -> assertEquals("Current task in Overview is not 0",
	0, getCurrentOverviewPage(launcher)));

	overview.flingForward();
	assertTrue("Launcher internal state is not Overview",
	isInState(() -> LauncherState.OVERVIEW));
	final Integer currentTaskAfterFlingForward = getFromLauncher(
	launcher -> getCurrentOverviewPage(launcher));
	executeOnLauncher(launcher -> assertTrue("Current task in Overview is still 0",
	currentTaskAfterFlingForward > 0));

	overview.flingBackward();
	assertTrue("Launcher internal state is not Overview",
	isInState(() -> LauncherState.OVERVIEW));
	executeOnLauncher(launcher -> assertTrue("Flinging back in Overview did nothing",
	getCurrentOverviewPage(launcher) < currentTaskAfterFlingForward));

	// Test opening a task.
	OverviewTask task = mLauncher.pressHome().switchToOverview().getCurrentTask();
	assertNotNull("overview.getCurrentTask() returned null (1)", task);
	assertNotNull("OverviewTask.open returned null", task.open());
	assertTrue("Test activity didn't open from Overview", mDevice.wait(Until.hasObject(
	By.pkg(getAppPackageName()).text("TestActivity2")),
	DEFAULT_UI_TIMEOUT));
	executeOnLauncher(launcher -> assertTrue(
	"Launcher activity is the top activity; expecting another activity to be the top "
	+ "one",
	isInBackground(launcher)));

	// Test dismissing a task.
	overview = mLauncher.pressHome().switchToOverview();
	assertTrue("Launcher internal state didn't switch to Overview",
	isInState(() -> LauncherState.OVERVIEW));
	final Integer numTasks = getFromLauncher(launcher -> getTaskCount(launcher));
	task = overview.getCurrentTask();
	assertNotNull("overview.getCurrentTask() returned null (2)", task);
	task.dismiss();
	executeOnLauncher(
	launcher -> assertEquals("Dismissing a task didn't remove 1 task from Overview",
	numTasks - 1, getTaskCount(launcher)));

	// Test UIDevice.pressHome, once we are in AllApps.
	mDevice.pressHome();
	waitForState("Launcher internal state didn't switch to Home", () -> LauncherState.NORMAL);

	// Test dismissing all tasks.
	mLauncher.getWorkspace().switchToOverview().dismissAllTasks();
	waitForState("Launcher internal state didn't switch to Home", () -> LauncherState.NORMAL);
	executeOnLauncher(
	launcher -> assertEquals("Still have tasks after dismissing all",
	0, getTaskCount(launcher)));
	}

	/**
	* Smoke test for action buttons: Presses all the buttons and makes sure no crashes occur.
	*/
	@Test
	@NavigationModeSwitch
	@PortraitLandscape
	public void testOverviewActions() throws Exception {
	// Experimenting for b/165029151:
	final Overview overview = mLauncher.pressHome().switchToOverview();
	if (overview.hasTasks()) overview.dismissAllTasks();
	mLauncher.pressHome();
	//

	startTestAppsWithCheck();
	OverviewActions actionsView =
	mLauncher.pressHome().switchToOverview().getOverviewActions();
	actionsView.clickAndDismissScreenshot();
	actionsView.clickAndDismissShare();
	}

	private int getCurrentOverviewPage(Launcher launcher) {
	return launcher.<RecentsView>getOverviewPanel().getCurrentPage();
	}

	private int getTaskCount(Launcher launcher) {
	return launcher.<RecentsView>getOverviewPanel().getTaskViewCount();
	}

	@Test
	@NavigationModeSwitch
	@PortraitLandscape
	public void testSwitchToOverview() throws Exception {
	assertNotNull("Workspace.switchToOverview() returned null",
	mLauncher.pressHome().switchToOverview());
	assertTrue("Launcher internal state didn't switch to Overview",
	isInState(() -> LauncherState.OVERVIEW));
	}

	@Test
	@NavigationModeSwitch
	@PortraitLandscape
	public void testBackground() throws Exception {
	startAppFast(resolveSystemApp(Intent.CATEGORY_APP_CALCULATOR));
	final Background background = getAndAssertBackground();

	assertNotNull("Background.switchToOverview() returned null", background.switchToOverview());
	assertTrue("Launcher internal state didn't switch to Overview",
	isInState(() -> LauncherState.OVERVIEW));
	}

	private Background getAndAssertBackground() {
	final Background background = mLauncher.getBackground();
	assertNotNull("Launcher.getBackground() returned null", background);
	executeOnLauncher(launcher -> assertTrue(
	"Launcher activity is the top activity; expecting another activity to be the top "
	+ "one",
	isInBackground(launcher)));
	return background;
	}

	@Test
	@PortraitLandscape
	public void testAllAppsFromHome() throws Exception {
	// Test opening all apps
	assertNotNull("switchToAllApps() returned null",
	mLauncher.getWorkspace().switchToAllApps());

	TaplTestsLauncher3.runAllAppsTest(this, mLauncher.getAllApps());

	// Testing pressHome.
	assertTrue("Launcher internal state is not All Apps",
	isInState(() -> LauncherState.ALL_APPS));
	assertNotNull("pressHome returned null", mLauncher.pressHome());
	assertTrue("Launcher internal state is not Home",
	isInState(() -> LauncherState.NORMAL));
	assertNotNull("getHome returned null", mLauncher.getWorkspace());
	}

	@Test
	@NavigationModeSwitch
	@PortraitLandscape
	public void testQuickSwitchFromApp() throws Exception {
	startTestActivity(2);
	startTestActivity(3);
	startTestActivity(4);

	Background background = getAndAssertBackground();
	background.quickSwitchToPreviousApp();
	assertTrue("The first app we should have quick switched to is not running",
	isTestActivityRunning(3));

	background = getAndAssertBackground();
	background.quickSwitchToPreviousApp();
	if (mLauncher.getNavigationModel() == NavigationModel.THREE_BUTTON) {
	// 3-button mode toggles between 2 apps, rather than going back further.
	assertTrue("Second quick switch should have returned to the first app.",
	isTestActivityRunning(4));
	} else {
	assertTrue("The second app we should have quick switched to is not running",
	isTestActivityRunning(2));
	}
	background = getAndAssertBackground();
	background.quickSwitchToPreviousAppSwipeLeft();
	assertTrue("The 2nd app we should have quick switched to is not running",
	isTestActivityRunning(3));
	getAndAssertBackground();
	}

	private boolean isTestActivityRunning(int activityNumber) {
	return mDevice.wait(Until.hasObject(By.pkg(getAppPackageName())
	.text("TestActivity" + activityNumber)),
	DEFAULT_UI_TIMEOUT);
	}

	@Test
	@NavigationModeSwitch
	@PortraitLandscape
	public void testQuickSwitchFromHome() throws Exception {
	startTestActivity(2);
	mLauncher.pressHome().quickSwitchToPreviousApp();
	assertTrue("The most recent task is not running after quick switching from home",
	isTestActivityRunning(2));
	getAndAssertBackground();
	}
	}