Update user restrictions on parent in TestDPC

* Previously, TestDPC invoked UserManager
  hasUserRestriction() when displaying the
  restrictions set on the parent instance.
  This was incorrect, as it was showing the
  user restrictions applied to the managed
  profile rather than the primary profile.
* This is now updated to use DevicePolicyManager
  getUserRestrictions() on the parent instance
  to check if a user restriction is set on the
  primary profile.

Bug: 149743941
Test: Manual testing
Change-Id: Iaa9e765d69aea251d8b8aecfecbb90502eb89abf

Author: Alex Johnston

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

@@ -4,7 +4,6 @@
 import android.content.ComponentName;
 import android.content.Context;
 import android.os.Bundle;
-import android.os.UserManager;
 import android.util.Log;
 import android.widget.Toast;
 
@@ -24,22 +23,21 @@ public class UserRestrictionsParentDisplayFragment extends BaseSearchablePolicyP
     private static final String TAG = "UserRestrictionsParent";
 
     private DevicePolicyManager mParentDevicePolicyManager;
-    private UserManager mUserManager;
     private ComponentName mAdminComponentName;
 
     @RequiresApi(api = Util.R_VERSION_CODE)
     @Override
     public void onCreate(Bundle savedInstanceState) {
         DevicePolicyManager mDevicePolicyManager = (DevicePolicyManager)
                 getActivity().getSystemService(Context.DEVICE_POLICY_SERVICE);
-        mUserManager = (UserManager) getActivity().getSystemService(Context.USER_SERVICE);
         mAdminComponentName = DeviceAdminReceiver.getComponentName(getActivity());
         mParentDevicePolicyManager = mDevicePolicyManager
                 .getParentProfileInstance(mAdminComponentName);
         super.onCreate(savedInstanceState);
         getActivity().getActionBar().setTitle(R.string.user_restrictions_management_title);
     }
 
+    @RequiresApi(api = Util.R_VERSION_CODE)
     @Override
     public void onCreatePreferences(Bundle savedInstanceState, String rootKey) {
         PreferenceScreen preferenceScreen = getPreferenceManager().createPreferenceScreen(
@@ -59,6 +57,7 @@ public void onCreatePreferences(Bundle savedInstanceState, String rootKey) {
         constrainPreferences();
     }
 
+    @RequiresApi(api = Util.R_VERSION_CODE)
     @Override
     public void onResume() {
         super.onResume();
@@ -70,6 +69,7 @@ public boolean isAvailable(Context context) {
         return true;
     }
 
+    @RequiresApi(api = Util.R_VERSION_CODE)
     @Override
     public boolean onPreferenceChange(Preference preference, Object newValue) {
         String restriction = preference.getKey();
@@ -89,16 +89,18 @@ public boolean onPreferenceChange(Preference preference, Object newValue) {
         }
     }
 
+    @RequiresApi(api = Util.R_VERSION_CODE)
     private void updateAllUserRestrictions() {
         for (UserRestriction restriction : UserRestriction.PROFILE_OWNER_ORG_DEVICE_RESTRICTIONS) {
             updateUserRestriction(restriction.key);
         }
     }
 
+    @RequiresApi(api = Util.R_VERSION_CODE)
     private void updateUserRestriction(String userRestriction) {
         DpcSwitchPreference preference = (DpcSwitchPreference) findPreference(userRestriction);
-        boolean disallowed = mUserManager.hasUserRestriction(userRestriction);
-        preference.setChecked(disallowed);
+        Bundle restrictions = mParentDevicePolicyManager.getUserRestrictions(mAdminComponentName);
+        preference.setChecked(restrictions.containsKey(userRestriction));
     }
 
     private void constrainPreferences() {