Make TestDPC use reflection for updated COSU APIs

Test: startuserInBackground, stopUser, logoutUser and clearApplicationUserData can be invoked
Bug: 73066020
Change-Id: Iec61f9355d0a2b054ed3098afbf61a7a5a72354b

Author: alexchau-google

app/src/main/java/com/afwsamples/testdpc/policy/PolicyManagementFragment.java

@@ -149,6 +149,7 @@
 import java.util.List;
 import java.util.Set;
 import java.util.TimeZone;
+import java.util.concurrent.Executor;
 import java.util.stream.Collectors;
 
 import static android.os.UserManager.DISALLOW_INSTALL_UNKNOWN_SOURCES;
@@ -382,6 +383,9 @@ public class PolicyManagementFragment extends BaseSearchablePolicyPreferenceFrag
     public static final String GMSCORE_BACKUP_TRANSPORT =
             "com.google.android.gms.backup.BackupTransportService";
 
+    private static final int USER_OPERATION_ERROR_UNKNOWN = 1;
+    private static final int USER_OPERATION_SUCCESS = 0;
+
     private DevicePolicyManager mDevicePolicyManager;
     private PackageManager mPackageManager;
     private String mPackageName;
@@ -1844,17 +1848,17 @@ private void showSwitchUserPrompt() {
     @TargetApi(28)
     private void showStartUserInBackgroundPrompt() {
         showChooseUserPrompt(R.string.start_user_in_background, userHandle -> {
-            boolean success = false;
+            int status = USER_OPERATION_ERROR_UNKNOWN;
             try {
-                success = (boolean) ReflectionUtil.invoke(
+                status = (int) ReflectionUtil.invoke(
                         mDevicePolicyManager,
                         "startUserInBackground",
                         mAdminComponentName,
                         userHandle);
             } catch (ReflectionUtil.ReflectionIsTemporaryException e) {
                 Log.e(TAG,"Can't invoke startUserInBackground", e);
             }
-            showToast(success
+            showToast(status == USER_OPERATION_SUCCESS
                     ? R.string.user_started_in_background
                     : R.string.failed_to_start_user_in_background);
         });
@@ -1867,9 +1871,18 @@ private void showStartUserInBackgroundPrompt() {
     @TargetApi(28)
     private void showStopUserPrompt() {
         showChooseUserPrompt(R.string.stop_user, userHandle -> {
-            boolean success =
-                    mDevicePolicyManager.stopUser(mAdminComponentName, userHandle);
-            showToast(success ? R.string.user_stopped : R.string.failed_to_stop_user);
+            int status = USER_OPERATION_ERROR_UNKNOWN;
+            try {
+                status = (int) ReflectionUtil.invoke(
+                        mDevicePolicyManager,
+                        "stopUser",
+                        mAdminComponentName,
+                        userHandle);
+            } catch (ReflectionUtil.ReflectionIsTemporaryException e) {
+                Log.e(TAG,"Can't invoke stopUser", e);
+            }
+            showToast(status == USER_OPERATION_SUCCESS ? R.string.user_stopped
+                    : R.string.failed_to_stop_user);
         });
     }
 
@@ -1908,8 +1921,17 @@ private void showChooseUserPrompt(int titleResId, UserCallback callback) {
      */
     @TargetApi(28)
     private void logoutUser() {
-        boolean success = mDevicePolicyManager.logoutUser(mAdminComponentName);
-        showToast(success ? R.string.user_logouted : R.string.failed_to_logout_user);
+        int status = USER_OPERATION_ERROR_UNKNOWN;
+        try {
+            status = (int) ReflectionUtil.invoke(
+                    mDevicePolicyManager,
+                    "logoutUser",
+                    mAdminComponentName);
+        } catch (ReflectionUtil.ReflectionIsTemporaryException e) {
+            Log.e(TAG,"Can't invoke logoutUser", e);
+        }
+        showToast(status == USER_OPERATION_SUCCESS ? R.string.user_logouted
+                : R.string.failed_to_logout_user);
     }
 
     /**
@@ -2826,14 +2848,21 @@ private void showClearAppDataPrompt() {
 
     @TargetApi(28)
     private void clearApplicationUserData(String packageName) {
-        if (!mDevicePolicyManager.clearApplicationUserData(
-                mAdminComponentName,
-                packageName,
-                new MainThreadExecutor(),
-                (__, succeed) -> showToast(
-                        succeed ? R.string.clear_app_data_success : R.string.clear_app_data_failure,
-                        packageName))) {
-            showToast(R.string.clear_app_data_failure, packageName);
+        try {
+            ReflectionUtil.invoke(
+                    mDevicePolicyManager, "clearApplicationUserData",
+                    new Class[]{ComponentName.class, String.class, Executor.class,
+                            DevicePolicyManager.OnClearApplicationUserDataListener.class},
+                    mAdminComponentName,
+                    packageName,
+                    new MainThreadExecutor(),
+                    (DevicePolicyManager.OnClearApplicationUserDataListener)
+                            (__, succeed) -> showToast(
+                                    succeed ? R.string.clear_app_data_success
+                                            : R.string.clear_app_data_failure,
+                                    packageName));
+        } catch (ReflectionUtil.ReflectionIsTemporaryException e) {
+            Log.e(LOG_TAG, "cannot invoke clearApplicationUserData", e);
         }
     }