Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit 78a55228 authored by Weng Su's avatar Weng Su
Browse files

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
parent 0b29ebd4

tests/robotests/src/com/android/settings/network/NetworkProviderSettingsTest.java

+4 −0
Original line number Diff line number Diff line
@@ -88,6 +88,8 @@ public class NetworkProviderSettingsTest { 
    @Mock

    private WifiManager mWifiManager;

    @Mock

    private UserManager mUserManager;

    @Mock

    private DataUsagePreference mDataUsagePreference;

    private Context mContext;

    private NetworkProviderSettings mNetworkProviderSettings;
@@ -115,6 +117,8 @@ public class NetworkProviderSettingsTest { 
        doReturn(mPreferenceManager).when(mNetworkProviderSettings).getPreferenceManager();

        doReturn(mPowerManager).when(mContext).getSystemService(PowerManager.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();

        mNetworkProviderSettings.mAddWifiNetworkPreference = new AddWifiNetworkPreference(mContext);

        mNetworkProviderSettings.mSavedNetworksPreference = new Preference(mContext);