commit 601bcf158c310092883225dd42073ce14541b405
author William Escande <wescande@google.com> Thu Oct 07 19:39:53 2021 +0200
committer William Escande <wescande@google.com> Thu Oct 21 12:32:54 2021 +0200
tree 82ab388f68de6098b7783f86bdc57a308409da3f
parent eb2a284f9ae7cd1dbc9ce79b1179869a5c7f495f

Add cts test for new PermissionManager api

Adding checkPermissionForDataDeliveryFromDataSource

Bug: 195144968
Tag: #refactor
Test: Build
Change-Id: I0d1abe0b5411e458116cf3be5b9d732766775fed

tests/tests/permission5/src/android/permission5/cts/PermissionCheckerTest.kt

diff --git a/tests/tests/permission5/src/android/permission5/cts/PermissionCheckerTest.kt b/tests/tests/permission5/src/android/permission5/cts/PermissionCheckerTest.kt
index c7f175d..b7762c6 100644
--- a/tests/tests/permission5/src/android/permission5/cts/PermissionCheckerTest.kt
+++ b/tests/tests/permission5/src/android/permission5/cts/PermissionCheckerTest.kt
@@ -113,6 +113,43 @@
         }
     }
 
+    @Test
+    fun testCheckPermissionForDataDeliveryFromDataSource() {
+        runWithShellPermissionIdentity({
+            assertThat(
+                permissionManager.checkPermissionForDataDeliveryFromDataSource(
+                    HELPER_PERMISSION_NAME, helperAttributionSource, null
+                )
+            ).isEqualTo(PermissionManager.PERMISSION_GRANTED)
+        }, android.Manifest.permission.UPDATE_APP_OPS_STATS)
+
+        runWithShellPermissionIdentity {
+            appOpsManager.setUidMode(HELPER_APP_OP_NAME, helperUid, AppOpsManager.MODE_IGNORED)
+        }
+
+        runWithShellPermissionIdentity({
+            assertThat(
+                permissionManager.checkPermissionForDataDeliveryFromDataSource(
+                    HELPER_PERMISSION_NAME, helperAttributionSource, null
+                )
+            ).isEqualTo(PermissionManager.PERMISSION_SOFT_DENIED)
+        }, android.Manifest.permission.UPDATE_APP_OPS_STATS)
+
+        runWithShellPermissionIdentity {
+            packageManager.revokeRuntimePermission(
+                HELPER_PACKAGE_NAME, HELPER_PERMISSION_NAME, currentUser
+            )
+        }
+
+        runWithShellPermissionIdentity({
+            assertThat(
+                permissionManager.checkPermissionForDataDeliveryFromDataSource(
+                    HELPER_PERMISSION_NAME, helperAttributionSource, null
+                )
+            ).isEqualTo(PermissionManager.PERMISSION_SOFT_DENIED)
+        }, android.Manifest.permission.UPDATE_APP_OPS_STATS)
+    }
+
     companion object {
         private const val HELPER_PACKAGE_NAME = "android.permission5.cts.blamed"
         private const val HELPER_PERMISSION_NAME = android.Manifest.permission.READ_CALENDAR