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

Commit ef9960a4 authored by TreeHugger Robot's avatar TreeHugger Robot Committed by Automerger Merge Worker
Browse files

Merge "Listen to correct intents in UserTracker" into tm-dev am: 87ab8129 am: 55b531c7 

Original change: https://googleplex-android-review.googlesource.com/c/platform/frameworks/base/+/18774497



Change-Id: Ib554962dde3ac6dd95b96cff77cbeb0a011d08d5
Signed-off-by: default avatarAutomerger Merge Worker <android-build-automerger-merge-worker@system.gserviceaccount.com>
parents b4d0b92e 55b531c7

packages/SystemUI/src/com/android/systemui/settings/UserTracker.kt

+2 −0
Original line number Diff line number Diff line
@@ -46,6 +46,8 @@ interface UserTracker : UserContentResolverProvider, UserContextProvider { 


    /**

     * List of profiles associated with the current user.

     *

     * Quiet work profiles will still appear here, but will have the `QUIET_MODE` flag.

     */

    val userProfiles: List<UserInfo>

packages/SystemUI/src/com/android/systemui/settings/UserTrackerImpl.kt

+9 −2
Original line number Diff line number Diff line
@@ -108,8 +108,12 @@ class UserTrackerImpl internal constructor( 


        val filter = IntentFilter().apply {

            addAction(Intent.ACTION_USER_SWITCHED)

            // These get called when a managed profile goes in or out of quiet mode.

            addAction(Intent.ACTION_MANAGED_PROFILE_AVAILABLE)

            addAction(Intent.ACTION_MANAGED_PROFILE_UNAVAILABLE)



            addAction(Intent.ACTION_MANAGED_PROFILE_REMOVED)

            addAction(Intent.ACTION_MANAGED_PROFILE_UNLOCKED)

        }

        context.registerReceiverForAllUsers(this, filter, null /* permission */, backgroundHandler)
@@ -121,7 +125,10 @@ class UserTrackerImpl internal constructor( 
            Intent.ACTION_USER_SWITCHED -> {

                handleSwitchUser(intent.getIntExtra(Intent.EXTRA_USER_HANDLE, UserHandle.USER_NULL))

            }

            Intent.ACTION_MANAGED_PROFILE_AVAILABLE, Intent.ACTION_MANAGED_PROFILE_UNAVAILABLE -> {

            Intent.ACTION_MANAGED_PROFILE_AVAILABLE,

            Intent.ACTION_MANAGED_PROFILE_UNAVAILABLE,

            Intent.ACTION_MANAGED_PROFILE_REMOVED,

            Intent.ACTION_MANAGED_PROFILE_UNLOCKED -> {

                handleProfilesChanged()

            }

        }
@@ -208,7 +215,7 @@ class UserTrackerImpl internal constructor( 
        pw.println("Initialized: $initialized")

        if (initialized) {

            pw.println("userId: $userId")

            val ids = userProfiles.map { it.id }

            val ids = userProfiles.map { it.toFullString() }

            pw.println("userProfiles: $ids")

        }

        val list = synchronized(callbacks) {

packages/SystemUI/tests/src/com/android/systemui/settings/UserTrackerImplTest.kt

+61 −0
Original line number Diff line number Diff line
@@ -189,6 +189,67 @@ class UserTrackerImplTest : SysuiTestCase() { 
        assertThat(tracker.userProfiles.map { it.id }).containsExactly(tracker.userId, profileID)

    }



    fun testManagedProfileUnavailable() {

        tracker.initialize(0)

        val profileID = tracker.userId + 10



        `when`(userManager.getProfiles(anyInt())).thenAnswer { invocation ->

            val id = invocation.getArgument<Int>(0)

            val info = UserInfo(id, "", UserInfo.FLAG_FULL)

            val infoProfile = UserInfo(

                    id + 10,

                    "",

                    "",

                    UserInfo.FLAG_MANAGED_PROFILE or UserInfo.FLAG_QUIET_MODE,

                    UserManager.USER_TYPE_PROFILE_MANAGED

            )

            infoProfile.profileGroupId = id

            listOf(info, infoProfile)

        }



        val intent = Intent(Intent.ACTION_MANAGED_PROFILE_UNAVAILABLE)

                .putExtra(Intent.EXTRA_USER, UserHandle.of(profileID))

        tracker.onReceive(context, intent)



        assertThat(tracker.userProfiles.map { it.id }).containsExactly(tracker.userId, profileID)

    }



    fun testManagedProfileStartedAndRemoved() {

        tracker.initialize(0)

        val profileID = tracker.userId + 10



        `when`(userManager.getProfiles(anyInt())).thenAnswer { invocation ->

            val id = invocation.getArgument<Int>(0)

            val info = UserInfo(id, "", UserInfo.FLAG_FULL)

            val infoProfile = UserInfo(

                    id + 10,

                    "",

                    "",

                    UserInfo.FLAG_MANAGED_PROFILE,

                    UserManager.USER_TYPE_PROFILE_MANAGED

            )

            infoProfile.profileGroupId = id

            listOf(info, infoProfile)

        }



        // Managed profile started

        val intent = Intent(Intent.ACTION_MANAGED_PROFILE_UNLOCKED)

                .putExtra(Intent.EXTRA_USER, UserHandle.of(profileID))

        tracker.onReceive(context, intent)



        assertThat(tracker.userProfiles.map { it.id }).containsExactly(tracker.userId, profileID)



        `when`(userManager.getProfiles(anyInt())).thenAnswer { invocation ->

            listOf(UserInfo(invocation.getArgument(0), "", UserInfo.FLAG_FULL))

        }



        val intent2 = Intent(Intent.ACTION_MANAGED_PROFILE_REMOVED)

                .putExtra(Intent.EXTRA_USER, UserHandle.of(profileID))

        tracker.onReceive(context, intent2)



        assertThat(tracker.userProfiles.map { it.id }).containsExactly(tracker.userId)

    }



    @Test

    fun testCallbackNotCalledOnAdd() {

        tracker.initialize(0)