Merge "Set Private Space usehandle to null if we receive...

    private final ActivityManager mActivityManager;

    private  int mErrorCode;

    @GuardedBy("this")

    @Nullable

    private UserHandle mUserHandle;

    private final KeyguardManager mKeyguardManager;

    /** This variable should be accessed via {@link #getProfileBroadcastReceiver()} only. */

            mContext.unregisterReceiver(/* receiver= */ this);

        }

        @GuardedBy("PrivateSpaceMaintainer.this")

        @Override

        public void onReceive(@NonNull Context context, @NonNull Intent intent) {

            UserHandle userHandle = intent.getParcelableExtra(Intent.EXTRA_USER, UserHandle.class);

            if (intent.getAction().equals(Intent.ACTION_PROFILE_REMOVED)) {

                // This applies to all profiles getting removed, since there is no way to tell if

                // it is a private profile that got removed.

                if (userHandle.equals(getPrivateProfileHandle())) {

                    mUserHandle = null;

                }

                removeSettingsAllTasks();

                unregisterBroadcastReceiver();

                return;

import android.content.ComponentName;

import android.content.ContentResolver;

import android.content.Context;

import android.content.Intent;

import android.content.pm.PackageManager;

import android.os.Flags;

import android.os.RemoteException;

import android.os.UserHandle;

import android.os.UserManager;

import android.platform.test.flag.junit.SetFlagsRule;

import android.provider.Settings;

import androidx.test.core.app.ApplicationProvider;

import androidx.test.ext.junit.runners.AndroidJUnit4;

import com.android.compatibility.common.util.BlockingBroadcastReceiver;

import com.android.settings.privatespace.PrivateSpaceMaintainer.ErrorDeletingPrivateSpace;

import org.junit.After;

import org.junit.runner.RunWith;

import org.mockito.MockitoAnnotations;

import java.util.Objects;

import java.util.concurrent.TimeUnit;

@RunWith(AndroidJUnit4.class)

public class PrivateSpaceMaintainerTest {

    private static final String TAG = "PSMaintainerTest";

                /* enabled */ 1,

                privateSpaceMaintainer.getPrivateProfileHandle().getIdentifier());

    }

    @Test

    public void profileRemovedFromUserManager_privateSpaceNoLongerExists() {

        PrivateSpaceMaintainer privateSpaceMaintainer = PrivateSpaceMaintainer.getInstance(

                mContext);

        privateSpaceMaintainer.createPrivateSpace();

        UserHandle privateSpaceUserHandle = privateSpaceMaintainer.getPrivateProfileHandle();

        assertThat(privateSpaceMaintainer.doesPrivateSpaceExist()).isTrue();

        Intent removedIntent = new Intent(Intent.ACTION_PROFILE_REMOVED);

        assertThat(privateSpaceUserHandle).isNotNull();

        final BlockingBroadcastReceiver receiver = new BlockingBroadcastReceiver(mContext,

                removedIntent.getAction());

        receiver.register();

        Objects.requireNonNull(mContext.getSystemService(UserManager.class)).removeUser(

                privateSpaceUserHandle);

        receiver.awaitForBroadcast(TimeUnit.SECONDS.toMillis(10));

        assertThat(privateSpaceMaintainer.doesPrivateSpaceExist()).isFalse();

    }

}