AppFilterCoordinatorTest.java | Explore in Territory

// Copyright 2024 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

package org.chromium.chrome.browser.history;

import static org.junit.Assert.assertEquals;

import static org.chromium.base.ThreadUtils.runOnUiThreadBlocking;
import static org.chromium.chrome.browser.history.AppFilterCoordinator.MAX_SHEET_HEIGHT_RATIO;
import static org.chromium.chrome.browser.history.AppFilterCoordinator.MAX_VISIBLE_ITEM_COUNT;

import android.app.Activity;
import android.graphics.Color;
import android.graphics.drawable.Drawable;
import android.view.ViewGroup;

import androidx.test.filters.MediumTest;
import androidx.test.filters.SmallTest;
import androidx.test.runner.lifecycle.Stage;

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

import org.chromium.base.ThreadUtils;
import org.chromium.base.test.BaseActivityTestRule;
import org.chromium.base.test.util.ApplicationTestUtils;
import org.chromium.base.test.util.Batch;
import org.chromium.base.test.util.CommandLineFlags;
import org.chromium.chrome.R;
import org.chromium.chrome.browser.flags.ChromeSwitches;
import org.chromium.chrome.browser.history.AppFilterCoordinator.AppInfo;
import org.chromium.chrome.test.ChromeJUnit4ClassRunner;
import org.chromium.components.browser_ui.bottomsheet.BottomSheetController;
import org.chromium.components.browser_ui.bottomsheet.BottomSheetControllerFactory;
import org.chromium.components.browser_ui.widget.scrim.ScrimCoordinator;
import org.chromium.ui.KeyboardVisibilityDelegate;
import org.chromium.ui.test.util.BlankUiTestActivity;

import java.util.ArrayList;
import java.util.List;

/** Integration tests for the app filter sheet for history page. */
@RunWith(ChromeJUnit4ClassRunner.class)
@Batch(Batch.PER_CLASS)
@CommandLineFlags.Add({ChromeSwitches.DISABLE_FIRST_RUN_EXPERIENCE})
public class AppFilterCoordinatorTest {
    private static final String APPID_YOUTUBE = "com.google.android.youtube";
    private static final String APPID_CHROME = "com.android.chrome";
    private static final String APPID_CALENDAR = "com.google.android.calendar";
    private static final String APPID_MESSAGE = "com.google.android.apps.messaging";

    private static final CharSequence APPLABEL_YOUTUBE = "YouTube";
    private static final CharSequence APPLABEL_CHROME = "Chrome";
    private static final CharSequence APPLABEL_CALENDAR = "Calendar";
    private static final CharSequence APPLABEL_MESSAGE = "Message";

    @Rule
    public final BaseActivityTestRule<BlankUiTestActivity> mActivityRule =
            new BaseActivityTestRule<>(BlankUiTestActivity.class);

    private BottomSheetController mBottomSheetController;
    private AppFilterCoordinator mAppFilterSheet;
    private AppInfo mCurrentApp;

    @Before
    public void setUp() throws InterruptedException {
        mActivityRule.launchActivity(null);
        Activity activity = getActivity();
        ApplicationTestUtils.waitForActivityState(activity, Stage.RESUMED);
        runOnUiThreadBlocking(
                () -> {
                    mBottomSheetController = createBottomSheetController();

                    Drawable icon = activity.getResources().getDrawable(R.drawable.ic_devices_16dp);
                    List<AppInfo> apps = new ArrayList<>();
                    apps.add(new AppInfo(APPID_YOUTUBE, icon, APPLABEL_YOUTUBE));
                    apps.add(new AppInfo(APPID_CHROME, icon, APPLABEL_CHROME));
                    apps.add(new AppInfo(APPID_CALENDAR, icon, APPLABEL_CALENDAR));
                    apps.add(new AppInfo(APPID_MESSAGE, icon, APPLABEL_MESSAGE));
                    mAppFilterSheet =
                            new AppFilterCoordinator(
                                    activity,
                                    activity.getWindow().getDecorView(),
                                    mBottomSheetController,
                                    this::onAppUpdated,
                                    apps);
                });
    }

    private BlankUiTestActivity getActivity() {
        return mActivityRule.getActivity();
    }

    private BottomSheetController createBottomSheetController() {
        ViewGroup activityContentView = getActivity().findViewById(android.R.id.content);
        ScrimCoordinator scrimCoordinator =
                new ScrimCoordinator(
                        getActivity(),
                        new ScrimCoordinator.SystemUiScrimDelegate() {
                            @Override
                            public void setStatusBarScrimFraction(float scrimFraction) {}

                            @Override
                            public void setNavigationBarScrimFraction(float scrimFraction) {}
                        },
                        activityContentView,
                        Color.WHITE);
        return BottomSheetControllerFactory.createBottomSheetController(
                () -> scrimCoordinator,
                (unused) -> {},
                getActivity().getWindow(),
                KeyboardVisibilityDelegate.getInstance(),
                () -> activityContentView,
                () -> 0);
    }

    private void onAppUpdated(AppInfo appInfo) {
        mCurrentApp = appInfo;
    }

    private void setCurrentAppInfo(String appId, CharSequence appLabel) {
        mCurrentApp = appId == null ? null : new AppInfo(appId, null, appLabel);
    }

    private int calcSheetHeight(int rowHeight, int baseViewHeight, int rowCount) {
        return AppFilterCoordinator.calculateSheetHeight(rowHeight, baseViewHeight, rowCount);
    }

    @Test
    @SmallTest
    public void testSheetSize() {
        final int rowHeight = 64;
        final int baseHeight = 1200;
        final int defaultMaxHeight = rowHeight * MAX_VISIBLE_ITEM_COUNT;

        int rowCount = MAX_VISIBLE_ITEM_COUNT - 1;
        assertEquals(rowHeight * rowCount, calcSheetHeight(rowHeight, baseHeight, rowCount));

        rowCount = MAX_VISIBLE_ITEM_COUNT;
        assertEquals(rowHeight * rowCount, calcSheetHeight(rowHeight, baseHeight, rowCount));

        rowCount = MAX_VISIBLE_ITEM_COUNT + 1;
        assertEquals(defaultMaxHeight, calcSheetHeight(rowHeight, baseHeight, rowCount));

        rowCount = MAX_VISIBLE_ITEM_COUNT * 2;
        assertEquals(defaultMaxHeight, calcSheetHeight(rowHeight, baseHeight, rowCount));

        final int smallBase = 300;
        final int maxHeight = (int) (smallBase * MAX_SHEET_HEIGHT_RATIO);

        rowCount = 2;
        assertEquals(rowHeight * rowCount, calcSheetHeight(rowHeight, smallBase, rowCount));

        rowCount = MAX_VISIBLE_ITEM_COUNT;
        assertEquals(maxHeight, calcSheetHeight(rowHeight, smallBase, rowCount));

        rowCount = 100;
        assertEquals(maxHeight, calcSheetHeight(rowHeight, smallBase, rowCount));
    }

    @Test
    @MediumTest
    public void testFullHistoryToApp() {
        assertEquals("Selected app is not correct.", null, mCurrentApp);

        ThreadUtils.runOnUiThreadBlocking(
                () -> {
                    mAppFilterSheet.openSheet(mCurrentApp);
                    mAppFilterSheet.clickItemForTesting(APPID_MESSAGE);
                });

        // Tapping an app selects it.
        assertEquals("Chosen app is not correct.", APPID_MESSAGE, mCurrentApp.id);
        assertEquals("Chosen label is not correct.", APPLABEL_MESSAGE, mCurrentApp.label);
    }

    @Test
    @MediumTest
    public void testSelectNewApp() {
        setCurrentAppInfo(APPID_CALENDAR, APPLABEL_CALENDAR);
        assertEquals("Selected app is not correct.", APPID_CALENDAR, mCurrentApp.id);
        ThreadUtils.runOnUiThreadBlocking(
                () -> {
                    mAppFilterSheet.openSheet(mCurrentApp);
                    mAppFilterSheet.clickItemForTesting(APPID_CHROME);
                });

        // Tapping an app makes it a newly selected one.
        assertEquals("Chosen app is not correct.", APPID_CHROME, mCurrentApp.id);
        assertEquals("Chosen label is not correct.", APPLABEL_CHROME, mCurrentApp.label);
    }

    @Test
    @MediumTest
    public void testUnselectApp() {
        setCurrentAppInfo(APPID_CALENDAR, APPLABEL_CALENDAR);
        assertEquals("Selected app is not correct.", APPID_CALENDAR, mCurrentApp.id);

        ThreadUtils.runOnUiThreadBlocking(
                () -> {
                    mAppFilterSheet.openSheet(mCurrentApp);
                    mAppFilterSheet.clickItemForTesting(APPID_CALENDAR);
                });

        // Tapping the already selected app unselects it.
        assertEquals("Chosen app is not correct.", null, mCurrentApp);

        // Open the sheet once more and select the app that was unselected right before.
        ThreadUtils.runOnUiThreadBlocking(
                () -> {
                    mAppFilterSheet.openSheet(mCurrentApp);
                    mAppFilterSheet.clickItemForTesting(APPID_CALENDAR);
                });
        assertEquals("Chosen app is not correct.", APPID_CALENDAR, mCurrentApp.id);
        assertEquals("Chosen label is not correct.", APPLABEL_CALENDAR, mCurrentApp.label);
    }

    @Test
    @MediumTest
    public void testResetSheetAtOpen() {
        assertEquals("Selected app is not correct.", null, mCurrentApp);

        ThreadUtils.runOnUiThreadBlocking(
                () -> {
                    mAppFilterSheet.openSheet(mCurrentApp);
                    mAppFilterSheet.clickItemForTesting(APPID_CALENDAR);
                });
        assertEquals("Chosen app should be Calendar.", APPID_CALENDAR, mCurrentApp.id);

        // Caller resets its state and opens the sheet again. The sheet should be reset in sync.
        setCurrentAppInfo(null, null);
        ThreadUtils.runOnUiThreadBlocking(
                () -> {
                    mAppFilterSheet.openSheet(mCurrentApp);
                });
        assertEquals(
                "No app should be selected.", null, mAppFilterSheet.getCurrentAppIdForTesting());

        setCurrentAppInfo(APPID_YOUTUBE, APPLABEL_YOUTUBE);
        ThreadUtils.runOnUiThreadBlocking(
                () -> {
                    mAppFilterSheet.openSheet(mCurrentApp);
                });
        assertEquals(
                "Chosen app should be YouTube.",
                APPID_YOUTUBE,
                mAppFilterSheet.getCurrentAppIdForTesting());
    }

    @Test
    @MediumTest
    public void testCloseSheetWithoutSelection() {
        setCurrentAppInfo(APPID_CALENDAR, APPLABEL_CALENDAR);
        assertEquals("Selected app is not correct.", APPID_CALENDAR, mCurrentApp.id);

        ThreadUtils.runOnUiThreadBlocking(
                () -> {
                    mAppFilterSheet.openSheet(mCurrentApp);
                    mAppFilterSheet.clickCloseButtonForTesting();
                });

        // Closing the sheet preserves the previously selected app.
        assertEquals("Chosen app is not correct.", APPID_CALENDAR, mCurrentApp.id);
        assertEquals("Chosen label is not correct.", APPLABEL_CALENDAR, mCurrentApp.label);
    }
}
chromium/chrome/browser/history/java/src/org/chromium/chrome/browser/history/AppFilterCoordinatorTest.java
