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

Commit 3b58a060 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: c14a4b42

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



Change-Id: I1dc1844da1f858a0b862981dfc10c0c8d2262ba5
Signed-off-by: default avatarAutomerger Merge Worker <android-build-automerger-merge-worker@system.gserviceaccount.com>
parents 4e32f05a c14a4b42

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);