/*
 * 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.assertCallbackNotReceived;
import static com.android.server.pm.shortcutmanagertest.ShortcutManagerTestUtils.assertCallbackReceived;
import static com.android.server.pm.shortcutmanagertest.ShortcutManagerTestUtils.assertWith;
import static com.android.server.pm.shortcutmanagertest.ShortcutManagerTestUtils.checkAssertSuccess;
import static com.android.server.pm.shortcutmanagertest.ShortcutManagerTestUtils.dumpsysShortcut;
import static com.android.server.pm.shortcutmanagertest.ShortcutManagerTestUtils.list;
import static com.android.server.pm.shortcutmanagertest.ShortcutManagerTestUtils.resetAll;
import static com.android.server.pm.shortcutmanagertest.ShortcutManagerTestUtils.setDefaultLauncher;
import static com.android.server.pm.shortcutmanagertest.ShortcutManagerTestUtils.waitUntil;

import static org.mockito.Mockito.mock;

import android.content.pm.LauncherApps;
import android.content.pm.ShortcutInfo;
import android.os.Handler;
import android.os.Looper;
import android.test.suitebuilder.annotation.SmallTest;

/**
 * Make sure switching between mPackageContext1..3 and mLauncherContext1..3 will work as intended.
 */
@SmallTest
public class ShortcutManagerSpoofingTest extends ShortcutManagerCtsTestsBase {
    /**
     * Create shortcuts from different packages and make sure they're really different.
     */
    public void testSpoofingPublisher() {
        runWithCaller(mPackageContext1, () -> {
            ShortcutInfo s1 = makeShortcut("s1", "title1");
            getManager().setDynamicShortcuts(list(s1));
        });
        runWithCaller(mPackageContext2, () -> {
            ShortcutInfo s1 = makeShortcut("s1", "title2");
            getManager().setDynamicShortcuts(list(s1));
        });
        runWithCaller(mPackageContext3, () -> {
            ShortcutInfo s1 = makeShortcut("s1", "title3");
            getManager().setDynamicShortcuts(list(s1));
        });

        runWithCaller(mPackageContext1, () -> {
            assertWith(getManager().getDynamicShortcuts())
                    .haveIds("s1")
                    .forShortcutWithId("s1", s -> {
                        assertEquals("title1", s.getShortLabel());
                    });
        });
        runWithCaller(mPackageContext2, () -> {
            assertWith(getManager().getDynamicShortcuts())
                    .haveIds("s1")
                    .forShortcutWithId("s1", s -> {
                        assertEquals("title2", s.getShortLabel());
                    });
        });
        runWithCaller(mPackageContext3, () -> {
            assertWith(getManager().getDynamicShortcuts())
                    .haveIds("s1")
                    .forShortcutWithId("s1", s -> {
                        assertEquals("title3", s.getShortLabel());
                    });
        });
    }

    public void testSpoofingLauncher() {
        final LauncherApps.Callback c0_1 = mock(LauncherApps.Callback.class);
        final LauncherApps.Callback c0_2 = mock(LauncherApps.Callback.class);
        final LauncherApps.Callback c0_3 = mock(LauncherApps.Callback.class);
        final Handler h = new Handler(Looper.getMainLooper());

        runWithCaller(mLauncherContext1, () -> getLauncherApps().registerCallback(c0_1, h));
        runWithCaller(mLauncherContext2, () -> getLauncherApps().registerCallback(c0_2, h));
        runWithCaller(mLauncherContext3, () -> getLauncherApps().registerCallback(c0_3, h));

        // Change the default launcher
        setDefaultLauncher(getInstrumentation(), mLauncherContext2);

        dumpsysShortcut(getInstrumentation());

        runWithCaller(mLauncherContext1,
                () -> assertFalse(getLauncherApps().hasShortcutHostPermission()));
        runWithCaller(mLauncherContext2,
                () -> assertTrue(getLauncherApps().hasShortcutHostPermission()));
        runWithCaller(mLauncherContext3,
                () -> assertFalse(getLauncherApps().hasShortcutHostPermission()));

        // Call a publisher API and make sure only launcher2 gets it.

        resetAll(list(c0_1, c0_2, c0_3));

        runWithCaller(mPackageContext1, () -> {
            ShortcutInfo s1 = makeShortcut("s1", "title1");
            getManager().setDynamicShortcuts(list(s1));
        });

        // Because of the handlers, callback calls are not synchronous.
        waitUntil("Launcher 2 didn't receive message", () ->
                checkAssertSuccess(() ->
                        assertCallbackReceived(c0_2, android.os.Process.myUserHandle(),
                                mPackageContext1.getPackageName(), "s1")
                )
        );

        assertCallbackNotReceived(c0_1);
        assertCallbackNotReceived(c0_3);
    }
}