Merge "Split policy removal test in two to test no call case." into main (88399290) · Commits · e / os / android_frameworks_base

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)