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

Commit 4f4580e5 authored by Aleksander Morgado's avatar Aleksander Morgado
Browse files

Added unit tests for ImeiInfoPreferenceController visibility 

The new test checks whether the item is available when
config_show_sim_info=false, when telephony is not data capable and
when the user is not admin,

The default visibility status for all tests is defined in the test
setup().

Bug: 395714454
Flag: EXEMPT test only
Test: atest ImeiInfoPreferenceControllerTest
Change-Id: Ic4df12da216e5c343cf696d931ff02dc942d105f
parent a66d27c4

tests/robotests/src/com/android/settings/deviceinfo/imei/ImeiInfoPreferenceControllerTest.java

+43 −3
Original line number Diff line number Diff line
@@ -20,7 +20,7 @@ import static android.telephony.TelephonyManager.PHONE_TYPE_CDMA; 
import static android.telephony.TelephonyManager.PHONE_TYPE_GSM;

import static android.telephony.TelephonyManager.PHONE_TYPE_NONE;



import static com.android.settings.core.BasePreferenceController.AVAILABLE;

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



import static org.mockito.ArgumentMatchers.anyBoolean;

import static org.mockito.Mockito.anyInt;
@@ -43,6 +43,7 @@ import androidx.preference.PreferenceCategory; 
import androidx.preference.PreferenceScreen;



import com.android.settings.R;

import com.android.settings.core.BasePreferenceController;

import com.android.settings.deviceinfo.simstatus.SlotSimStatus;



import org.junit.Before;
@@ -90,11 +91,15 @@ public class ImeiInfoPreferenceControllerTest { 


        mResources = spy(mContext.getResources());

        when(mContext.getResources()).thenReturn(mResources);

        when(mResources.getBoolean(R.bool.config_show_sim_info)).thenReturn(true);



        mockService(Context.TELEPHONY_SERVICE, TelephonyManager.class, mTelephonyManager);

        mockService(Context.USER_SERVICE, UserManager.class, mUserManager);



        // Availability defaults

        when(mResources.getBoolean(R.bool.config_show_sim_info)).thenReturn(true);

        when(mTelephonyManager.isDataCapable()).thenReturn(true);

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



        when(mScreen.getContext()).thenReturn(mContext);

        final String categoryKey = "device_detail_category";

        when(mScreen.findPreference(categoryKey)).thenReturn(mCategory);
@@ -109,7 +114,6 @@ public class ImeiInfoPreferenceControllerTest { 
                    }

                });

        controller.init(mFragment, slotSimStatus);

        doReturn(AVAILABLE).when(controller).getAvailabilityStatus();

        doReturn(preference).when(controller).createNewPreference(mContext);



        when(mScreen.findPreference(key)).thenReturn(preference);
@@ -228,6 +232,42 @@ public class ImeiInfoPreferenceControllerTest { 
        verify(mFragment).getChildFragmentManager();

    }



    @Test

    public void getAvailabilityStatus_showSimInfo_telephonyDataCapable_userAdmindisplayed() {

        setupPhoneCount(1, PHONE_TYPE_GSM, PHONE_TYPE_NONE);



        // Use defaults

        assertThat(mController.getAvailabilityStatus()).isEqualTo(

                BasePreferenceController.AVAILABLE);

    }



    @Test

    public void getAvailabilityStatus_notShowSimInfo_telephonyDataCapable_userAdmin_notDisplayed() {

        setupPhoneCount(1, PHONE_TYPE_GSM, PHONE_TYPE_NONE);



        when(mResources.getBoolean(R.bool.config_show_sim_info)).thenReturn(false);

        assertThat(mController.getAvailabilityStatus()).isEqualTo(

                BasePreferenceController.UNSUPPORTED_ON_DEVICE);

    }



    @Test

    public void getAvailabilityStatus_showSimInfo_notTelephonyDataCapable_userAdmin_notDisplayed() {

        setupPhoneCount(1, PHONE_TYPE_GSM, PHONE_TYPE_NONE);



        when(mTelephonyManager.isDataCapable()).thenReturn(false);

        assertThat(mController.getAvailabilityStatus()).isEqualTo(

                BasePreferenceController.UNSUPPORTED_ON_DEVICE);

    }



    @Test

    public void getAvailabilityStatus_showSimInfo_telephonyDataCapable_notUserAdmin_notDisplayed() {

        setupPhoneCount(1, PHONE_TYPE_GSM, PHONE_TYPE_NONE);



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

        assertThat(mController.getAvailabilityStatus()).isEqualTo(

                BasePreferenceController.UNSUPPORTED_ON_DEVICE);

    }



    private <T> void mockService(String serviceName, Class<T> serviceClass, T service) {

        when(mContext.getSystemServiceName(serviceClass)).thenReturn(serviceName);

        when(mContext.getSystemService(serviceName)).thenReturn(service);