Create CompatUIShellTestCase

import android.testing.AndroidTestingRunner

import androidx.test.filters.SmallTest

import com.android.internal.R

import com.android.wm.shell.ShellTestCase

import com.android.wm.shell.desktopmode.DesktopTestHelpers.createFreeformTask

import org.junit.Assert.assertFalse

import org.junit.Assert.assertTrue

 */

@RunWith(AndroidTestingRunner::class)

@SmallTest

class AppCompatUtilsTest : ShellTestCase() {

class AppCompatUtilsTest : CompatUIShellTestCase() {

    @Test

    fun testIsTopActivityExemptFromDesktopWindowing_onlyTransparentActivitiesInStack() {

        assertTrue(isTopActivityExemptFromDesktopWindowing(mContext,

import android.platform.test.annotations.DisableFlags;

import android.platform.test.annotations.EnableFlags;

import android.platform.test.annotations.RequiresFlagsDisabled;

import android.platform.test.flag.junit.CheckFlagsRule;

import android.platform.test.flag.junit.DeviceFlagsValueProvider;

import android.platform.test.flag.junit.SetFlagsRule;

import android.testing.AndroidTestingRunner;

import android.view.InsetsSource;

import android.view.InsetsState;

import com.android.window.flags.Flags;

import com.android.wm.shell.ShellTaskOrganizer;

import com.android.wm.shell.ShellTestCase;

import com.android.wm.shell.common.DisplayController;

import com.android.wm.shell.common.DisplayImeController;

import com.android.wm.shell.common.DisplayInsetsController;

import dagger.Lazy;

import java.util.Optional;

import org.junit.Assert;

import org.junit.Before;

import org.junit.Rule;

import org.junit.Test;

import org.junit.runner.RunWith;

import org.mockito.ArgumentCaptor;

import org.mockito.Mock;

import org.mockito.MockitoAnnotations;

import java.util.Optional;

/**

 * Tests for {@link CompatUIController}.

 *

 */

@RunWith(AndroidTestingRunner.class)

@SmallTest

public class CompatUIControllerTest extends ShellTestCase {

public class CompatUIControllerTest extends CompatUIShellTestCase {

    private static final int DISPLAY_ID = 0;

    private static final int TASK_ID = 12;

    @Rule

    public final CheckFlagsRule mCheckFlagsRule =

            DeviceFlagsValueProvider.createCheckFlagsRule();

    @Rule

    public final SetFlagsRule mSetFlagsRule = new SetFlagsRule();

    private CompatUIController mController;

    private ShellInit mShellInit;

    @Mock

import android.app.TaskInfo;

import android.graphics.Rect;

import android.platform.test.annotations.RequiresFlagsDisabled;

import android.platform.test.flag.junit.CheckFlagsRule;

import android.platform.test.flag.junit.DeviceFlagsValueProvider;

import android.testing.AndroidTestingRunner;

import android.util.Pair;

import android.view.LayoutInflater;

import com.android.window.flags.Flags;

import com.android.wm.shell.R;

import com.android.wm.shell.ShellTaskOrganizer;

import com.android.wm.shell.ShellTestCase;

import com.android.wm.shell.common.DisplayLayout;

import com.android.wm.shell.common.SyncTransactionQueue;

import com.android.wm.shell.compatui.CompatUIController.CompatUIHintsState;

import junit.framework.Assert;

import org.junit.Before;

import org.junit.Rule;

import org.junit.Test;

import org.junit.runner.RunWith;

import org.mockito.ArgumentCaptor;

 */

@RunWith(AndroidTestingRunner.class)

@SmallTest

public class CompatUILayoutTest extends ShellTestCase {

public class CompatUILayoutTest extends CompatUIShellTestCase {

    private static final int TASK_ID = 1;

    @Rule

    public final CheckFlagsRule mCheckFlagsRule =

            DeviceFlagsValueProvider.createCheckFlagsRule();

    @Mock private SyncTransactionQueue mSyncTransactionQueue;

    @Mock private Consumer<CompatUIEvent> mCallback;

    @Mock private Consumer<Pair<TaskInfo, ShellTaskOrganizer.TaskListener>> mOnRestartButtonClicked;

/*

 * Copyright (C) 2024 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 com.android.wm.shell.compatui;

import android.platform.test.flag.junit.CheckFlagsRule;

import android.platform.test.flag.junit.DeviceFlagsValueProvider;

import android.platform.test.flag.junit.SetFlagsRule;

import com.android.wm.shell.ShellTestCase;

import org.junit.Rule;

/**

 * Base class for CompatUI tests.

 */

public class CompatUIShellTestCase extends ShellTestCase {

    @Rule

    public final CheckFlagsRule mCheckFlagsRule =

            DeviceFlagsValueProvider.createCheckFlagsRule();

    @Rule

    public final SetFlagsRule mSetFlagsRule = new SetFlagsRule();

}

import androidx.test.filters.SmallTest;

import com.android.wm.shell.ShellTestCase;

import org.junit.Before;

import org.junit.Test;

import org.junit.runner.RunWith;

 */

@RunWith(AndroidTestingRunner.class)

@SmallTest

public class CompatUIStatusManagerTest extends ShellTestCase {

public class CompatUIStatusManagerTest extends CompatUIShellTestCase {

    private FakeCompatUIStatusManagerTest mTestState;

    private CompatUIStatusManager mStatusManager;