Home | History | Annotate | Download | only in simstatus 
      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.simstatus;
     18 
     19 import static org.mockito.ArgumentMatchers.anyString;
     20 import static org.mockito.Mockito.doReturn;
     21 import static org.mockito.Mockito.mock;
     22 import static org.mockito.Mockito.spy;
     23 import static org.mockito.Mockito.verify;
     24 import static org.mockito.Mockito.when;
     25 
     26 import android.app.Fragment;
     27 import android.app.FragmentManager;
     28 import android.content.Context;
     29 import android.os.UserManager;
     30 import android.support.v7.preference.Preference;
     31 import android.support.v7.preference.PreferenceScreen;
     32 import android.telephony.TelephonyManager;
     33 
     34 import com.android.settings.R;
     35 import com.android.settings.testutils.SettingsRobolectricTestRunner;
     36 
     37 import org.junit.Before;
     38 import org.junit.Test;
     39 import org.junit.runner.RunWith;
     40 import org.mockito.Answers;
     41 import org.mockito.Mock;
     42 import org.mockito.MockitoAnnotations;
     43 import org.robolectric.RuntimeEnvironment;
     44 import org.robolectric.util.ReflectionHelpers;
     45 
     46 @RunWith(SettingsRobolectricTestRunner.class)
     47 public class SimStatusPreferenceControllerTest {
     48 
     49     @Mock
     50     private Preference mPreference;
     51     @Mock
     52     private Preference mSecondSimPreference;
     53     @Mock
     54     private PreferenceScreen mScreen;
     55     @Mock
     56     private TelephonyManager mTelephonyManager;
     57     @Mock
     58     private UserManager mUserManager;
     59     @Mock
     60     private Fragment mFragment;
     61 
     62     private Context mContext;
     63     private SimStatusPreferenceController mController;
     64 
     65     @Before
     66     public void setUp() {
     67         MockitoAnnotations.initMocks(this);
     68         mContext = spy(RuntimeEnvironment.application);
     69         doReturn(mUserManager).when(mContext).getSystemService(UserManager.class);
     70         mController = spy(new SimStatusPreferenceController(mContext, mFragment));
     71         doReturn(true).when(mController).isAvailable();
     72         when(mScreen.getContext()).thenReturn(mContext);
     73         doReturn(mSecondSimPreference).when(mController).createNewPreference(mContext);
     74         ReflectionHelpers.setField(mController, "mTelephonyManager", mTelephonyManager);
     75         when(mScreen.findPreference(mController.getPreferenceKey())).thenReturn(mPreference);
     76         final String prefKey = mController.getPreferenceKey();
     77         when(mPreference.getKey()).thenReturn(prefKey);
     78         when(mPreference.isVisible()).thenReturn(true);
     79     }
     80 
     81     @Test
     82     public void displayPreference_multiSim_shouldAddSecondPreference() {
     83         when(mTelephonyManager.getPhoneCount()).thenReturn(2);
     84 
     85         mController.displayPreference(mScreen);
     86 
     87         verify(mScreen).addPreference(mSecondSimPreference);
     88     }
     89 
     90     @Test
     91     public void updateState_singleSim_shouldSetSingleSimTitleAndSummary() {
     92         when(mTelephonyManager.getPhoneCount()).thenReturn(1);
     93         mController.displayPreference(mScreen);
     94 
     95         mController.updateState(mPreference);
     96 
     97         verify(mPreference).setTitle(mContext.getString(R.string.sim_status_title));
     98         verify(mPreference).setSummary(anyString());
     99     }
    100 
    101     @Test
    102     public void updateState_multiSim_shouldSetMultiSimTitleAndSummary() {
    103         when(mTelephonyManager.getPhoneCount()).thenReturn(2);
    104         mController.displayPreference(mScreen);
    105 
    106         mController.updateState(mPreference);
    107 
    108         verify(mPreference).setTitle(
    109                 mContext.getString(R.string.sim_status_title_sim_slot, 1 /* sim slot */));
    110         verify(mSecondSimPreference).setTitle(
    111                 mContext.getString(R.string.sim_status_title_sim_slot, 2 /* sim slot */));
    112         verify(mPreference).setSummary(anyString());
    113         verify(mSecondSimPreference).setSummary(anyString());
    114     }
    115 
    116     @Test
    117     public void handlePreferenceTreeClick_shouldStartDialogFragment() {
    118         when(mFragment.getChildFragmentManager()).thenReturn(
    119                 mock(FragmentManager.class, Answers.RETURNS_DEEP_STUBS));
    120         mController.displayPreference(mScreen);
    121 
    122         mController.handlePreferenceTreeClick(mPreference);
    123 
    124         verify(mFragment).getChildFragmentManager();
    125     }
    126 }
    127