tests/tests/shortcutmanager/src/android/content/pm/cts/shortcutmanager/ShortcutManagerMaxCountTest.java

/*
 * Copyright (C) 2016 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 android.content.pm.cts.shortcutmanager;

import static com.android.server.pm.shortcutmanagertest.ShortcutManagerTestUtils.assertDynamicShortcutCountExceeded;
import static com.android.server.pm.shortcutmanagertest.ShortcutManagerTestUtils.assertWith;
import static com.android.server.pm.shortcutmanagertest.ShortcutManagerTestUtils.list;
import static com.android.server.pm.shortcutmanagertest.ShortcutManagerTestUtils.retryUntil;
import static com.android.server.pm.shortcutmanagertest.ShortcutManagerTestUtils.setDefaultLauncher;

import android.test.suitebuilder.annotation.SmallTest;

@SmallTest
public class ShortcutManagerMaxCountTest extends ShortcutManagerCtsTestsBase {
    /**
     * Basic tests: single app, single activity, no manifest shortcuts.
     */
    public void testNumDynamicShortcuts() {
        runWithCaller(mPackageContext1, () -> {
            assertTrue(getManager().setDynamicShortcuts(list(makeShortcut("s1"))));
            assertTrue(getManager().setDynamicShortcuts(list(
                    makeShortcut("s1"),
                    makeShortcut("s2"),
                    makeShortcut("s3"),
                    makeShortcut("s4"),
                    makeShortcut("s5")
            )));
            assertWith(getManager().getDynamicShortcuts())
                    .haveIds("s1", "s2", "s3", "s4", "s5")
                    .areAllDynamic()
                    .areAllEnabled();
            assertTrue(getManager().setDynamicShortcuts(list(
                    makeShortcut("s1x"),
                    makeShortcut("s2x"),
                    makeShortcut("s3x"),
                    makeShortcut("s4x"),
                    makeShortcut("s5x")
            )));

            assertDynamicShortcutCountExceeded(() -> {
                getManager().setDynamicShortcuts(list(
                        makeShortcut("s1y"),
                        makeShortcut("s2y"),
                        makeShortcut("s3y"),
                        makeShortcut("s4y"),
                        makeShortcut("s5y"),
                        makeShortcut("s6y")));
            });
            assertWith(getManager().getDynamicShortcuts())
                    .haveIds("s1x", "s2x", "s3x", "s4x", "s5x")
                    .areAllDynamic()
                    .areAllEnabled();

            assertDynamicShortcutCountExceeded(() -> {
                getManager().addDynamicShortcuts(list(
                        makeShortcut("s1y")));
            });
            assertWith(getManager().getDynamicShortcuts())
                    .haveIds("s1x", "s2x", "s3x", "s4x", "s5x")
                    .areAllDynamic()
                    .areAllEnabled();
            getManager().removeDynamicShortcuts(list("s5x"));
            assertTrue(getManager().addDynamicShortcuts(list(
                    makeShortcut("s1y"))));

            assertWith(getManager().getDynamicShortcuts())
                    .haveIds("s1x", "s2x", "s3x", "s4x", "s1y")
                    .areAllDynamic()
                    .areAllEnabled();

            getManager().removeAllDynamicShortcuts();

            assertTrue(getManager().addDynamicShortcuts(list(
                    makeShortcut("s1"),
                    makeShortcut("s2"),
                    makeShortcut("s3"),
                    makeShortcut("s4"),
                    makeShortcut("s5")
            )));
            assertWith(getManager().getDynamicShortcuts())
                    .haveIds("s1", "s2", "s3", "s4", "s5")
                    .areAllDynamic()
                    .areAllEnabled();
        });
    }

    /**
     * Manifest shortcuts are included in the count too.
     */
    public void testWithManifest() throws Exception {
        runWithCaller(mPackageContext1, () -> {
            enableManifestActivity("Launcher_manifest_1", true);
            enableManifestActivity("Launcher_manifest_2", true);

            retryUntil(() -> getManager().getManifestShortcuts().size() == 3,
                    "Manifest shortcuts didn't show up");

        });

        runWithCaller(mPackageContext1, () -> {
            assertWith(getManager().getManifestShortcuts())
                    .haveIds("ms1", "ms21", "ms22")
                    .areAllManifest()
                    .areAllEnabled()
                    .areAllNotPinned()

                    .selectByIds("ms1")
                    .forAllShortcuts(sa -> {
                        assertEquals(getActivity("Launcher_manifest_1"), sa.getActivity());
                    })

                    .revertToOriginalList()
                    .selectByIds("ms21", "ms22")
                    .forAllShortcuts(sa -> {
                        assertEquals(getActivity("Launcher_manifest_2"), sa.getActivity());
                    });

        });

        // Note since max counts is per activity, testNumDynamicShortcuts_single should just pass.
        testNumDynamicShortcuts();

        // Launcher_manifest_1 has one manifest, so can only add 4 dynamic shortcuts.
        runWithCaller(mPackageContext1, () -> {
            setTargetActivityOverride("Launcher_manifest_1");

            assertTrue(getManager().setDynamicShortcuts(list(
                    makeShortcut("s1"),
                    makeShortcut("s2"),
                    makeShortcut("s3"),
                    makeShortcut("s4")
            )));
            assertWith(getManager().getDynamicShortcuts())
                    .selectByActivity(getActivity("Launcher_manifest_1"))
                    .haveIds("s1", "s2", "s3", "s4")
                    .areAllEnabled();

            assertDynamicShortcutCountExceeded(() -> getManager().setDynamicShortcuts(list(
                    makeShortcut("s1x"),
                    makeShortcut("s2x"),
                    makeShortcut("s3x"),
                    makeShortcut("s4x"),
                    makeShortcut("s5x")
            )));
            // Not changed.
            assertWith(getManager().getDynamicShortcuts())
                    .selectByActivity(getActivity("Launcher_manifest_1"))
                    .haveIds("s1", "s2", "s3", "s4")
                    .areAllEnabled();
        });

        // Launcher_manifest_2 has two manifests, so can only add 3.
        runWithCaller(mPackageContext1, () -> {
            setTargetActivityOverride("Launcher_manifest_2");

            assertTrue(getManager().addDynamicShortcuts(list(
                    makeShortcut("s1"),
                    makeShortcut("s2"),
                    makeShortcut("s3")
            )));
            assertWith(getManager().getDynamicShortcuts())
                    .selectByActivity(getActivity("Launcher_manifest_2"))
                    .haveIds("s1", "s2", "s3")
                    .areAllEnabled();

            assertDynamicShortcutCountExceeded(() -> getManager().addDynamicShortcuts(list(
                    makeShortcut("s1x")
            )));
            // Not added.
            assertWith(getManager().getDynamicShortcuts())
                    .selectByActivity(getActivity("Launcher_manifest_2"))
                    .haveIds("s1", "s2", "s3")
                    .areAllEnabled();
        });
    }

    public void testChangeActivity() {
        runWithCaller(mPackageContext1, () -> {
            setTargetActivityOverride("Launcher");
            assertTrue(getManager().setDynamicShortcuts(list(
                    makeShortcut("s1"),
                    makeShortcut("s2"),
                    makeShortcut("s3"),
                    makeShortcut("s4"),
                    makeShortcut("s5")
            )));
            assertWith(getManager().getDynamicShortcuts())
                    .selectByActivity(getActivity("Launcher"))
                    .haveIds("s1", "s2", "s3", "s4", "s5")
                    .areAllDynamic()
                    .areAllEnabled();

            setTargetActivityOverride("Launcher2");

            assertTrue(getManager().addDynamicShortcuts(list(
                    makeShortcut("s1b"),
                    makeShortcut("s2b"),
                    makeShortcut("s3b"),
                    makeShortcut("s4b"),
                    makeShortcut("s5b")
            )));
            assertWith(getManager().getDynamicShortcuts())
                    .selectByActivity(getActivity("Launcher"))
                    .haveIds("s1", "s2", "s3", "s4", "s5")
                    .areAllDynamic()
                    .areAllEnabled()

                    .revertToOriginalList()
                    .selectByActivity(getActivity("Launcher2"))
                    .haveIds("s1b", "s2b", "s3b", "s4b", "s5b")
                    .areAllDynamic()
                    .areAllEnabled();

            // Moving one from L1 to L2 is not allowed.
            assertDynamicShortcutCountExceeded(() -> getManager().updateShortcuts(list(
                    makeShortcut("s1", getActivity("Launcher2"))
            )));

            assertWith(getManager().getDynamicShortcuts())
                    .selectByActivity(getActivity("Launcher"))
                    .haveIds("s1", "s2", "s3", "s4", "s5")
                    .areAllDynamic()
                    .areAllEnabled()

                    .revertToOriginalList()
                    .selectByActivity(getActivity("Launcher2"))
                    .haveIds("s1b", "s2b", "s3b", "s4b", "s5b")
                    .areAllDynamic()
                    .areAllEnabled();

            // But swapping shortcuts will work.
            assertTrue(getManager().updateShortcuts(list(
                    makeShortcut("s1", getActivity("Launcher2")),
                    makeShortcut("s1b", getActivity("Launcher"))
            )));

            assertWith(getManager().getDynamicShortcuts())
                    .selectByActivity(getActivity("Launcher"))
                    .haveIds("s1b", "s2", "s3", "s4", "s5")
                    .areAllDynamic()
                    .areAllEnabled()

                    .revertToOriginalList()
                    .selectByActivity(getActivity("Launcher2"))
                    .haveIds("s1", "s2b", "s3b", "s4b", "s5b")
                    .areAllDynamic()
                    .areAllEnabled();
        });
    }

    public void testWithPinned() {
        runWithCaller(mPackageContext1, () -> {
            assertTrue(getManager().setDynamicShortcuts(list(
                    makeShortcut("s1"),
                    makeShortcut("s2"),
                    makeShortcut("s3"),
                    makeShortcut("s4"),
                    makeShortcut("s5")
            )));
        });

        setDefaultLauncher(getInstrumentation(), mLauncherContext1);

        runWithCaller(mLauncherContext1, () -> {
            getLauncherApps().pinShortcuts(mPackageContext1.getPackageName(),
                    list("s1", "s2", "s3", "s4", "s5"), getUserHandle());
        });

        runWithCaller(mPackageContext1, () -> {
            assertTrue(getManager().setDynamicShortcuts(list(
                    makeShortcut("s6"),
                    makeShortcut("s7"),
                    makeShortcut("s8"),
                    makeShortcut("s9"),
                    makeShortcut("s10")
            )));

            assertWith(getManager().getDynamicShortcuts())
                    .haveIds("s6", "s7", "s8", "s9", "s10")
                    .areAllEnabled()
                    .areAllNotPinned();

            assertWith(getManager().getPinnedShortcuts())
                    .haveIds("s1", "s2", "s3", "s4", "s5")
                    .areAllEnabled()
                    .areAllNotDynamic();
        });
    }
}