Added transfer-ownership command.

Test: adb shell dumpsys activity service --user 10 com.afwsamples.testdpc  transfer-ownership com.android.cts.verifier/.managedprovisioning.DeviceAdminTestReceive

Bug: 171350084
Bug: 179100903

Test: adb shell dumpsys activity service --user 10 com.afwsamples.testdpc

Change-Id: If06f13113e4cc6cf174790191ea3a5ac0026c3c9
(cherry picked from commit 257a793666382a6348ac856ea298e31f84c5c517)

Author: the-felipeal

app/src/main/java/com/afwsamples/testdpc/DevicePolicyManagerGateway.java

@@ -18,6 +18,7 @@
 import android.app.admin.DevicePolicyManager;
 import android.content.ComponentName;
 import android.graphics.Bitmap;
+import android.os.PersistableBundle;
 import android.os.UserHandle;
 import android.os.UserManager;
 import androidx.annotation.NonNull;
@@ -218,6 +219,12 @@ void setPasswordQuality(int quality, @NonNull Consumer<Void> onSuccess,
      */
     int getPasswordQuality();
 
+    /**
+     * See {@link android.app.admin.DevicePolicyManager#transferOwnership(ComponentName, ComponentName, android.os.PersistableBundle)}.
+     */
+    void transferOwnership(@NonNull ComponentName target, @Nullable PersistableBundle bundle,
+            @NonNull Consumer<Void> onSuccess, @NonNull Consumer<Exception> onError);
+
     /**
      * Used on error callbacks to indicate a {@link android.app.admin.DevicePolicyManager} method
      * call failed.

app/src/main/java/com/afwsamples/testdpc/DevicePolicyManagerGatewayImpl.java

@@ -21,6 +21,7 @@
 import android.content.Context;
 import android.graphics.Bitmap;
 import android.os.Bundle;
+import android.os.PersistableBundle;
 import android.os.UserHandle;
 import android.os.UserManager;
 import android.util.Log;
@@ -427,6 +428,19 @@ public int getPasswordQuality() {
         return quality;
     }
 
+    @Override
+    public void transferOwnership(ComponentName target, PersistableBundle bundle,
+            Consumer<Void> onSuccess, Consumer<Exception> onError) {
+        Log.d(TAG, "transferOwnership(" + target + ", " + bundle + ")");
+
+        try {
+            mDevicePolicyManager.transferOwnership(mAdminComponentName, target, bundle);
+            onSuccess.accept(null);
+        } catch (Exception e) {
+            onError.accept(e);
+        }
+    }
+
     @Override
     public String toString() {
         return "DevicePolicyManagerGatewayImpl[" + mAdminComponentName + "]";

app/src/main/java/com/afwsamples/testdpc/ShellCommand.java

@@ -73,6 +73,7 @@ final class ShellCommand {
     private static final String CMD_CLEAR_PROFILE_OWNER = "clear-profile-owner";
     private static final String CMD_SET_PASSWORD_QUALITY = "set-password-quality";
     private static final String CMD_GET_PASSWORD_QUALITY = "get-password-quality";
+    private static final String CMD_TRANSFER_OWNERSHIP = "transfer-ownership";
 
     private static final String ARG_FLAGS = "--flags";
 
@@ -179,6 +180,9 @@ public void run() {
             case CMD_GET_PASSWORD_QUALITY:
                 execute(() -> getPasswordQuality());
                 break;
+            case CMD_TRANSFER_OWNERSHIP:
+                execute(() -> transferOwnership());
+                break;
             default:
                 mWriter.printf("Invalid command: %s\n\n", cmd);
                 showUsage();
@@ -241,6 +245,8 @@ private void showUsage() {
         mWriter.printf("\t%s <QUALITY> - set password quality\n",
                 CMD_SET_PASSWORD_QUALITY);
         mWriter.printf("\t%s - get password quality\n", CMD_GET_PASSWORD_QUALITY);
+        mWriter.printf("\t%s [ADMIN]- transfer ownership to the given admin\n",
+                CMD_TRANSFER_OWNERSHIP);
     }
 
     private void createUser() {
@@ -495,6 +501,18 @@ private void getPasswordQuality() {
         mWriter.printf("password quality: %d\n", mDevicePolicyManagerGateway.getPasswordQuality());
     }
 
+    private void transferOwnership() {
+        // TODO(b/171350084): check args
+        String flatTarget = mArgs[1];
+        ComponentName target = ComponentName.unflattenFromString(flatTarget);
+
+        Log.i(TAG, "transferOwnership(" + target + ")");
+
+        mDevicePolicyManagerGateway.transferOwnership(target, /* bundle= */ null,
+                (v) -> onSuccess("Ownership transferred to %s", flatTarget),
+                (e) -> onError(e, "Error transferring ownership to %s", flatTarget));
+    }
+
     private void execute(@NonNull Runnable r) {
         try {
             r.run();

app/src/main/java/com/afwsamples/testdpc/transferownership/PickTransferComponentFragment.java

@@ -107,6 +107,7 @@ private String performTransfer(ComponentName target) {
         try {
             PersistableBundle persistableBundle = new PersistableBundle();
             persistableBundle.putString("random_key", "random_value");
+            // TODO: use DevicePolicyManagerGateway instead
             mDevicePolicyManager.transferOwnership(source, target, persistableBundle);
             return "Success!";
         } catch (Exception e) {