Home | History | Annotate | Download | only in imei 
      1 /*
      2  * Copyright (C) 2017 The Android Open Source Project
      3  *
      4  * Licensed under the Apache License, Version 2.0 (the "License");
      5  * you may not use this file except in compliance with the License.
      6  * You may obtain a copy of the License at
      7  *
      8  *      http://www.apache.org/licenses/LICENSE-2.0
      9  *
     10  * Unless required by applicable law or agreed to in writing, software
     11  * distributed under the License is distributed on an "AS IS" BASIS,
     12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
     13  * See the License for the specific language governing permissions and
     14  * limitations under the License.
     15  */
     16 
     17 package com.android.settings.deviceinfo.imei;
     18 
     19 import static android.telephony.TelephonyManager.PHONE_TYPE_CDMA;
     20 import static android.telephony.TelephonyManager.PHONE_TYPE_GSM;
     21 import static org.mockito.ArgumentMatchers.anyInt;
     22 import static org.mockito.Mockito.doReturn;
     23 import static org.mockito.Mockito.mock;
     24 import static org.mockito.Mockito.spy;
     25 import static org.mockito.Mockito.verify;
     26 import static org.mockito.Mockito.when;
     27 
     28 import android.app.Fragment;
     29 import android.app.FragmentManager;
     30 import android.content.Context;
     31 import android.os.UserManager;
     32 import android.support.v7.preference.Preference;
     33 import android.support.v7.preference.PreferenceScreen;
     34 import android.telephony.TelephonyManager;
     35 
     36 import com.android.settings.R;
     37 import com.android.settings.testutils.SettingsRobolectricTestRunner;
     38 
     39 import org.junit.Before;
     40 import org.junit.Test;
     41 import org.junit.runner.RunWith;
     42 import org.mockito.Answers;
     43 import org.mockito.Mock;
     44 import org.mockito.MockitoAnnotations;
     45 import org.robolectric.RuntimeEnvironment;
     46 import org.robolectric.util.ReflectionHelpers;
     47 
     48 @RunWith(SettingsRobolectricTestRunner.class)
     49 public class ImeiInfoPreferenceControllerTest {
     50 
     51     @Mock
     52     private Preference mPreference;
     53     @Mock
     54     private Preference mSecondSimPreference;
     55     @Mock
     56     private PreferenceScreen mScreen;
     57     @Mock
     58     private TelephonyManager mTelephonyManager;
     59     @Mock
     60     private UserManager mUserManager;
     61     @Mock
     62     private Fragment mFragment;
     63 
     64     private Context mContext;
     65     private ImeiInfoPreferenceController mController;
     66 
     67     @Before
     68     public void setUp() {
     69         MockitoAnnotations.initMocks(this);
     70         mContext = spy(RuntimeEnvironment.application);
     71         doReturn(mUserManager).when(mContext).getSystemService(UserManager.class);
     72         mController = spy(new ImeiInfoPreferenceController(mContext, mFragment));
     73         doReturn(true).when(mController).isAvailable();
     74         when(mScreen.getContext()).thenReturn(mContext);
     75         doReturn(mSecondSimPreference).when(mController).createNewPreference(mContext);
     76         ReflectionHelpers.setField(mController, "mTelephonyManager", mTelephonyManager);
     77         when(mScreen.findPreference(mController.getPreferenceKey())).thenReturn(mPreference);
     78         final String prefKey = mController.getPreferenceKey();
     79         when(mPreference.getKey()).thenReturn(prefKey);
     80         when(mPreference.isVisible()).thenReturn(true);
     81     }
     82 
     83     @Test
     84     public void displayPreference_multiSimGsm_shouldAddSecondPreference() {
     85         ReflectionHelpers.setField(mController, "mIsMultiSim", true);
     86         when(mTelephonyManager.getPhoneCount()).thenReturn(2);
     87         when(mTelephonyManager.getPhoneType()).thenReturn(PHONE_TYPE_GSM);
     88 
     89         mController.displayPreference(mScreen);
     90 
     91         verify(mScreen).addPreference(mSecondSimPreference);
     92     }
     93 
     94     @Test
     95     public void displayPreference_singleSimCdmaPhone_shouldSetSingleSimCdmaTitleAndMeid() {
     96         ReflectionHelpers.setField(mController, "mIsMultiSim", false);
     97         final String meid = "125132215123";
     98         when(mTelephonyManager.getPhoneType()).thenReturn(PHONE_TYPE_CDMA);
     99         doReturn(meid).when(mController).getMeid(anyInt());
    100 
    101         mController.displayPreference(mScreen);
    102 
    103         verify(mPreference).setTitle(mContext.getString(R.string.status_meid_number));
    104         verify(mPreference).setSummary(meid);
    105     }
    106 
    107     @Test
    108     public void displayPreference_multiSimCdmaPhone_shouldSetMultiSimCdmaTitleAndMeid() {
    109         ReflectionHelpers.setField(mController, "mIsMultiSim", true);
    110         final String meid = "125132215123";
    111         when(mTelephonyManager.getPhoneCount()).thenReturn(2);
    112         when(mTelephonyManager.getPhoneType()).thenReturn(PHONE_TYPE_CDMA);
    113         doReturn(meid).when(mController).getMeid(anyInt());
    114 
    115         mController.displayPreference(mScreen);
    116 
    117         verify(mPreference).setTitle(mContext.getString(R.string.meid_multi_sim, 1 /* sim slot */));
    118         verify(mSecondSimPreference).setTitle(
    119                 mContext.getString(R.string.meid_multi_sim, 2 /* sim slot */));
    120         verify(mPreference).setSummary(meid);
    121         verify(mSecondSimPreference).setSummary(meid);
    122     }
    123 
    124     @Test
    125     public void displayPreference_singleSimGsmPhone_shouldSetSingleSimGsmTitleAndImei() {
    126         ReflectionHelpers.setField(mController, "mIsMultiSim", false);
    127         final String imei = "125132215123";
    128         when(mTelephonyManager.getPhoneType()).thenReturn(PHONE_TYPE_GSM);
    129         when(mTelephonyManager.getImei(anyInt())).thenReturn(imei);
    130 
    131         mController.displayPreference(mScreen);
    132 
    133         verify(mPreference).setTitle(mContext.getString(R.string.status_imei));
    134         verify(mPreference).setSummary(imei);
    135     }
    136 
    137     @Test
    138     public void displayPreference_multiSimGsmPhone_shouldSetMultiSimGsmTitleAndImei() {
    139         ReflectionHelpers.setField(mController, "mIsMultiSim", true);
    140         final String imei = "125132215123";
    141         when(mTelephonyManager.getPhoneCount()).thenReturn(2);
    142         when(mTelephonyManager.getPhoneType()).thenReturn(PHONE_TYPE_GSM);
    143         when(mTelephonyManager.getImei(anyInt())).thenReturn(imei);
    144 
    145         mController.displayPreference(mScreen);
    146 
    147         verify(mPreference).setTitle(mContext.getString(R.string.imei_multi_sim, 1 /* sim slot */));
    148         verify(mSecondSimPreference).setTitle(
    149                 mContext.getString(R.string.imei_multi_sim, 2 /* sim slot */));
    150         verify(mPreference).setSummary(imei);
    151         verify(mSecondSimPreference).setSummary(imei);
    152     }
    153 
    154     @Test
    155     public void handlePreferenceTreeClick_shouldStartDialogFragment() {
    156         when(mFragment.getChildFragmentManager())
    157             .thenReturn(mock(FragmentManager.class, Answers.RETURNS_DEEP_STUBS));
    158         when(mPreference.getTitle()).thenReturn("SomeTitle");
    159         mController.displayPreference(mScreen);
    160 
    161         mController.handlePreferenceTreeClick(mPreference);
    162 
    163         verify(mFragment).getChildFragmentManager();
    164     }
    165 }
    166