Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit a98c9912 authored by Elis Elliott's avatar Elis Elliott Committed by Automerger Merge Worker
Browse files

Merge "Add version checks in tests that use Dynamo in DocsUi." into tm-dev am:... 

Merge "Add version checks in tests that use Dynamo in DocsUi." into tm-dev am: 5b8423e7 am: c2e5d967

Original change: https://googleplex-android-review.googlesource.com/c/platform/packages/apps/DocumentsUI/+/18619447



Change-Id: I0b15de252cc15bf48e35f6e240aba45f421213f3
Signed-off-by: default avatarAutomerger Merge Worker <android-build-automerger-merge-worker@system.gserviceaccount.com>
parents 453eb7ea c2e5d967

tests/unit/com/android/documentsui/dirlist/MessageTest.java

+23 −15
Original line number Diff line number Diff line
@@ -46,6 +46,7 @@ import com.android.documentsui.base.UserId; 
import com.android.documentsui.testing.TestActionHandler;

import com.android.documentsui.testing.TestEnv;

import com.android.documentsui.testing.UserManagers;

import com.android.modules.utils.build.SdkLevel;



import org.junit.Before;

import org.junit.Test;
@@ -60,7 +61,6 @@ public final class MessageTest { 
    };

    private UserManager mUserManager;

    private DevicePolicyManager mDevicePolicyManager;

    private DevicePolicyResourcesManager mDevicePolicyResourcesManager;

    private TestActionHandler mTestActionHandler;



    @Before
@@ -69,13 +69,11 @@ public final class MessageTest { 
        mUserManager = UserManagers.create();

        mTestActionHandler = new TestActionHandler();

        mDevicePolicyManager = mock(DevicePolicyManager.class);

        mDevicePolicyResourcesManager = mock(DevicePolicyResourcesManager.class);

        when(mContext.getSystemService(Context.USER_SERVICE)).thenReturn(mUserManager);

        when(mContext.getSystemServiceName(DevicePolicyManager.class))

                .thenReturn(Context.DEVICE_POLICY_SERVICE);

        when(mContext.getSystemService(Context.DEVICE_POLICY_SERVICE))

                .thenReturn(mDevicePolicyManager);

        when(mDevicePolicyManager.getResources()).thenReturn(mDevicePolicyResourcesManager);

        when(mContext.getResources()).thenReturn(

                InstrumentationRegistry.getInstrumentation().getTargetContext().getResources());

        DocumentsAdapter.Environment env =
@@ -91,12 +89,17 @@ public final class MessageTest { 
        Model.Update error = new Model.Update(

                new CrossProfileNoPermissionException(),

                /* isRemoteActionsEnabled= */ true);

        if (SdkLevel.isAtLeastT()) {

            String title = mContext.getString(R.string.cant_select_work_files_error_title);

            String message = mContext.getString(R.string.cant_select_work_files_error_message);

        when(mDevicePolicyResourcesManager.getString(eq(CANT_SELECT_WORK_FILES_TITLE), any()))

            DevicePolicyResourcesManager devicePolicyResourcesManager = mock(

                    DevicePolicyResourcesManager.class);

            when(mDevicePolicyManager.getResources()).thenReturn(devicePolicyResourcesManager);

            when(devicePolicyResourcesManager.getString(eq(CANT_SELECT_WORK_FILES_TITLE), any()))

                    .thenReturn(title);

        when(mDevicePolicyResourcesManager.getString(eq(CANT_SELECT_WORK_FILES_MESSAGE), any()))

            when(devicePolicyResourcesManager.getString(eq(CANT_SELECT_WORK_FILES_MESSAGE), any()))

                    .thenReturn(message);

        }



        mInflateMessage.update(error);
@@ -115,12 +118,17 @@ public final class MessageTest { 
        Model.Update error = new Model.Update(

                new CrossProfileQuietModeException(mUserId),

                /* isRemoteActionsEnabled= */ true);

        if (SdkLevel.isAtLeastT()) {

            String title = mContext.getString(R.string.quiet_mode_error_title);

            String text = mContext.getString(R.string.quiet_mode_button);

        when(mDevicePolicyResourcesManager.getString(eq(WORK_PROFILE_OFF_ERROR_TITLE), any()))

            DevicePolicyResourcesManager devicePolicyResourcesManager = mock(

                    DevicePolicyResourcesManager.class);

            when(mDevicePolicyManager.getResources()).thenReturn(devicePolicyResourcesManager);

            when(devicePolicyResourcesManager.getString(eq(WORK_PROFILE_OFF_ERROR_TITLE), any()))

                    .thenReturn(title);

        when(mDevicePolicyResourcesManager.getString(eq(WORK_PROFILE_OFF_ENABLE_BUTTON), any()))

            when(devicePolicyResourcesManager.getString(eq(WORK_PROFILE_OFF_ENABLE_BUTTON), any()))

                    .thenReturn(text);

        }



        mInflateMessage.update(error);