/*
 * 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.server.wm;

import static android.server.wm.WindowManagerState.TRANSIT_ACTIVITY_CLOSE;
import static android.server.wm.WindowManagerState.TRANSIT_ACTIVITY_OPEN;
import static android.server.wm.WindowManagerState.TRANSIT_TASK_CLOSE;
import static android.server.wm.WindowManagerState.TRANSIT_TASK_OPEN;
import static android.server.wm.WindowManagerState.TRANSIT_TRANSLUCENT_ACTIVITY_CLOSE;
import static android.server.wm.WindowManagerState.TRANSIT_WALLPAPER_CLOSE;
import static android.server.wm.WindowManagerState.TRANSIT_WALLPAPER_INTRA_CLOSE;
import static android.server.wm.WindowManagerState.TRANSIT_WALLPAPER_INTRA_OPEN;
import static android.server.wm.WindowManagerState.TRANSIT_WALLPAPER_OPEN;
import static android.server.wm.app.Components.BOTTOM_ACTIVITY;
import static android.server.wm.app.Components.BOTTOM_NON_RESIZABLE_ACTIVITY;
import static android.server.wm.app.Components.BottomActivity.EXTRA_BOTTOM_WALLPAPER;
import static android.server.wm.app.Components.BottomActivity.EXTRA_STOP_DELAY;
import static android.server.wm.app.Components.TOP_ACTIVITY;
import static android.server.wm.app.Components.TOP_NON_RESIZABLE_ACTIVITY;
import static android.server.wm.app.Components.TOP_NON_RESIZABLE_WALLPAPER_ACTIVITY;
import static android.server.wm.app.Components.TOP_WALLPAPER_ACTIVITY;
import static android.server.wm.app.Components.TRANSLUCENT_TOP_ACTIVITY;
import static android.server.wm.app.Components.TRANSLUCENT_TOP_NON_RESIZABLE_ACTIVITY;
import static android.server.wm.app.Components.TRANSLUCENT_TOP_WALLPAPER_ACTIVITY;
import static android.server.wm.app.Components.TopActivity.EXTRA_FINISH_DELAY;

import static org.junit.Assert.assertEquals;

import android.content.ComponentName;
import android.platform.test.annotations.Presubmit;

import org.junit.Test;

/**
 * This test tests the transition type selection logic in ActivityManager/WindowManager.
 * BottomActivity is started first, then TopActivity, and we check the transition type that the
 * system selects when TopActivity enters or exits under various setups.
 *
 * Note that we only require the correct transition type to be reported (eg. TRANSIT_ACTIVITY_OPEN,
 * TRANSIT_TASK_CLOSE, TRANSIT_WALLPAPER_OPEN, etc.). The exact animation is unspecified and can be
 * overridden.
 *
 * <p>Build/Install/Run:
 *     atest CtsWindowManagerDeviceTestCases:TransitionSelectionTests
 */
@Presubmit
public class TransitionSelectionTests extends ActivityManagerTestBase {

    // Test activity open/close under normal timing
    @Test
    public void testOpenActivity_NeitherWallpaper() {
        testOpenActivity(false /*bottomWallpaper*/, false /*topWallpaper*/,
                false /*slowStop*/, TRANSIT_ACTIVITY_OPEN);
    }

    @Test
    public void testCloseActivity_NeitherWallpaper() {
        testCloseActivity(false /*bottomWallpaper*/, false /*topWallpaper*/,
                false /*slowStop*/, TRANSIT_ACTIVITY_CLOSE);
    }

    @Test
    public void testOpenActivity_BottomWallpaper() {
        testOpenActivity(true /*bottomWallpaper*/, false /*topWallpaper*/,
                false /*slowStop*/, TRANSIT_WALLPAPER_CLOSE);
    }

    @Test
    public void testCloseActivity_BottomWallpaper() {
        testCloseActivity(true /*bottomWallpaper*/, false /*topWallpaper*/,
                false /*slowStop*/, TRANSIT_WALLPAPER_OPEN);
    }

    @Test
    public void testOpenActivity_BothWallpaper() {
        testOpenActivity(true /*bottomWallpaper*/, true /*topWallpaper*/,
                false /*slowStop*/, TRANSIT_WALLPAPER_INTRA_OPEN);
    }

    @Test
    public void testCloseActivity_BothWallpaper() {
        testCloseActivity(true /*bottomWallpaper*/, true /*topWallpaper*/,
                false /*slowStop*/, TRANSIT_WALLPAPER_INTRA_CLOSE);
    }

    //------------------------------------------------------------------------//

    // Test task open/close under normal timing
    @Test
    public void testOpenTask_NeitherWallpaper() {
        testOpenTask(false /*bottomWallpaper*/, false /*topWallpaper*/,
            true /* topResizable */, false /*slowStop*/, TRANSIT_TASK_OPEN);
    }

    @Test
    public void testCloseTask_NeitherWallpaper() {
        testCloseTask(false /*bottomWallpaper*/, false /*topWallpaper*/,
            true /* topResizable */, false /*slowStop*/, TRANSIT_TASK_CLOSE);
    }

    @Test
    public void testOpenTask_BottomWallpaper_TopNonResizable() {
        testOpenTask(true /*bottomWallpaper*/, false /*topWallpaper*/,
            false /* topResizable */, false /*slowStop*/, TRANSIT_WALLPAPER_CLOSE);
    }

    @Test
    public void testCloseTask_BottomWallpaper_TopNonResizable() {
        testCloseTask(true /*bottomWallpaper*/, false /*topWallpaper*/,
            false /* topResizable */, false /*slowStop*/, TRANSIT_WALLPAPER_OPEN);
    }

    @Test
    public void testOpenTask_BothWallpaper() {
        testOpenTask(true /*bottomWallpaper*/, true /*topWallpaper*/,
            false /* topResizable */, false /*slowStop*/, TRANSIT_WALLPAPER_INTRA_OPEN);
    }

    @Test
    public void testCloseTask_BothWallpaper() {
        testCloseTask(true /*bottomWallpaper*/, true /*topWallpaper*/,
            false /* topResizable */, false /*slowStop*/, TRANSIT_WALLPAPER_INTRA_CLOSE);
    }

    //------------------------------------------------------------------------//

    // Test activity close -- bottom activity slow in stopping
    // These simulate the case where the bottom activity is resumed
    // before AM receives its activitiyStopped
    @Test
    public void testCloseActivity_NeitherWallpaper_SlowStop() {
        testCloseActivity(false /*bottomWallpaper*/, false /*topWallpaper*/,
                true /*slowStop*/, TRANSIT_ACTIVITY_CLOSE);
    }

    @Test
    public void testCloseActivity_BottomWallpaper_SlowStop() {
        testCloseActivity(true /*bottomWallpaper*/, false /*topWallpaper*/,
                true /*slowStop*/, TRANSIT_WALLPAPER_OPEN);
    }

    @Test
    public void testCloseActivity_BothWallpaper_SlowStop() {
        testCloseActivity(true /*bottomWallpaper*/, true /*topWallpaper*/,
                true /*slowStop*/, TRANSIT_WALLPAPER_INTRA_CLOSE);
    }

    //------------------------------------------------------------------------//

    // Test task close -- bottom task top activity slow in stopping
    // These simulate the case where the bottom activity is resumed
    // before AM receives its activitiyStopped
    @Test
    public void testCloseTask_NeitherWallpaper_SlowStop() {
        testCloseTask(false /*bottomWallpaper*/, false /*topWallpaper*/,
            true /* topResizable */, true /*slowStop*/, TRANSIT_TASK_CLOSE);
    }

    @Test
    public void testCloseTask_BottomWallpaper_TopNonResizable_SlowStop() {
        testCloseTask(true /*bottomWallpaper*/, false /*topWallpaper*/,
            false /* topResizable */, true /*slowStop*/, TRANSIT_WALLPAPER_OPEN);
    }

    @Test
    public void testCloseTask_BothWallpaper_SlowStop() {
        testCloseTask(true /*bottomWallpaper*/, true /*topWallpaper*/,
            false /* topResizable */, true /*slowStop*/, TRANSIT_WALLPAPER_INTRA_CLOSE);
    }

    //------------------------------------------------------------------------//

    /// Test closing of translucent activity/task
    @Test
    public void testCloseActivity_NeitherWallpaper_Translucent() {
        testCloseActivityTranslucent(false /*bottomWallpaper*/, false /*topWallpaper*/,
                TRANSIT_TRANSLUCENT_ACTIVITY_CLOSE);
    }

    @Test
    public void testCloseActivity_BottomWallpaper_Translucent() {
        testCloseActivityTranslucent(true /*bottomWallpaper*/, false /*topWallpaper*/,
                TRANSIT_TRANSLUCENT_ACTIVITY_CLOSE);
    }

    @Test
    public void testCloseActivity_BothWallpaper_Translucent() {
        testCloseActivityTranslucent(true /*bottomWallpaper*/, true /*topWallpaper*/,
                TRANSIT_WALLPAPER_INTRA_CLOSE);
    }

    @Test
    public void testCloseTask_NeitherWallpaper_Translucent() {
        testCloseTaskTranslucent(false /*bottomWallpaper*/, false /*topWallpaper*/,
                TRANSIT_TRANSLUCENT_ACTIVITY_CLOSE);
    }

    @Test
    public void testCloseTask_BottomWallpaper_Translucent() {
        testCloseTaskTranslucent(true /*bottomWallpaper*/, false /*topWallpaper*/,
                TRANSIT_TRANSLUCENT_ACTIVITY_CLOSE);
    }

    @Test
    public void testCloseTask_BothWallpaper_Translucent() {
        testCloseTaskTranslucent(true /*bottomWallpaper*/, true /*topWallpaper*/,
                TRANSIT_WALLPAPER_INTRA_CLOSE);
    }

    //------------------------------------------------------------------------//

    private void testOpenActivity(boolean bottomWallpaper,
            boolean topWallpaper, boolean slowStop, String expectedTransit) {
        testTransitionSelection(true /*testOpen*/, false /*testNewTask*/,
            bottomWallpaper, topWallpaper, false /*topTranslucent*/, true /* topResizable */,
            slowStop, expectedTransit);
    }

    private void testCloseActivity(boolean bottomWallpaper,
            boolean topWallpaper, boolean slowStop, String expectedTransit) {
        testTransitionSelection(false /*testOpen*/, false /*testNewTask*/,
            bottomWallpaper, topWallpaper, false /*topTranslucent*/, true /* topResizable */,
            slowStop, expectedTransit);
    }

    private void testOpenTask(boolean bottomWallpaper,
        boolean topWallpaper, boolean topResizable, boolean slowStop, String expectedTransit) {
        testTransitionSelection(true /*testOpen*/, true /*testNewTask*/,
            bottomWallpaper, topWallpaper, false /*topTranslucent*/, topResizable, slowStop,
            expectedTransit);
    }

    private void testCloseTask(boolean bottomWallpaper,
        boolean topWallpaper, boolean topResizable, boolean slowStop, String expectedTransit) {
        testTransitionSelection(false /*testOpen*/, true /*testNewTask*/,
            bottomWallpaper, topWallpaper, false /*topTranslucent*/,
            topResizable /* topResizable */, slowStop, expectedTransit);
    }

    private void testCloseActivityTranslucent(boolean bottomWallpaper,
            boolean topWallpaper, String expectedTransit) {
        testTransitionSelection(false /*testOpen*/, false /*testNewTask*/,
            bottomWallpaper, topWallpaper, true /*topTranslucent*/, true /* topResizable */,
            false /*slowStop*/, expectedTransit);
    }

    private void testCloseTaskTranslucent(boolean bottomWallpaper,
            boolean topWallpaper, String expectedTransit) {
        testTransitionSelection(false /*testOpen*/, true /*testNewTask*/,
            bottomWallpaper, topWallpaper, true /*topTranslucent*/, true /* topResizable */,
            false /*slowStop*/, expectedTransit);
    }

    //------------------------------------------------------------------------//

    private void testTransitionSelection(
        boolean testOpen, boolean testNewTask,
        boolean bottomWallpaper, boolean topWallpaper, boolean topTranslucent,
        boolean topResizable, boolean testSlowStop, String expectedTransit) {
        final ComponentName bottomComponent = bottomWallpaper
            ? BOTTOM_NON_RESIZABLE_ACTIVITY : BOTTOM_ACTIVITY;
        String bottomStartCmd = getAmStartCmd(bottomComponent);
        if (bottomWallpaper) {
            bottomStartCmd += " --ez " + EXTRA_BOTTOM_WALLPAPER + " true";
        }
        if (testSlowStop) {
            bottomStartCmd += " --ei " + EXTRA_STOP_DELAY + " 3000";
        }
        executeShellCommand(bottomStartCmd);

        mWmState.computeState(bottomComponent);

        final ComponentName topActivity;
        if (topTranslucent && !topResizable) {
            topActivity = TRANSLUCENT_TOP_NON_RESIZABLE_ACTIVITY;
        } else if (topTranslucent && topWallpaper) {
            topActivity = TRANSLUCENT_TOP_WALLPAPER_ACTIVITY;
        } else if (topTranslucent) {
            topActivity = TRANSLUCENT_TOP_ACTIVITY;
        } else if (!topResizable && topWallpaper) {
            topActivity = TOP_NON_RESIZABLE_WALLPAPER_ACTIVITY;
        } else if (!topResizable) {
            topActivity = TOP_NON_RESIZABLE_ACTIVITY;
        } else if (topWallpaper) {
            topActivity = TOP_WALLPAPER_ACTIVITY;
        } else {
            topActivity = TOP_ACTIVITY;
        }
        String topStartCmd = getAmStartCmd(topActivity);
        if (testNewTask) {
            topStartCmd += " -f 0x18000000";
        }
        if (!testOpen) {
            topStartCmd += " --ei " + EXTRA_FINISH_DELAY + " 1000";
        }
        executeShellCommand(topStartCmd);

        Condition.waitFor("Retrieving correct transition", () -> {
            if (testOpen) {
                mWmState.computeState(topActivity);
            } else {
                mWmState.computeState(bottomComponent);
            }
            return expectedTransit.equals(
                    mWmState.getDefaultDisplayLastTransition());
        });
        assertEquals("Picked wrong transition", expectedTransit,
                mWmState.getDefaultDisplayLastTransition());
    }
}