| /* |
| * Copyright (C) 2018 The Android Open Source Project |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| package com.android.settings; |
| |
| import static com.google.common.truth.Truth.assertThat; |
| |
| import static org.mockito.ArgumentMatchers.any; |
| import static org.mockito.Mockito.doNothing; |
| import static org.mockito.Mockito.doReturn; |
| import static org.mockito.Mockito.mock; |
| import static org.mockito.Mockito.never; |
| import static org.mockito.Mockito.spy; |
| import static org.mockito.Mockito.verify; |
| import static org.mockito.Mockito.when; |
| |
| import android.bluetooth.BluetoothAdapter; |
| import android.content.BroadcastReceiver; |
| import android.content.Context; |
| import android.content.Intent; |
| import android.content.IntentFilter; |
| import android.net.ConnectivityManager; |
| import android.net.TetheringManager; |
| import android.net.wifi.WifiManager; |
| import android.os.UserHandle; |
| import android.os.UserManager; |
| import android.util.FeatureFlagUtils; |
| |
| import androidx.fragment.app.FragmentActivity; |
| import androidx.preference.Preference; |
| import androidx.preference.SwitchPreference; |
| |
| import com.android.settings.core.FeatureFlags; |
| |
| import org.junit.Before; |
| import org.junit.Test; |
| import org.junit.runner.RunWith; |
| import org.mockito.ArgumentCaptor; |
| import org.mockito.Mock; |
| import org.mockito.MockitoAnnotations; |
| import org.robolectric.RobolectricTestRunner; |
| import org.robolectric.RuntimeEnvironment; |
| |
| import java.util.ArrayList; |
| import java.util.List; |
| |
| @RunWith(RobolectricTestRunner.class) |
| public class TetherSettingsTest { |
| |
| private Context mContext; |
| |
| @Mock |
| private ConnectivityManager mConnectivityManager; |
| @Mock |
| private UserManager mUserManager; |
| @Mock |
| private TetheringManager mTetheringManager; |
| |
| @Before |
| public void setUp() { |
| mContext = spy(RuntimeEnvironment.application); |
| |
| MockitoAnnotations.initMocks(this); |
| doReturn(mConnectivityManager) |
| .when(mContext).getSystemService(Context.CONNECTIVITY_SERVICE); |
| doReturn(mUserManager) |
| .when(mContext).getSystemService(Context.USER_SERVICE); |
| doReturn(mTetheringManager) |
| .when(mContext).getSystemService(Context.TETHERING_SERVICE); |
| |
| setupIsTetherAvailable(true); |
| |
| when(mTetheringManager.getTetherableUsbRegexs()).thenReturn(new String[0]); |
| when(mTetheringManager.getTetherableBluetoothRegexs()).thenReturn(new String[0]); |
| } |
| |
| @Test |
| public void testTetherNonIndexableKeys_tetherAvailable_keysNotReturned() { |
| FeatureFlagUtils.setEnabled(mContext, FeatureFlags.TETHER_ALL_IN_ONE, false); |
| // To let TetherUtil.isTetherAvailable return true, select one of the combinations |
| setupIsTetherAvailable(true); |
| |
| final List<String> niks = |
| TetherSettings.SEARCH_INDEX_DATA_PROVIDER.getNonIndexableKeys(mContext); |
| |
| assertThat(niks).doesNotContain(TetherSettings.KEY_TETHER_PREFS_SCREEN); |
| assertThat(niks).doesNotContain(TetherSettings.KEY_WIFI_TETHER); |
| } |
| |
| @Test |
| public void testTetherNonIndexableKeys_tetherNotAvailable_keysReturned() { |
| // To let TetherUtil.isTetherAvailable return false, select one of the combinations |
| setupIsTetherAvailable(false); |
| |
| final List<String> niks = |
| TetherSettings.SEARCH_INDEX_DATA_PROVIDER.getNonIndexableKeys(mContext); |
| |
| assertThat(niks).contains(TetherSettings.KEY_TETHER_PREFS_SCREEN); |
| assertThat(niks).contains(TetherSettings.KEY_WIFI_TETHER); |
| } |
| |
| @Test |
| public void testTetherNonIndexableKeys_usbNotAvailable_usbKeyReturned() { |
| when(mTetheringManager.getTetherableUsbRegexs()).thenReturn(new String[0]); |
| |
| final List<String> niks = |
| TetherSettings.SEARCH_INDEX_DATA_PROVIDER.getNonIndexableKeys(mContext); |
| |
| assertThat(niks).contains(TetherSettings.KEY_USB_TETHER_SETTINGS); |
| } |
| |
| @Test |
| public void testTetherNonIndexableKeys_usbAvailable_usbKeyNotReturned() { |
| FeatureFlagUtils.setEnabled(mContext, FeatureFlags.TETHER_ALL_IN_ONE, false); |
| // We can ignore the condition of Utils.isMonkeyRunning() |
| // In normal case, monkey and robotest should not execute at the same time |
| when(mTetheringManager.getTetherableUsbRegexs()).thenReturn(new String[]{"dummyRegex"}); |
| |
| final List<String> niks = |
| TetherSettings.SEARCH_INDEX_DATA_PROVIDER.getNonIndexableKeys(mContext); |
| |
| assertThat(niks).doesNotContain(TetherSettings.KEY_USB_TETHER_SETTINGS); |
| } |
| |
| @Test |
| public void testTetherNonIndexableKeys_bluetoothNotAvailable_bluetoothKeyReturned() { |
| when(mTetheringManager.getTetherableBluetoothRegexs()).thenReturn(new String[0]); |
| |
| final List<String> niks = |
| TetherSettings.SEARCH_INDEX_DATA_PROVIDER.getNonIndexableKeys(mContext); |
| |
| assertThat(niks).contains(TetherSettings.KEY_ENABLE_BLUETOOTH_TETHERING); |
| } |
| |
| @Test |
| public void testTetherNonIndexableKeys_bluetoothAvailable_bluetoothKeyNotReturned() { |
| FeatureFlagUtils.setEnabled(mContext, FeatureFlags.TETHER_ALL_IN_ONE, false); |
| when(mTetheringManager.getTetherableBluetoothRegexs()) |
| .thenReturn(new String[]{"dummyRegex"}); |
| |
| final List<String> niks = |
| TetherSettings.SEARCH_INDEX_DATA_PROVIDER.getNonIndexableKeys(mContext); |
| |
| assertThat(niks).doesNotContain(TetherSettings.KEY_ENABLE_BLUETOOTH_TETHERING); |
| } |
| |
| @Test |
| public void testSetFooterPreferenceTitle_isStaApConcurrencySupported_showStaApString() { |
| final TetherSettings spyTetherSettings = spy(new TetherSettings()); |
| when(spyTetherSettings.getContext()).thenReturn(mContext); |
| final Preference mockPreference = mock(Preference.class); |
| when(spyTetherSettings.findPreference(TetherSettings.KEY_TETHER_PREFS_FOOTER)) |
| .thenReturn(mockPreference); |
| final WifiManager mockWifiManager = mock(WifiManager.class); |
| when(mContext.getSystemService(Context.WIFI_SERVICE)).thenReturn(mockWifiManager); |
| when(mockWifiManager.isStaApConcurrencySupported()).thenReturn(true); |
| |
| spyTetherSettings.setFooterPreferenceTitle(); |
| |
| verify(mockPreference, never()).setTitle(R.string.tethering_footer_info); |
| verify(mockPreference).setTitle(R.string.tethering_footer_info_sta_ap_concurrency); |
| } |
| |
| @Test |
| public void testBluetoothState_updateBluetoothState_bluetoothTetheringStateOn() { |
| final TetherSettings spyTetherSettings = spy(new TetherSettings()); |
| when(spyTetherSettings.getContext()).thenReturn(mContext); |
| final SwitchPreference mockSwitchPreference = mock(SwitchPreference.class); |
| when(spyTetherSettings.findPreference(TetherSettings.KEY_ENABLE_BLUETOOTH_TETHERING)) |
| .thenReturn(mockSwitchPreference); |
| final FragmentActivity mockActivity = mock(FragmentActivity.class); |
| when(spyTetherSettings.getActivity()).thenReturn(mockActivity); |
| final ArgumentCaptor<BroadcastReceiver> captor = |
| ArgumentCaptor.forClass(BroadcastReceiver.class); |
| when(mockActivity.registerReceiver(captor.capture(), any(IntentFilter.class))) |
| .thenReturn(null); |
| // Bluetooth tethering state is on |
| when(spyTetherSettings.getBluetoothState()).thenReturn(BluetoothAdapter.STATE_ON); |
| when(spyTetherSettings.isBluetoothTetheringOn()).thenReturn(true); |
| |
| spyTetherSettings.setupTetherPreference(); |
| spyTetherSettings.registerReceiver(); |
| updateOnlyBluetoothState(spyTetherSettings); |
| |
| // Simulate Bluetooth tethering state changed |
| final BroadcastReceiver receiver = captor.getValue(); |
| final Intent bluetoothTetheringChanged = |
| new Intent(TetherSettings.BLUETOOTH_TETHERING_STATE_CHANGED); |
| receiver.onReceive(mockActivity, bluetoothTetheringChanged); |
| |
| verify(mockSwitchPreference).setEnabled(true); |
| verify(mockSwitchPreference).setChecked(true); |
| } |
| |
| @Test |
| public void testBluetoothState_updateBluetoothState_bluetoothTetheringStateOff() { |
| final TetherSettings spyTetherSettings = spy(new TetherSettings()); |
| when(spyTetherSettings.getContext()).thenReturn(mContext); |
| final SwitchPreference mockSwitchPreference = mock(SwitchPreference.class); |
| when(spyTetherSettings.findPreference(TetherSettings.KEY_ENABLE_BLUETOOTH_TETHERING)) |
| .thenReturn(mockSwitchPreference); |
| final FragmentActivity mockActivity = mock(FragmentActivity.class); |
| when(spyTetherSettings.getActivity()).thenReturn(mockActivity); |
| final ArgumentCaptor<BroadcastReceiver> captor = |
| ArgumentCaptor.forClass(BroadcastReceiver.class); |
| when(mockActivity.registerReceiver(captor.capture(), any(IntentFilter.class))) |
| .thenReturn(null); |
| // Bluetooth tethering state is off |
| when(spyTetherSettings.getBluetoothState()).thenReturn(BluetoothAdapter.STATE_ON); |
| when(spyTetherSettings.isBluetoothTetheringOn()).thenReturn(false); |
| |
| spyTetherSettings.setupTetherPreference(); |
| spyTetherSettings.registerReceiver(); |
| updateOnlyBluetoothState(spyTetherSettings); |
| |
| // Simulate Bluetooth tethering state changed |
| final BroadcastReceiver receiver = captor.getValue(); |
| final Intent bluetoothTetheringChanged = |
| new Intent(TetherSettings.BLUETOOTH_TETHERING_STATE_CHANGED); |
| receiver.onReceive(mockActivity, bluetoothTetheringChanged); |
| |
| verify(mockSwitchPreference).setEnabled(true); |
| verify(mockSwitchPreference).setChecked(false); |
| } |
| |
| private void updateOnlyBluetoothState(TetherSettings tetherSettings) { |
| doReturn(mTetheringManager).when(tetherSettings) |
| .getSystemService(Context.TETHERING_SERVICE); |
| when(mTetheringManager.getTetherableIfaces()).thenReturn(new String[0]); |
| when(mTetheringManager.getTetheredIfaces()).thenReturn(new String[0]); |
| when(mTetheringManager.getTetheringErroredIfaces()).thenReturn(new String[0]); |
| doNothing().when(tetherSettings).updateUsbState(any(String[].class), any(String[].class), |
| any(String[].class)); |
| doNothing().when(tetherSettings).updateEthernetState(any(String[].class), |
| any(String[].class)); |
| } |
| |
| private void setupIsTetherAvailable(boolean returnValue) { |
| when(mConnectivityManager.isTetheringSupported()).thenReturn(true); |
| |
| // For RestrictedLockUtils.checkIfRestrictionEnforced |
| final int userId = UserHandle.myUserId(); |
| List<UserManager.EnforcingUser> enforcingUsers = new ArrayList<>(); |
| when(mUserManager.getUserRestrictionSources( |
| UserManager.DISALLOW_CONFIG_TETHERING, UserHandle.of(userId))) |
| .thenReturn(enforcingUsers); |
| |
| // For RestrictedLockUtils.hasBaseUserRestriction |
| when(mUserManager.hasBaseUserRestriction( |
| UserManager.DISALLOW_CONFIG_TETHERING, UserHandle.of(userId))) |
| .thenReturn(!returnValue); |
| } |
| } |