Merge "Merge branch 'upstream-google' of sso://android.googlesource.com/platform/external/robolectric into upgrade" into main

Author: Treehugger Robot

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

@@ -1,6 +1,7 @@
 package org.robolectric.shadows;
 
 import static android.os.Build.VERSION_CODES.M;
+import static android.os.Build.VERSION_CODES.P;
 import static android.os.Build.VERSION_CODES.Q;
 import static com.google.common.truth.Truth.assertThat;
 import static org.junit.Assert.assertThrows;
@@ -13,6 +14,7 @@
 import android.app.backup.RestoreObserver;
 import android.app.backup.RestoreSession;
 import android.app.backup.RestoreSet;
+import android.os.UserHandle;
 import androidx.test.core.app.ApplicationProvider;
 import androidx.test.ext.junit.runners.AndroidJUnit4;
 import com.google.common.collect.ImmutableList;
@@ -93,6 +95,13 @@ public void isBackupEnabled_noPermission_shouldThrowSecurityException() {
     assertThrows(SecurityException.class, () -> backupManager.isBackupEnabled());
   }
 
+  @Test
+  @Config(minSdk = P)
+  public void setBackupS_setToFalse_shouldDisableBackup() {
+    shadowOf(backupManager).setBackupServiceActive(UserHandle.of(0), false);
+    assertThat(backupManager.isBackupServiceActive(UserHandle.of(0))).isFalse();
+  }
+
   @Test
   public void getAvailableRestoreSets_shouldCallbackToRestoreSetsAvailable() {
     RestoreSession restoreSession = backupManager.beginRestoreSession();

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

@@ -135,6 +135,7 @@
 import org.robolectric.annotation.Config;
 import org.robolectric.annotation.GetInstallerPackageNameMode;
 import org.robolectric.annotation.GetInstallerPackageNameMode.Mode;
+import org.robolectric.junit.rules.SetSystemPropertyRule;
 import org.robolectric.shadow.api.Shadow;
 import org.robolectric.shadows.ShadowPackageManager.PackageSetting;
 import org.robolectric.shadows.ShadowPackageManager.ResolveInfoComparator;
@@ -162,6 +163,7 @@ public class ShadowPackageManagerTest {
   public static final String ORIGINATING_PACKAGE_NAME = "originating.package";
   public static final String UPDATE_OWNER_PACKAGE_NAME = "update.owner.package";
 
+  @Rule public final SetSystemPropertyRule setSystemPropertyRule = new SetSystemPropertyRule();
   @Rule public TemporaryFolder temporaryFolder = new TemporaryFolder();
   private Context context;
   private PackageManager packageManager;
@@ -2198,6 +2200,23 @@ public void getApplicationIcon_applicationInfo_matches() {
     assertThat(packageManager.getApplicationIcon(applicationInfo)).isSameInstanceAs(d);
   }
 
+  @Test
+  public void getApplicationIcon_fromInfoNoShadow_returnsDrawable() {
+    setSystemPropertyRule.set("robolectric.useValidGetApplicationIcon", "true");
+    ApplicationInfo applicationInfo = new ApplicationInfo();
+    applicationInfo.packageName = TEST_PACKAGE_NAME;
+
+    Drawable d = packageManager.getApplicationIcon(applicationInfo);
+    assertThat(d).isNotNull();
+  }
+
+  @Test
+  public void getApplicationIcon_fromPackageNoShadow_throws() {
+    setSystemPropertyRule.set("robolectric.useValidGetApplicationIcon", "true");
+    assertThrows(
+        NameNotFoundException.class, () -> packageManager.getApplicationIcon(TEST_PACKAGE_NAME));
+  }
+
   @Test
   public void hasSystemFeature() {
     // uninitialized

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

@@ -3,6 +3,7 @@
 import static android.os.Build.VERSION_CODES.N;
 import static android.os.Build.VERSION_CODES.Q;
 import static com.google.common.truth.Truth.assertThat;
+import static org.junit.Assert.assertThrows;
 import static org.junit.Assert.fail;
 import static org.robolectric.Shadows.shadowOf;
 
@@ -95,6 +96,14 @@ public void startForegroundWithForegroundServiceType() {
     assertThat(service.getForegroundServiceType()).isEqualTo(64);
   }
 
+  @Test
+  public void startForegroundWithException() {
+    shadowOf(service).setThrowInStartForeground(new RuntimeException("test"));
+    Notification n = notBuilder.build();
+    RuntimeException e = assertThrows(RuntimeException.class, () -> service.startForeground(23, n));
+    assertThat(e).hasMessageThat().isEqualTo("test");
+  }
+
   @Test
   public void stopForeground() {
     service.stopForeground(true);

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

@@ -0,0 +1,76 @@
+package org.robolectric.shadows;
+
+import static android.os.Build.VERSION_CODES.O;
+import static com.google.common.truth.Truth.assertThat;
+import static org.robolectric.Shadows.shadowOf;
+
+import android.app.Activity;
+import android.app.UiAutomation;
+import android.graphics.Bitmap;
+import android.os.Looper;
+import android.os.SystemClock;
+import android.view.View;
+import android.view.ViewGroup;
+import android.view.animation.Transformation;
+import android.view.animation.TranslateAnimation;
+import androidx.test.ext.junit.runners.AndroidJUnit4;
+import androidx.test.platform.app.InstrumentationRegistry;
+import java.time.Instant;
+import java.util.concurrent.TimeUnit;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.robolectric.Robolectric;
+import org.robolectric.annotation.Config;
+import org.robolectric.annotation.GraphicsMode;
+import org.robolectric.annotation.GraphicsMode.Mode;
+
+@RunWith(AndroidJUnit4.class)
+@Config(minSdk = O)
+@GraphicsMode(Mode.NATIVE)
+public class ViewAnimationTest {
+  @Test
+  public void viewAnimations_interpolateWhenDrawIsCalled() {
+    try {
+      ShadowView.setUseRealViewAnimations(true);
+      Activity activity = Robolectric.setupActivity(Activity.class);
+      View view = new View(activity);
+      ViewGroup.LayoutParams params = new ViewGroup.LayoutParams(100, 100);
+      view.setLayoutParams(params);
+      ((ViewGroup) activity.findViewById(android.R.id.content)).addView(view);
+
+      shadowOf(Looper.getMainLooper()).runToEndOfTasks();
+
+      long startTime = SystemClock.uptimeMillis();
+
+      final TestTranslateAnimation animation = new TestTranslateAnimation(0, 100, 0, 0);
+      animation.setDuration(1000);
+      view.startAnimation(animation);
+
+      shadowOf(Looper.getMainLooper()).idleFor(500, TimeUnit.MILLISECONDS);
+      UiAutomation uiAutomation = InstrumentationRegistry.getInstrumentation().getUiAutomation();
+      Bitmap bitmap = uiAutomation.takeScreenshot();
+      assertThat(bitmap).isNotNull();
+      assertThat(animation.lastAnimationTime.toEpochMilli()).isEqualTo(startTime + 500);
+    } finally {
+      ShadowView.setUseRealViewAnimations(false);
+    }
+  }
+
+  public static class TestTranslateAnimation extends TranslateAnimation {
+
+    public Instant lastAnimationTime;
+
+    public TestTranslateAnimation(
+        float fromXDelta, float toXDelta, float fromYDelta, float toYDelta) {
+      super(fromXDelta, toXDelta, fromYDelta, toYDelta);
+    }
+
+    @Override
+    public boolean getTransformation(
+        long currentTime, Transformation outTransformation, float scale) {
+      boolean result = super.getTransformation(currentTime, outTransformation, scale);
+      lastAnimationTime = Instant.ofEpochMilli(currentTime);
+      return result;
+    }
+  }
+}

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

@@ -1285,12 +1285,26 @@ protected boolean isSafeMode() {
 
   @Implementation
   protected Drawable getApplicationIcon(String packageName) throws NameNotFoundException {
-    return applicationIcons.get(packageName);
+    Drawable result = applicationIcons.get(packageName);
+    if (useValidGetApplicationIcon() && result == null) {
+      throw new NameNotFoundException(packageName);
+    }
+    return result;
+  }
+
+  private static boolean useValidGetApplicationIcon() {
+    return Boolean.parseBoolean(
+        System.getProperty("robolectric.useValidGetApplicationIcon", "false"));
   }
 
   @Implementation
   protected Drawable getApplicationIcon(ApplicationInfo info) throws NameNotFoundException {
-    return getApplicationIcon(info.packageName);
+    try {
+      return getApplicationIcon(info.packageName);
+    } catch (NameNotFoundException e) {
+      // This method should not return null. Return a default icon instead.
+      return getDefaultActivityIcon();
+    }
   }
 
   @Implementation

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

@@ -1,6 +1,7 @@
 package org.robolectric.shadows;
 
 import static android.os.Build.VERSION_CODES.M;
+import static android.os.Build.VERSION_CODES.P;
 
 import android.app.backup.BackupManager;
 import android.app.backup.BackupTransport;
@@ -14,6 +15,7 @@
 import android.os.IBinder;
 import android.os.Looper;
 import android.os.RemoteException;
+import android.os.UserHandle;
 import com.google.common.collect.ArrayListMultimap;
 import com.google.common.collect.ListMultimap;
 import java.util.ArrayList;
@@ -88,6 +90,17 @@ protected boolean isBackupEnabled() {
     return serviceState.backupEnabled;
   }
 
+  @Implementation(minSdk = P)
+  @HiddenApi // SystemApi
+  protected boolean isBackupServiceActive(UserHandle user) {
+    enforceBackupPermission("isBackupServiceActive");
+    return serviceState.backupServiceActive;
+  }
+
+  public void setBackupServiceActive(UserHandle user, boolean makeActive) {
+    serviceState.backupServiceActive = makeActive;
+  }
+
   @Implementation
   @HiddenApi // SystemApi
   protected RestoreSession beginRestoreSession() {
@@ -263,6 +276,7 @@ private void post(RemoteRunnable runnable) {
   }
 
   private static class BackupManagerServiceState {
+    boolean backupServiceActive = true;
     boolean backupEnabled = true;
     boolean nullRestoreData;
     long lastRestoreToken = 0L;

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

@@ -24,6 +24,7 @@ public class ShadowService extends ShadowContextWrapper {
   private boolean selfStopped = false;
   private boolean foregroundStopped;
   private boolean notificationShouldRemoved;
+  private Exception exceptionForStartForeground = null;
   private int stopSelfId;
   private int stopSelfResultId;
   private int foregroundServiceType;
@@ -55,7 +56,10 @@ protected boolean stopSelfResult(int id) {
   }
 
   @Implementation
-  protected void startForeground(int id, Notification notification) {
+  protected void startForeground(int id, Notification notification) throws Exception {
+    if (exceptionForStartForeground != null) {
+      throw exceptionForStartForeground;
+    }
     foregroundStopped = false;
     lastForegroundNotificationId = id;
     lastForegroundNotification = notification;
@@ -70,7 +74,8 @@ protected void startForeground(int id, Notification notification) {
 
   @Implementation(minSdk = Q)
   protected void startForeground(
-      int id, Notification notification, @ForegroundServiceType int foregroundServiceType) {
+      int id, Notification notification, @ForegroundServiceType int foregroundServiceType)
+      throws Exception {
     startForeground(id, notification);
     this.foregroundServiceType = foregroundServiceType;
   }
@@ -153,4 +158,22 @@ public int getStopSelfId() {
   public int getStopSelfResultId() {
     return stopSelfResultId;
   }
+
+  /**
+   * Configures the ShadowService so that calls to startForeground() will throw the given Exception.
+   * It can throw: {@link android.app.ForegroundServiceStartNotAllowedException}:
+   * https://developer.android.com/reference/android/app/ForegroundServiceStartNotAllowedException
+   * Or {@link android.app.InvalidForegroundServiceTypeException}:
+   * https://developer.android.com/reference/android/app/InvalidForegroundServiceTypeException Or
+   * {@link android.app.MissingForegroundServiceTypeException}:
+   * https://developer.android.com/reference/android/app/MissingForegroundServiceTypeException Or
+   * {@link java.lang.SecurityException}:
+   * https://developer.android.com/reference/java/lang/SecurityException
+   *
+   * <p>Details in:
+   * https://developer.android.com/reference/android/app/Service#startForeground(int,%20android.app.Notification)
+   */
+  public void setThrowInStartForeground(Exception e) {
+    exceptionForStartForeground = e;
+  }
 }

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

@@ -24,6 +24,7 @@
 import android.graphics.Paint;
 import android.graphics.Point;
 import android.os.IBinder;
+import android.os.SystemClock;
 import android.provider.Settings;
 import android.view.Display;
 import android.view.InputEvent;
@@ -47,6 +48,7 @@
 import org.robolectric.RuntimeEnvironment;
 import org.robolectric.annotation.Implementation;
 import org.robolectric.annotation.Implements;
+import org.robolectric.shadow.api.Shadow;
 import org.robolectric.util.ReflectionHelpers;
 
 /** Shadow for {@link UiAutomation}. */
@@ -123,8 +125,12 @@ protected Bitmap takeScreenshot() throws Exception {
                   Bitmap.createBitmap(displaySize.x, displaySize.y, Bitmap.Config.ARGB_8888);
               Canvas screenshotCanvas = new Canvas(screenshot);
               Paint paint = new Paint();
+              long drawingTime = SystemClock.uptimeMillis();
               for (Root root : getViewRoots().reverse()) {
                 View rootView = root.getRootView();
+                if (ShadowView.useRealViewAnimations()) {
+                  ((ShadowView) Shadow.extract(rootView)).setDrawingTime(drawingTime);
+                }
                 if (rootView.getWidth() <= 0 || rootView.getHeight() <= 0) {
                   continue;
                 }