Merge "Clean up ShadowUserManager"

import android.content.ContentResolver;

import android.content.Intent;

import android.os.UserHandle;

import android.os.UserManager;

import android.provider.Settings;

import android.view.View;

import android.widget.RadioButton;

import com.android.settings.RestrictedRadioButton;

import com.android.settings.notification.RedactionInterstitial.RedactionInterstitialFragment;

import com.android.settings.testutils.shadow.ShadowRestrictedLockUtilsInternal;

import com.android.settings.testutils.shadow.ShadowUserManager;

import com.android.settings.testutils.shadow.ShadowUtils;

import org.junit.After;

import org.robolectric.RobolectricTestRunner;

import org.robolectric.RuntimeEnvironment;

import org.robolectric.annotation.Config;

import org.robolectric.shadow.api.Shadow;

import org.robolectric.shadows.ShadowUserManager;

@RunWith(RobolectricTestRunner.class)

@Config(shadows = {

        ShadowUtils.class,

        ShadowRestrictedLockUtilsInternal.class,

        ShadowUserManager.class,

})

public class RedactionInterstitialTest {

    private RedactionInterstitial mActivity;

    @Test

    public void managedProfileNoRestrictionsTest() {

        setupSettings(1 /* show */, 1 /* showUnredacted */);

        ShadowUserManager.getShadow().addManagedProfile(UserHandle.myUserId());

        final ShadowUserManager sum =

                Shadow.extract(RuntimeEnvironment.application.getSystemService(UserManager.class));

        sum.setManagedProfile(true);

        setupActivity();

        assertHideAllVisible(false);

    @Test

    public void managedProfileUnredactedRestrictionTest() {

        setupSettings(1 /* show */, 1 /* showUnredacted */);

        ShadowUserManager.getShadow().addManagedProfile(UserHandle.myUserId());

        final ShadowUserManager sum =

                Shadow.extract(RuntimeEnvironment.application.getSystemService(UserManager.class));

        sum.setManagedProfile(true);

        ShadowRestrictedLockUtilsInternal.setKeyguardDisabledFeatures(

                KEYGUARD_DISABLE_UNREDACTED_NOTIFICATIONS);

        setupActivity();

        mRestrictionSources.put(restrictionKey + userHandle.getIdentifier(), enforcers);

    }

    @Implementation

    protected boolean isManagedProfile(@UserIdInt int userId) {

        return mManagedProfiles.contains(userId);

    }

    public void addManagedProfile(int userId) {

        mManagedProfiles.add(userId);

    }

    @Implementation

    protected boolean isQuietModeEnabled(UserHandle userHandle) {

        return mIsQuietModeEnabled;