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

import java.lang.Exception;
import java.lang.String;
import java.util.ArrayList;
import java.util.List;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import junit.framework.Assert;

import static com.android.ddmlib.Log.LogLevel.INFO;

import com.android.tradefed.device.DeviceNotAvailableException;
import com.android.tradefed.log.LogUtil.CLog;

public class ActivityManagerReplaceWindowTests extends ActivityManagerTestBase {

    private static final String SLOW_CREATE_ACTIVITY_NAME = "SlowCreateActivity";
    private static final String NO_RELAUNCH_ACTIVITY_NAME = "NoRelaunchActivity";

    private static final String AM_MOVE_TASK = "am stack movetask ";

    private List<String> mTempWindowTokens = new ArrayList();

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

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

    private void testReplaceWindow_Dock(boolean relaunch) throws Exception {
        final String activityName =
                relaunch ? SLOW_CREATE_ACTIVITY_NAME : NO_RELAUNCH_ACTIVITY_NAME;
        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 (SLOW_CREATE_ACTIVITY_NAME.equals(activityName)) {
            Thread.sleep(2000);
        }

        CLog.logAndDisplay(INFO, "==========Before Docking========");
        final String oldToken = getWindowToken(windowName, activityName, true);

        // Move to docked stack
        final int taskId = getActivityTaskId(activityName);
        final String cmd = AM_MOVE_TASK + taskId + " " + DOCKED_STACK_ID + " true";
        executeShellCommand(cmd);

        // Sleep 5 seconds, then check if the window is replaced properly.
        Thread.sleep(5000);

        CLog.logAndDisplay(INFO, "==========After Docking========");
        final String newToken = getWindowToken(windowName, activityName, false);

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

    private String getWindowToken(String windowName, String activityName, boolean visibleOnly)
            throws Exception {
        mAmWmState.computeState(mDevice, visibleOnly, new String[] {activityName});

        mAmWmState.assertVisibility(activityName, true);

        mAmWmState.getWmState().getMatchingWindowTokens(windowName, mTempWindowTokens);

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

        return mTempWindowTokens.get(0);
    }
}