Home | History | Annotate | Download | only in development 
      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.development;
     18 
     19 import static com.android.settings.development.LocalTerminalPreferenceController.TERMINAL_APP_PACKAGE;
     20 import static org.mockito.Mockito.doReturn;
     21 import static org.mockito.Mockito.never;
     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.content.Context;
     27 import android.content.pm.PackageManager;
     28 import android.os.UserManager;
     29 import android.support.v14.preference.SwitchPreference;
     30 import android.support.v7.preference.PreferenceScreen;
     31 
     32 import com.android.settings.testutils.SettingsRobolectricTestRunner;
     33 import com.android.settingslib.wrapper.PackageManagerWrapper;
     34 
     35 import org.junit.Before;
     36 import org.junit.Test;
     37 import org.junit.runner.RunWith;
     38 import org.mockito.Mock;
     39 import org.mockito.MockitoAnnotations;
     40 
     41 @RunWith(SettingsRobolectricTestRunner.class)
     42 public class LocalTerminalPreferenceControllerTest {
     43 
     44     @Mock
     45     private Context mContext;
     46     @Mock
     47     private SwitchPreference mPreference;
     48     @Mock
     49     private PreferenceScreen mPreferenceScreen;
     50     @Mock
     51     private UserManager mUserManager;
     52     @Mock
     53     private PackageManagerWrapper mPackageManagerWrapper;
     54 
     55     private LocalTerminalPreferenceController mController;
     56 
     57     @Before
     58     public void setup() {
     59         MockitoAnnotations.initMocks(this);
     60         when(mContext.getSystemService(Context.USER_SERVICE)).thenReturn(mUserManager);
     61         when(mUserManager.isAdminUser()).thenReturn(true);
     62         mController = spy(new LocalTerminalPreferenceController(mContext));
     63         doReturn(true).when(mController).isAvailable();
     64         doReturn(mPackageManagerWrapper).when(mController).getPackageManagerWrapper();
     65         when(mPreferenceScreen.findPreference(mController.getPreferenceKey()))
     66             .thenReturn(mPreference);
     67         mController.displayPreference(mPreferenceScreen);
     68     }
     69 
     70     @Test
     71     public void displayPreference_shouldDisablePreferenceWhenNotAdmin() {
     72         when(mUserManager.isAdminUser()).thenReturn(false);
     73         mController.displayPreference(mPreferenceScreen);
     74 
     75         verify(mPreference).setEnabled(false);
     76     }
     77 
     78     @Test
     79     public void onPreferenceChanged_turnOnTerminal() {
     80         mController.onPreferenceChange(null, true);
     81 
     82         verify(mPackageManagerWrapper).setApplicationEnabledSetting(TERMINAL_APP_PACKAGE,
     83                 PackageManager.COMPONENT_ENABLED_STATE_ENABLED, 0);
     84     }
     85 
     86     @Test
     87     public void onPreferenceChanged_turnOffTerminal() {
     88         mController.onPreferenceChange(null, false);
     89 
     90         verify(mPackageManagerWrapper).setApplicationEnabledSetting(TERMINAL_APP_PACKAGE,
     91                 PackageManager.COMPONENT_ENABLED_STATE_DEFAULT, 0);
     92     }
     93 
     94     @Test
     95     public void updateState_preferenceShouldBeChecked() {
     96         when(mPackageManagerWrapper.getApplicationEnabledSetting(TERMINAL_APP_PACKAGE)).thenReturn(
     97                 PackageManager.COMPONENT_ENABLED_STATE_ENABLED);
     98         mController.updateState(mPreference);
     99 
    100         verify(mPreference).setChecked(true);
    101     }
    102 
    103     @Test
    104     public void updateState_preferenceShouldNotBeChecked() {
    105         when(mPackageManagerWrapper.getApplicationEnabledSetting(TERMINAL_APP_PACKAGE)).thenReturn(
    106                 PackageManager.COMPONENT_ENABLED_STATE_DEFAULT);
    107         mController.updateState(mPreference);
    108 
    109         verify(mPreference).setChecked(false);
    110     }
    111 
    112     @Test
    113     public void onDeveloperOptionsSwitchDisabled_shouldDisablePreference() {
    114         mController.onDeveloperOptionsSwitchDisabled();
    115 
    116         verify(mPackageManagerWrapper).setApplicationEnabledSetting(TERMINAL_APP_PACKAGE,
    117                 PackageManager.COMPONENT_ENABLED_STATE_DEFAULT, 0);
    118         verify(mPreference).setChecked(false);
    119         verify(mPreference).setEnabled(false);
    120     }
    121 
    122     @Test
    123     public void onDeveloperOptionsSwitchEnabled_shouldDoNothingWhenNotAdmin() {
    124         when(mUserManager.isAdminUser()).thenReturn(false);
    125         mController.onDeveloperOptionsSwitchDisabled();
    126 
    127         verify(mPreference, never()).setEnabled(true);
    128     }
    129 
    130     @Test
    131     public void onDeveloperOptionsSwitchEnabled_preferenceShouldBeEnabledWhenAdmin() {
    132         mController.onDeveloperOptionsSwitchEnabled();
    133 
    134         verify(mPreference).setEnabled(true);
    135     }
    136 }
    137