tests/tests/permission3/src/android/permission3/cts/MediaPermissionTest.kt

/*
 * Copyright (C) 2022 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package android.permission3.cts

import android.Manifest
import android.os.Build
import androidx.test.filters.SdkSuppress
import com.android.compatibility.common.util.SystemUtil
import org.junit.Assume
import org.junit.Test

/**
 * Tests media storage supergroup behavior. I.e., on a T+ platform, for legacy (targetSdk<T) apps,
 * the storage permission groups (STORAGE, AURAL, and VISUAL) form a supergroup, which effectively
 * treats them as one group and therefore their permission state must always be equal.
 */
@SdkSuppress(minSdkVersion = Build.VERSION_CODES.TIRAMISU, codeName = "Tiramisu")
class MediaPermissionTest : BaseUsePermissionTest() {
    private fun assertStorageAndMediaPermissionState(state: Boolean) {
        for (permission in STORAGE_AND_MEDIA_PERMISSIONS) {
            assertAppHasPermission(permission, state)
        }
    }

    @Test
    fun testWhenRESIsGrantedFromGrantDialogThenShouldGrantAllPermissions() {
        installPackage(APP_APK_PATH_23)
        requestAppPermissionsAndAssertResult(
            android.Manifest.permission.READ_EXTERNAL_STORAGE to true
        ) {
            clickPermissionRequestAllowButton()
        }
        assertStorageAndMediaPermissionState(true)
    }

    @Test
    fun testWhenRESIsGrantedManuallyThenShouldGrantAllPermissions() {
        installPackage(APP_APK_PATH_23)
        grantAppPermissions(android.Manifest.permission.READ_EXTERNAL_STORAGE, targetSdk = 23)
        assertStorageAndMediaPermissionState(true)
    }

    @Test
    fun testWhenAuralIsGrantedManuallyThenShouldGrantAllPermissions() {
        installPackage(APP_APK_PATH_23)
        grantAppPermissions(android.Manifest.permission.READ_MEDIA_AUDIO, targetSdk = 23)
        assertStorageAndMediaPermissionState(true)
    }

    @Test
    fun testWhenVisualIsGrantedManuallyThenShouldGrantAllPermissions() {
        installPackage(APP_APK_PATH_23)
        grantAppPermissions(android.Manifest.permission.READ_MEDIA_VIDEO, targetSdk = 23)
        assertStorageAndMediaPermissionState(true)
    }

    @Test
    fun testWhenRESIsDeniedFromGrantDialogThenShouldDenyAllPermissions() {
        installPackage(APP_APK_PATH_23)
        requestAppPermissionsAndAssertResult(
            android.Manifest.permission.READ_EXTERNAL_STORAGE to false
        ) {
            clickPermissionRequestDenyButton()
        }
        assertStorageAndMediaPermissionState(false)
    }

    @Test
    fun testWhenRESIsDeniedManuallyThenShouldDenyAllPermissions() {
        installPackage(APP_APK_PATH_23)
        grantAppPermissions(android.Manifest.permission.READ_EXTERNAL_STORAGE, targetSdk = 23)
        revokeAppPermissions(android.Manifest.permission.READ_EXTERNAL_STORAGE, targetSdk = 23)
        assertStorageAndMediaPermissionState(false)
    }

    @Test
    fun testWhenAuralIsDeniedManuallyThenShouldDenyAllPermissions() {
        installPackage(APP_APK_PATH_23)
        grantAppPermissions(android.Manifest.permission.READ_MEDIA_AUDIO, targetSdk = 23)
        revokeAppPermissions(android.Manifest.permission.READ_MEDIA_AUDIO, targetSdk = 23)
        assertStorageAndMediaPermissionState(false)
    }

    @Test
    fun testWhenVisualIsDeniedManuallyThenShouldDenyAllPermissions() {
        // TODO: Re-enable after b/239249703 is fixed
        Assume.assumeFalse("skip on TV due to flaky", isTv)
        installPackage(APP_APK_PATH_23)
        grantAppPermissions(android.Manifest.permission.READ_MEDIA_VIDEO, targetSdk = 23)
        revokeAppPermissions(android.Manifest.permission.READ_MEDIA_VIDEO, targetSdk = 23)
        assertStorageAndMediaPermissionState(false)
    }

    @Test
    fun testWhenA33AppRequestsStorageThenNoDialogAndNoGrant() {
        installPackage(APP_APK_PATH_MEDIA_PERMISSION_33_WITH_STORAGE)
        requestAppPermissions(
            android.Manifest.permission.READ_EXTERNAL_STORAGE,
            android.Manifest.permission.WRITE_EXTERNAL_STORAGE
        ) {
        }
        assertStorageAndMediaPermissionState(false)
    }

    @Test
    fun testWhenA33AppRequestsAuralThenDialogAndGrant() {
        installPackage(APP_APK_PATH_LATEST)
        requestAppPermissions(android.Manifest.permission.READ_MEDIA_AUDIO) {
            clickPermissionRequestAllowButton()
        }
        assertAppHasPermission(android.Manifest.permission.READ_EXTERNAL_STORAGE, false)
        assertAppHasPermission(android.Manifest.permission.WRITE_EXTERNAL_STORAGE, false)
        assertAppHasPermission(android.Manifest.permission.READ_MEDIA_AUDIO, true)
        assertAppHasPermission(android.Manifest.permission.READ_MEDIA_VIDEO, false)
        assertAppHasPermission(android.Manifest.permission.READ_MEDIA_IMAGES, false)
    }

    @Test
    fun testWhenA33AppRequestsVisualThenDialogAndGrant() {
        installPackage(APP_APK_PATH_LATEST)
        requestAppPermissions(
            android.Manifest.permission.READ_MEDIA_VIDEO,
            android.Manifest.permission.READ_MEDIA_IMAGES
        ) {
            clickPermissionRequestAllowButton()
        }
        assertAppHasPermission(android.Manifest.permission.READ_EXTERNAL_STORAGE, false)
        assertAppHasPermission(android.Manifest.permission.WRITE_EXTERNAL_STORAGE, false)
        assertAppHasPermission(android.Manifest.permission.READ_MEDIA_AUDIO, false)
        assertAppHasPermission(android.Manifest.permission.READ_MEDIA_VIDEO, true)
        assertAppHasPermission(android.Manifest.permission.READ_MEDIA_IMAGES, true)
    }

    @Test
    fun testWhenA30AppRequestsStorageWhenMediaPermsHaveRWRFlag() {
        installPackage(APP_APK_PATH_30)

        requestAppPermissionsAndAssertResult(
            Manifest.permission.READ_EXTERNAL_STORAGE to true
        ) {
            clickPermissionRequestAllowButton()
        }

        fun setRevokeWhenRequested(permission: String) = SystemUtil.runShellCommandOrThrow(
            "pm set-permission-flags android.permission3.cts.usepermission " +
                permission + " revoke-when-requested")
        setRevokeWhenRequested("android.permission.READ_MEDIA_AUDIO")
        setRevokeWhenRequested("android.permission.READ_MEDIA_VIDEO")
        setRevokeWhenRequested("android.permission.READ_MEDIA_IMAGES")

        requestAppPermissionsAndAssertResult(
            Manifest.permission.READ_EXTERNAL_STORAGE to true
        ) {
            // No dialog should appear
        }

        assertAppHasPermission(Manifest.permission.READ_EXTERNAL_STORAGE, true)
        assertAppHasPermission(Manifest.permission.READ_MEDIA_AUDIO, true)
        assertAppHasPermission(Manifest.permission.READ_MEDIA_VIDEO, true)
        assertAppHasPermission(Manifest.permission.READ_MEDIA_IMAGES, true)
    }
}