commit d292e434d2571dd5beb70250a97896089cc9fbe1
author Googler <noreply@google.com> Thu Sep 02 17:32:52 2021 -0700
committer Michael Hoisie <hoisie@google.com> Mon Sep 13 21:53:12 2021 -0700
tree 54d10f04f6e8d8be37ebdd25501e8a027ed6fb63
parent 6c7a59f42e59d2bfd1864943f76735134f946cae

Add ability to uninstall app widgets to ShadowAppWidgetManager.java

PiperOrigin-RevId: 394579316

robolectric/src/test/java/org/robolectric/shadows/ShadowAppWidgetManagerTest.java

diff --git a/robolectric/src/test/java/org/robolectric/shadows/ShadowAppWidgetManagerTest.java b/robolectric/src/test/java/org/robolectric/shadows/ShadowAppWidgetManagerTest.java
index cab172f..3774edb 100644
--- a/robolectric/src/test/java/org/robolectric/shadows/ShadowAppWidgetManagerTest.java
+++ b/robolectric/src/test/java/org/robolectric/shadows/ShadowAppWidgetManagerTest.java
@@ -210,6 +210,45 @@
   }
 
   @Test
+  public void removeInstalledProviders_returnsWidgetList() {
+    AppWidgetProviderInfo info1 = new AppWidgetProviderInfo();
+    info1.label = "abc";
+    AppWidgetProviderInfo info2 = new AppWidgetProviderInfo();
+    info2.label = "def";
+    shadowAppWidgetManager.addInstalledProvider(info1);
+    shadowAppWidgetManager.addInstalledProvider(info2);
+    List<AppWidgetProviderInfo> installedProviders = appWidgetManager.getInstalledProviders();
+    assertEquals(2, installedProviders.size());
+    assertEquals(info1, installedProviders.get(0));
+    assertEquals(info2, installedProviders.get(1));
+    assertTrue(shadowAppWidgetManager.removeInstalledProvider(info1));
+    installedProviders = appWidgetManager.getInstalledProviders();
+    assertEquals(1, installedProviders.size());
+    assertEquals(info2, installedProviders.get(0));
+  }
+
+  @Test
+  public void tryRemoveNotInstalledProviders_noChange() {
+    AppWidgetProviderInfo info1 = new AppWidgetProviderInfo();
+    info1.label = "abc";
+    AppWidgetProviderInfo info2 = new AppWidgetProviderInfo();
+    info2.label = "def";
+    AppWidgetProviderInfo info3 = new AppWidgetProviderInfo();
+    info2.label = "efa";
+    shadowAppWidgetManager.addInstalledProvider(info1);
+    shadowAppWidgetManager.addInstalledProvider(info2);
+    List<AppWidgetProviderInfo> installedProviders = appWidgetManager.getInstalledProviders();
+    assertEquals(2, installedProviders.size());
+    assertEquals(info1, installedProviders.get(0));
+    assertEquals(info2, installedProviders.get(1));
+    assertFalse(shadowAppWidgetManager.removeInstalledProvider(info3));
+    installedProviders = appWidgetManager.getInstalledProviders();
+    assertEquals(2, installedProviders.size());
+    assertEquals(info1, installedProviders.get(0));
+    assertEquals(info2, installedProviders.get(1));
+  }
+
+  @Test
   @Config(minSdk = L)
   public void getInstalledProvidersForProfile_returnsWidgetList() {
     UserHandle userHandle = UserHandle.CURRENT;

shadows/framework/src/main/java/org/robolectric/shadows/ShadowAppWidgetManager.java

diff --git a/shadows/framework/src/main/java/org/robolectric/shadows/ShadowAppWidgetManager.java b/shadows/framework/src/main/java/org/robolectric/shadows/ShadowAppWidgetManager.java
index 4fe89a8..b747854 100644
--- a/shadows/framework/src/main/java/org/robolectric/shadows/ShadowAppWidgetManager.java
+++ b/shadows/framework/src/main/java/org/robolectric/shadows/ShadowAppWidgetManager.java
@@ -155,6 +155,10 @@
     installedProviders.add(appWidgetProviderInfo);
   }
 
+  public boolean removeInstalledProvider(AppWidgetProviderInfo appWidgetProviderInfo) {
+    return installedProviders.remove(appWidgetProviderInfo);
+  }
+
   public void addInstalledProvidersForProfile(
       UserHandle userHandle, AppWidgetProviderInfo appWidgetProviderInfo) {
     installedProvidersForProfile.put(userHandle, appWidgetProviderInfo);