Add shadows for DevicePolicyManager#setGlobalSetting

PiperOrigin-RevId: 512558848
robolectric · Apr 3, 2023 · 9f640b6 · 9f640b6
1 parent 16e72ad
commit 9f640b6
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 0 deletions.
diff --git a/robolectric/src/test/java/org/robolectric/shadows/ShadowDevicePolicyManagerTest.java b/robolectric/src/test/java/org/robolectric/shadows/ShadowDevicePolicyManagerTest.java
@@ -54,6 +54,7 @@
 import android.os.PersistableBundle;
 import android.os.UserHandle;
 import android.os.UserManager;
+import android.provider.Settings;
 import androidx.test.core.app.ApplicationProvider;
 import androidx.test.ext.junit.runners.AndroidJUnit4;
 import com.google.common.collect.ImmutableSet;
@@ -2357,6 +2358,28 @@ public void getUserProvisioningState_returnsSetUserProvisioningState() {
     assertThat(devicePolicyManager.getUserProvisioningState()).isEqualTo(STATE_USER_UNMANAGED);
   }
 
+  @Config(minSdk = M)
+  @Test
+  public void setGlobalSetting_globalSettingSet() {
+    String enabled = "1";
+
+    devicePolicyManager.setGlobalSetting(testComponent, Settings.Global.ADB_ENABLED, enabled);
+
+    assertThat(Settings.Global.getString(context.getContentResolver(), Settings.Global.ADB_ENABLED))
+        .isEqualTo(enabled);
+  }
+
+  @Config(minSdk = M)
+  @Test
+  public void setGlobalSetting_settingCanBeReturnedAsInteger() {
+    String enabled = "1";
+
+    devicePolicyManager.setGlobalSetting(testComponent, Settings.Global.ADB_ENABLED, enabled);
+
+    assertThat(Settings.Global.getInt(context.getContentResolver(), Settings.Global.ADB_ENABLED))
+        .isEqualTo(1);
+  }
+
   private ServiceConnection buildServiceConnection() {
     return new ServiceConnection() {
       @Override

diff --git a/shadows/framework/src/main/java/org/robolectric/shadows/ShadowDevicePolicyManager.java b/shadows/framework/src/main/java/org/robolectric/shadows/ShadowDevicePolicyManager.java
@@ -51,6 +51,7 @@
 import android.os.PersistableBundle;
 import android.os.Process;
 import android.os.UserHandle;
+import android.provider.Settings;
 import android.text.TextUtils;
 import com.android.internal.util.Preconditions;
 import com.google.common.collect.ImmutableList;
@@ -1321,6 +1322,11 @@ protected int getPermissionPolicy(ComponentName admin) {
     return permissionPolicy;
   }
 
+  @Implementation(minSdk = M)
+  protected void setGlobalSetting(@NonNull ComponentName admin, String setting, String value) {
+    Settings.Global.putString(context.getContentResolver(), setting, value);
+  }
+
   /**
    * Grants a particular device policy for an active ComponentName.
    *