Improved how setPermissionGrantState() handle failures.

Test: build
Bug: 171350084

Change-Id: I3e67beb57fc3ae2ab91dbbc11ea503afb0541be3

Author: the-felipeal

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

@@ -395,7 +395,7 @@ void setApplicationRestrictions(String packageName, Bundle settings,
      * See {@link android.app.admin.DevicePolicyManager#setPermissionGrantState(ComponentName, String, String, int)}.
      */
     void setPermissionGrantState(String packageName, String permission, int grantState,
-            @NonNull Consumer<Boolean> onSuccess, @NonNull Consumer<Exception> onError);
+            @NonNull Consumer<Void> onSuccess, @NonNull Consumer<Exception> onError);
 
     /**
      * See {@link android.app.admin.DevicePolicyManager#getPermissionGrantState(ComponentName, String, String)}.

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

@@ -776,13 +776,19 @@ public void setApplicationRestrictions(String packageName, Bundle settings,
 
     @Override
     public void setPermissionGrantState(String packageName, String permission, int grantState,
-            Consumer<Boolean> onSuccess, Consumer<Exception> onError) {
+            Consumer<Void> onSuccess, Consumer<Exception> onError) {
+        String stateName = Util.grantStateToString(grantState);
         Log.d(TAG, "setPermissionGrantState(" + packageName + ", " + permission + "): "
-                + Util.grantStateToString(grantState));
+                + stateName);
         try {
-            boolean result = mDevicePolicyManager.setPermissionGrantState(mAdminComponentName,
+            boolean success = mDevicePolicyManager.setPermissionGrantState(mAdminComponentName,
                     packageName, permission, grantState);
-            onSuccess.accept(result);
+            if (success) {
+                onSuccess.accept(null);
+            } else {
+                onError.accept(new FailedOperationException("setPermissionGrantState(%s, %s, %s)",
+                        packageName, permission, stateName));
+            }
         } catch (Exception e) {
             onError.accept(e);
         }