Fix the broken test on NetworkProviderSettingsTest

- Mock the UserManager for restricted component

Bug: 182147117
Test: manual test
make RunSettingsRoboTests ROBOTEST_FILTER=NetworkProviderSettingsTest

Change-Id: Ica5bfc8ecd0a7a786ec6fec439998696eed04266
This commit is contained in:
Weng Su
2021-03-09 17:38:40 +08:00
parent 0b29ebd420
commit 78a5522879

View File

@@ -88,6 +88,8 @@ public class NetworkProviderSettingsTest {
@Mock @Mock
private WifiManager mWifiManager; private WifiManager mWifiManager;
@Mock @Mock
private UserManager mUserManager;
@Mock
private DataUsagePreference mDataUsagePreference; private DataUsagePreference mDataUsagePreference;
private Context mContext; private Context mContext;
private NetworkProviderSettings mNetworkProviderSettings; private NetworkProviderSettings mNetworkProviderSettings;
@@ -115,6 +117,8 @@ public class NetworkProviderSettingsTest {
doReturn(mPreferenceManager).when(mNetworkProviderSettings).getPreferenceManager(); doReturn(mPreferenceManager).when(mNetworkProviderSettings).getPreferenceManager();
doReturn(mPowerManager).when(mContext).getSystemService(PowerManager.class); doReturn(mPowerManager).when(mContext).getSystemService(PowerManager.class);
doReturn(mWifiManager).when(mContext).getSystemService(WifiManager.class); doReturn(mWifiManager).when(mContext).getSystemService(WifiManager.class);
doReturn(mUserManager).when(mContext).getSystemService(Context.USER_SERVICE);
when(mUserManager.hasBaseUserRestriction(any(), any())).thenReturn(true);
doReturn(mContext).when(mPreferenceManager).getContext(); doReturn(mContext).when(mPreferenceManager).getContext();
mNetworkProviderSettings.mAddWifiNetworkPreference = new AddWifiNetworkPreference(mContext); mNetworkProviderSettings.mAddWifiNetworkPreference = new AddWifiNetworkPreference(mContext);
mNetworkProviderSettings.mSavedNetworksPreference = new Preference(mContext); mNetworkProviderSettings.mSavedNetworksPreference = new Preference(mContext);