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

Commit acb58ebf authored by Omer Ozer's avatar Omer Ozer
Browse files

Add factory reset wizard test coverage to 

factory reset preference controller

Bug: 305244192
Test: RunSettingsRoboTests
Change-Id: I0c0a67b41cc73589863a71e24617451f5b49de5e
parent c2251539

src/com/android/settings/system/FactoryResetPreferenceController.java

+6 −2
Original line number Diff line number Diff line
@@ -28,6 +28,7 @@ import androidx.activity.result.ActivityResultLauncher; 
import androidx.activity.result.contract.ActivityResultContracts;

import androidx.preference.Preference;



import com.android.internal.annotations.VisibleForTesting;

import com.android.settings.Settings;

import com.android.settings.core.BasePreferenceController;

import com.android.settings.factory_reset.Flags;
@@ -36,11 +37,14 @@ public class FactoryResetPreferenceController extends BasePreferenceController { 


    private static final String TAG = "FactoryResetPreference";



    private static final String ACTION_PREPARE_FACTORY_RESET =

    @VisibleForTesting

    static final String ACTION_PREPARE_FACTORY_RESET =

            "com.android.settings.ACTION_PREPARE_FACTORY_RESET";



    private final UserManager mUm;

    private ActivityResultLauncher<Intent> mFactoryResetPreparationLauncher;



    @VisibleForTesting

    ActivityResultLauncher<Intent> mFactoryResetPreparationLauncher;



    public FactoryResetPreferenceController(Context context, String preferenceKey) {

        super(context, preferenceKey);

tests/robotests/src/com/android/settings/system/FactoryResetPreferenceControllerTest.java

+82 −20
Original line number Diff line number Diff line
@@ -17,75 +17,125 @@ package com.android.settings.system; 


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



import static org.mockito.ArgumentMatchers.any;

import static org.mockito.ArgumentMatchers.anyInt;

import static org.mockito.ArgumentMatchers.anyString;

import static org.mockito.ArgumentMatchers.eq;

import static org.mockito.Mockito.verify;

import static org.mockito.Mockito.when;



import android.Manifest;

import android.content.Context;

import android.content.Intent;

import android.content.pm.ActivityInfo;

import android.content.pm.PackageInfo;

import android.content.pm.PackageManager;

import android.content.pm.ResolveInfo;

import android.content.pm.UserInfo;

import android.os.UserHandle;

import android.os.UserManager;

import android.platform.test.annotations.RequiresFlagsEnabled;

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

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

import android.provider.Settings;



import com.android.settings.testutils.shadow.ShadowUserManager;

import com.android.settings.testutils.shadow.ShadowUtils;

import androidx.activity.result.ActivityResultLauncher;

import androidx.preference.Preference;



import com.google.common.collect.ImmutableList;



import org.junit.After;

import org.junit.Before;

import org.junit.Ignore;

import org.junit.Rule;

import org.junit.Test;

import org.junit.runner.RunWith;

import org.mockito.ArgumentCaptor;

import org.mockito.Mock;

import org.mockito.Mockito;

import org.mockito.MockitoAnnotations;

import org.robolectric.RobolectricTestRunner;

import org.robolectric.RuntimeEnvironment;

import org.robolectric.annotation.Config;



@RunWith(RobolectricTestRunner.class)

@Config(shadows = ShadowUserManager.class)

public class FactoryResetPreferenceControllerTest {



    @Rule

    public final CheckFlagsRule mCheckFlagsRule = DeviceFlagsValueProvider.createCheckFlagsRule();



    private static final String FACTORY_RESET_KEY = "factory_reset";

    private static final String FACTORY_RESET_APP_PACKAGE = "com.frw_app";



    private ShadowUserManager mShadowUserManager;

    @Mock private ActivityResultLauncher<Intent> mFactoryResetLauncher;

    @Mock private Preference mPreference;

    @Mock private Context mContext;

    @Mock private PackageManager mPackageManager;

    @Mock private UserManager mUserManager;

    private ResolveInfo mFactoryResetAppResolveInfo;

    private PackageInfo mFactoryResetAppPackageInfo;



    private Context mContext;

    private FactoryResetPreferenceController mController;



    @Before

    public void setUp() {

        mContext = RuntimeEnvironment.application;

        mShadowUserManager = ShadowUserManager.getShadow();



    public void setUp() throws PackageManager.NameNotFoundException {

        MockitoAnnotations.initMocks(this);

        when(mContext.getPackageManager()).thenReturn(mPackageManager);

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

        mController = new FactoryResetPreferenceController(mContext, FACTORY_RESET_KEY);

        mFactoryResetAppResolveInfo = new ResolveInfo();

        mFactoryResetAppResolveInfo.activityInfo = new ActivityInfo();

        mFactoryResetAppResolveInfo.activityInfo.packageName = FACTORY_RESET_APP_PACKAGE;

        mFactoryResetAppPackageInfo = new PackageInfo();

        mFactoryResetAppPackageInfo.requestedPermissions =

                new String[] {Manifest.permission.PREPARE_FACTORY_RESET};

        mFactoryResetAppPackageInfo.requestedPermissionsFlags = new int[] {

                PackageInfo.REQUESTED_PERMISSION_GRANTED

        };

        when(mPackageManager.resolveActivity(any(), anyInt()))

                .thenReturn(mFactoryResetAppResolveInfo);

        when(mPackageManager.getPackageInfo(anyString(), anyInt()))

                .thenReturn(mFactoryResetAppPackageInfo);

        when(mPreference.getKey()).thenReturn(FACTORY_RESET_KEY);

        mController.mFactoryResetPreparationLauncher = mFactoryResetLauncher;



    }



    @After

    public void tearDown() {

        ShadowUtils.reset();

        mShadowUserManager.setIsAdminUser(false);

        mShadowUserManager.setIsDemoUser(false);

        Settings.Global.putInt(mContext.getContentResolver(), Settings.Global.DEVICE_DEMO_MODE, 0);

        Mockito.reset(mUserManager, mPackageManager);

        Settings.Global.putInt(RuntimeEnvironment.application.getContentResolver(),

                Settings.Global.DEVICE_DEMO_MODE, 0);

    }



    @Ignore("b/314930928")

    @Test

    public void isAvailable_systemUser() {

        mShadowUserManager.setIsAdminUser(true);

        when(mUserManager.isAdminUser()).thenReturn(true);



        assertThat(mController.isAvailable()).isTrue();

    }



    @Test

    public void isAvailable_nonSystemUser() {

        mShadowUserManager.setIsAdminUser(false);

        mShadowUserManager.setIsDemoUser(false);

        when(mUserManager.isAdminUser()).thenReturn(false);

        when(mUserManager.isDemoUser()).thenReturn(false);



        assertThat(mController.isAvailable()).isFalse();

    }



    @Test

    public void isAvailable_demoUser() {

        mShadowUserManager.setIsAdminUser(false);

        when(mUserManager.isAdminUser()).thenReturn(false);



        // Place the device in demo mode.

        Settings.Global.putInt(mContext.getContentResolver(), Settings.Global.DEVICE_DEMO_MODE, 1);

        Settings.Global.putInt(RuntimeEnvironment.application.getContentResolver(),

                Settings.Global.DEVICE_DEMO_MODE, 1);



        // Indicate the user is a demo user.

        mShadowUserManager.addUser(UserHandle.myUserId(), "test", UserInfo.FLAG_DEMO);

        when(mUserManager.getUserProfiles())

                .thenReturn(ImmutableList.of(new UserHandle(UserHandle.myUserId())));

        when(mUserManager.getUserInfo(eq(UserHandle.myUserId())))

                .thenReturn(new UserInfo(UserHandle.myUserId(), "test", UserInfo.FLAG_DEMO));



        assertThat(mController.isAvailable()).isFalse();

    }
@@ -94,4 +144,16 @@ public class FactoryResetPreferenceControllerTest { 
    public void getPreferenceKey() {

        assertThat(mController.getPreferenceKey()).isEqualTo(FACTORY_RESET_KEY);

    }



    @Test

    @RequiresFlagsEnabled(com.android.settings.factory_reset.Flags.FLAG_ENABLE_FACTORY_RESET_WIZARD)

    public void handlePreference_factoryResetWizardEnabled() {

        ArgumentCaptor<Intent> intentArgumentCaptor = ArgumentCaptor.forClass(Intent.class);



        assertThat(mController.handlePreferenceTreeClick(mPreference)).isTrue();

        verify(mFactoryResetLauncher).launch(intentArgumentCaptor.capture());

        assertThat(intentArgumentCaptor.getValue()).isNotNull();

        assertThat(intentArgumentCaptor.getValue().getAction())

                .isEqualTo(FactoryResetPreferenceController.ACTION_PREPARE_FACTORY_RESET);

    }

}