/*
 * Copyright (C) 2022 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.other;

import static android.app.ActivityManager.LOCK_TASK_MODE_LOCKED;
import static android.app.ActivityManager.LOCK_TASK_MODE_NONE;
import static android.app.ActivityManager.LOCK_TASK_MODE_PINNED;
import static android.content.Intent.FLAG_ACTIVITY_NEW_TASK;
import static android.server.wm.app.Components.TEST_ACTIVITY;

import static com.google.common.truth.Truth.assertThat;

import android.app.Activity;
import android.app.ActivityOptions;
import android.content.ComponentName;
import android.content.Intent;
import android.platform.test.annotations.Presubmit;
import android.server.wm.ActivityManagerTestBase;
import android.server.wm.SystemLockTaskModeSession;
import android.server.wm.app.Components;

import com.android.compatibility.common.util.ApiTest;

import org.junit.After;
import org.junit.Test;

/** Build/Install/Run: atest CtsWindowManagerDeviceOther:LockTaskModeTests */
@Presubmit
public class LockTaskModeTests extends ActivityManagerTestBase {
    private static final String[] LOCK_TASK_PACKAGES_ALLOWLIST = {Components.getPackageName()};

    @After
    public void tearDown() {
        Components.forceStopPackage();
    }

    @Test
    @ApiTest(apis = {"android.app.ActivityTaskManager#updateLockTaskPackages",
            "android.app.ActivityTaskManager#startSystemLockTaskMode"})
    public void
            startAllowedPackageIntoLockTaskMode_anotherAppPinned_exitsPinningEntersLockTaskMode() {
        clearLockTaskPackages();
        assertThat(mAm.getLockTaskModeState()).isEqualTo(LOCK_TASK_MODE_NONE);

        try (TestActivitySession<TestActivity> session = createManagedTestActivitySession()) {
            // pin app
            session.setFinishAfterClose(true);
            session.launchTestActivityOnDisplaySync(TestActivity.class, getMainDisplayId());
            final ComponentName activityComponent = session.getActivity().getComponentName();
            final int taskId = mWmState.getRootTaskByActivity(activityComponent).getTaskId();

            try (SystemLockTaskModeSession ignored = new SystemLockTaskModeSession(taskId)) {
                waitForOrFail(
                        "Task should in app pinning mode",
                        () -> mAm.getLockTaskModeState() == LOCK_TASK_MODE_PINNED);

                // setup lock task mode allowlist
                runWithShellPermission(
                        () -> mAtm.updateLockTaskPackages(mContext, LOCK_TASK_PACKAGES_ALLOWLIST));

                // start allowed package into lock task mode
                mContext.startActivity(
                        new Intent().addFlags(FLAG_ACTIVITY_NEW_TASK).setComponent(TEST_ACTIVITY),
                        ActivityOptions.makeBasic().setLockTaskEnabled(true).toBundle());
                waitAndAssertResumedActivity(TEST_ACTIVITY, "Activity must be started and resumed");
                waitForOrFail(
                        "Task should in lock task mode",
                        () -> mAm.getLockTaskModeState() == LOCK_TASK_MODE_LOCKED);
            }
        } finally {
            // cleanup
            clearLockTaskPackages();
            waitForOrFail(
                    "Task should not in app pinning mode",
                    () -> mAm.getLockTaskModeState() == LOCK_TASK_MODE_NONE);
        }
    }

    private void clearLockTaskPackages() {
        // clear lock task
        runWithShellPermission(() -> mAtm.updateLockTaskPackages(mContext, new String[0]));
    }

    /* An empty Activity which will be used in app pinning. */
    public static final class TestActivity extends Activity {}
}