tests/tapl/com/android/launcher3/tapl/HomeAllApps.java

/*
 * Copyright (C) 2022 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.launcher3.tapl;

import static com.android.launcher3.testing.shared.TestProtocol.NORMAL_STATE_ORDINAL;

import androidx.annotation.NonNull;
import androidx.test.uiautomator.UiObject2;

import com.android.launcher3.testing.shared.TestProtocol;

public class HomeAllApps extends AllApps {

    HomeAllApps(LauncherInstrumentation launcher) {
        super(launcher);
    }

    /**
     * Swipes up or down to dismiss to Workspace.
     * @param swipeDown Swipe all apps down to dismiss, otherwise swipe up to dismiss by going home.
     *
     * @return the Workspace object.
     */
    @NonNull
    public Workspace switchToWorkspace(boolean swipeDown) {
        try (LauncherInstrumentation.Closable e = mLauncher.eventsCheck();
             LauncherInstrumentation.Closable c =
                     mLauncher.addContextLayer("want to switch from all apps to workspace")) {
            UiObject2 allAppsContainer = verifyActiveContainer();

            final int startX = allAppsContainer.getVisibleCenter().x;
            final int startY = swipeDown ? getTopVisibleIconBounds(allAppsContainer).centerY()
                    : mLauncher.getDevice().getDisplayHeight();
            final int endY =
                    swipeDown ? mLauncher.getDevice().getDisplayHeight() : getTopVisibleIconBounds(
                            allAppsContainer).centerY();
            LauncherInstrumentation.log(
                    "switchToWorkspace: startY = " + startY + ", endY = " + endY
                            + ", slop = " + mLauncher.getTouchSlop()
                            + ", swipeDown = " + swipeDown);

            mLauncher.swipeToState(
                    startX,
                    startY,
                    startX,
                    endY,
                    5 /* steps */,
                    NORMAL_STATE_ORDINAL,
                    swipeDown ? LauncherInstrumentation.GestureScope.DONT_EXPECT_PILFER
                            : LauncherInstrumentation.GestureScope.EXPECT_PILFER);

            try (LauncherInstrumentation.Closable c1 = mLauncher.addContextLayer(
                    "swiped to workspace")) {
                return mLauncher.getWorkspace();
            }
        }
    }

    @Override
    protected LauncherInstrumentation.ContainerType getContainerType() {
        return LauncherInstrumentation.ContainerType.HOME_ALL_APPS;
    }

    @NonNull
    @Override
    public HomeAppIcon getAppIcon(String appName) {
        return (AllAppsAppIcon) super.getAppIcon(appName);
    }

    @NonNull
    @Override
    protected HomeAppIcon createAppIcon(UiObject2 icon) {
        return new AllAppsAppIcon(mLauncher, icon);
    }

    @Override
    protected boolean hasSearchBox() {
        return true;
    }

    @Override
    protected int getAppsListRecyclerTopPadding() {
        return mLauncher.getTestInfo(TestProtocol.REQUEST_ALL_APPS_TOP_PADDING)
                .getInt(TestProtocol.TEST_INFO_RESPONSE_FIELD);
    }

    @NonNull
    @Override
    public Qsb getQsb() {
        return new AllAppsQsb(mLauncher, verifyActiveContainer());
    }

    @Override
    protected int getAllAppsScroll() {
        return mLauncher.getTestInfo(TestProtocol.REQUEST_APPS_LIST_SCROLL_Y)
                .getInt(TestProtocol.TEST_INFO_RESPONSE_FIELD);
    }

    @Override
    protected void verifyVisibleContainerOnDismiss() {
        mLauncher.getWorkspace();
    }

    @Override
    public boolean isHomeState() {
        return true;
    }
}