1 /* 2 * Copyright (C) 2022 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17 package com.android.permissioncontroller.permissionui.ui.handheld 18 19 import android.content.Intent 20 import android.permission.cts.PermissionUtils.grantPermission 21 import android.permission.cts.PermissionUtils.install 22 import android.permission.cts.PermissionUtils.revokePermission 23 import android.permission.cts.PermissionUtils.uninstallApp 24 import androidx.test.uiautomator.By 25 import androidx.test.ext.junit.runners.AndroidJUnit4 26 import com.android.compatibility.common.util.SystemUtil.eventually 27 import com.android.compatibility.common.util.SystemUtil.runWithShellPermissionIdentity 28 import com.android.compatibility.common.util.UiAutomatorUtils2.waitFindObject 29 import com.android.permissioncontroller.permissionui.getUsageCountsFromUi 30 import com.android.permissioncontroller.permissionui.wakeUpScreen 31 import com.google.common.truth.Truth.assertThat 32 import org.junit.After 33 import org.junit.Before 34 import org.junit.Test 35 import org.junit.runner.RunWith 36 37 /** 38 * Simple tests for {@link ManageCustomPermissionsFragment} 39 */ 40 @RunWith(AndroidJUnit4::class) 41 class ManageCustomPermissionsFragmentTest : BaseHandheldPermissionUiTest() { 42 private val ONE_PERMISSION_DEFINER_APK = 43 "/data/local/tmp/permissioncontroller/tests/permissionui/" + 44 "PermissionUiDefineAdditionalPermissionApp.apk" 45 private val PERMISSION_USER_APK = 46 "/data/local/tmp/permissioncontroller/tests/permissionui/" + 47 "PermissionUiUseAdditionalPermissionApp.apk" 48 private val DEFINER_PKG = "com.android.permissioncontroller.tests.appthatdefinespermission" 49 private val USER_PKG = "com.android.permissioncontroller.tests.appthatrequestpermission" 50 51 private val PERM_LABEL = "Permission A" 52 private val PERM = "com.android.permissioncontroller.tests.A" 53 private val ADDITIONAL_PERMISSIONS_LABEL = "Additional permissions" 54 55 @Before setupnull56 fun setup() { 57 wakeUpScreen() 58 59 runWithShellPermissionIdentity { 60 instrumentationContext.startActivity(Intent(Intent.ACTION_MANAGE_PERMISSIONS).apply { 61 addFlags(Intent.FLAG_ACTIVITY_NEW_TASK) 62 }) 63 } 64 } 65 66 @Test groupSummaryGetsUpdatedWhenPermissionGetsGrantednull67 fun groupSummaryGetsUpdatedWhenPermissionGetsGranted() { 68 install(ONE_PERMISSION_DEFINER_APK) 69 install(PERMISSION_USER_APK) 70 waitFindObject(By.textContains(ADDITIONAL_PERMISSIONS_LABEL)).click() 71 waitFindObject(By.textContains(PERM_LABEL)) 72 73 val original = getUsageCountsFromUi(PERM_LABEL) 74 75 grantPermission(USER_PKG, PERM) 76 eventually { 77 assertThat(getUsageCountsFromUi(PERM_LABEL).granted).isEqualTo(original.granted + 1) 78 } 79 } 80 81 @Test groupSummaryGetsUpdatedWhenPermissionGetsRevokednull82 fun groupSummaryGetsUpdatedWhenPermissionGetsRevoked() { 83 install(ONE_PERMISSION_DEFINER_APK) 84 install(PERMISSION_USER_APK) 85 waitFindObject(By.textContains(ADDITIONAL_PERMISSIONS_LABEL)).click() 86 waitFindObject(By.textContains(PERM_LABEL)) 87 88 val original = getUsageCountsFromUi(PERM_LABEL) 89 90 grantPermission(USER_PKG, PERM) 91 eventually { 92 assertThat(getUsageCountsFromUi(PERM_LABEL)).isNotEqualTo(original) 93 } 94 95 revokePermission(USER_PKG, PERM) 96 eventually { 97 assertThat(getUsageCountsFromUi(PERM_LABEL)).isEqualTo(original) 98 } 99 } 100 101 @After tearDownnull102 fun tearDown() { 103 uninstallApp(DEFINER_PKG) 104 uninstallApp(USER_PKG) 105 106 uiDevice.pressBack() 107 } 108 } 109