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

Commit d2101762 authored by Rafael Prado's avatar Rafael Prado
Browse files

Split policy removal test in two to test no call case. 

Flag: android.app.supervision.flags.enable_remove_policies_on_supervision_disable
Bug: 414575617
Change-Id: I97a4ed278a09c1ee1fbe13ead950af521b566ca1
parent 5d978bb6

services/tests/servicestests/src/com/android/server/supervision/SupervisionServiceTest.kt

+9 −6
Original line number Diff line number Diff line
@@ -263,18 +263,21 @@ class SupervisionServiceTest { 
    @Test

    @RequiresFlagsEnabled(Flags.FLAG_ENABLE_REMOVE_POLICIES_ON_SUPERVISION_DISABLE)

    fun setSupervisionEnabledForUser_removesPoliciesWhenDisabling() {

        assertThat(service.isSupervisionEnabledForUser(USER_ID)).isFalse()

        service.setSupervisionEnabledForUser(USER_ID, true)



        verify(mockDpmInternal, never()).removePoliciesForAdmins(any(), any())

        assertThat(service.isSupervisionEnabledForUser(USER_ID)).isTrue()



        service.setSupervisionEnabledForUser(USER_ID, false)



        assertThat(service.isSupervisionEnabledForUser(USER_ID)).isFalse()

        verify(mockDpmInternal).removePoliciesForAdmins(eq(systemSupervisionPackage), eq(USER_ID))

    }



    @Test

    @RequiresFlagsEnabled(Flags.FLAG_ENABLE_REMOVE_POLICIES_ON_SUPERVISION_DISABLE)

    fun setSupervisionEnabledForUser_doesntRemovePoliciesWhenEnabling() {

        service.setSupervisionEnabledForUser(USER_ID, true)



        assertThat(service.isSupervisionEnabledForUser(USER_ID)).isTrue()

        verify(mockDpmInternal, never()).removePoliciesForAdmins(any(), any())

    }



    @Test

    fun setSupervisionEnabledForUser_internal() {

        putSecureSetting(BROWSER_CONTENT_FILTERS_ENABLED, 1)