/*
 * 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.ComponentNameUtils.getWindowName;
import static android.server.wm.StateLogger.log;
import static android.server.wm.app.Components.NO_RELAUNCH_ACTIVITY;
import static android.server.wm.app.Components.SLOW_CREATE_ACTIVITY;

import static org.junit.Assert.assertEquals;
import static org.junit.Assume.assumeTrue;

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

import org.junit.Before;
import org.junit.Test;

import java.util.List;
import java.util.concurrent.TimeUnit;

/**
 * Build/Install/Run:
 *     atest CtsWindowManagerDeviceTestCases:ReplaceWindowTests
 */
@Presubmit
public class ReplaceWindowTests extends ActivityManagerTestBase {

    @Before
    @Override
    public void setUp() throws Exception {
        super.setUp();

        assumeTrue("Skipping test: no multi-window support", supportsSplitScreenMultiWindow());
    }

    @Test
    public void testReplaceWindow_Dock_Relaunch() throws Exception {
        testReplaceWindow_Dock(true);
    }

    @Test
    public void testReplaceWindow_Dock_NoRelaunch() throws Exception {
        testReplaceWindow_Dock(false);
    }

    private void testReplaceWindow_Dock(boolean relaunch) throws Exception {
        final ComponentName activityName = relaunch ? SLOW_CREATE_ACTIVITY : NO_RELAUNCH_ACTIVITY;
        final String windowName = getWindowName(activityName);
        final String amStartCmd = getAmStartCmd(activityName);

        executeShellCommand(amStartCmd);

        // Sleep 2 seconds, then check if the window is started properly. SlowCreateActivity
        // will do a sleep inside its onCreate() to simulate a slow-starting app. So instead of
        // relying on WindowManagerState's retrying mechanism, we do an explicit sleep to avoid
        // excess spews from WindowManagerState.
        if (relaunch) {
            SystemClock.sleep(TimeUnit.SECONDS.toMillis(2));
        }

        log("==========Before Docking========");
        final String oldToken = getWindowToken(windowName, activityName);

        // Move to docked stack
        putActivityInPrimarySplit(activityName);

        // Sleep 5 seconds, then check if the window is replaced properly.
        SystemClock.sleep(TimeUnit.SECONDS.toMillis(5));

        log("==========After Docking========");
        final String newToken = getWindowToken(windowName, activityName);

        // For both relaunch and not relaunch case, we'd like the window to be kept.
        assertEquals("Window replaced while docking.", oldToken, newToken);
    }

    private String getWindowToken(String windowName, ComponentName activityName) throws Exception {
        mWmState.computeState(activityName);

        mWmState.assertVisibility(activityName, true);

        final List<String> windowTokens =
                mWmState.getMatchingWindowTokens(windowName);

        assertEquals("Should have exactly one window for the activity.",
                1, windowTokens.size());

        return windowTokens.get(0);
    }
}