Set the calling activity in ActivityScenario when run in Robolectric.

This change makes sure that when you use ActivityScenario.launchActivityForResult, your Activity under test will have getCallingActivity() and getCallingPackage() not null (set to the package of the test).

This is needed, if your Activity wants to make sure, that it is being called for result.

PiperOrigin-RevId: 582479442

integration_tests/androidx_test/build.gradle

@@ -10,6 +10,7 @@ android {
     namespace 'org.robolectric.integration.axt'
 
     defaultConfig {
+        testApplicationId 'org.robolectric.integrationtests.axt'
         minSdk 19
         targetSdk 34
         multiDexEnabled true

integration_tests/androidx_test/src/test/java/org/robolectric/integrationtests/axt/ActivityScenarioTest.java

@@ -291,4 +291,27 @@ public void onActivity_runsOnMainLooperThread() {
           });
     }
   }
+
+  @Test
+  public void getCallingActivity_empty() {
+    try (ActivityScenario<TranscriptActivity> activityScenario =
+        ActivityScenario.launch(TranscriptActivity.class)) {
+      activityScenario.onActivity(
+          activity -> {
+            assertThat(activity.getCallingActivity()).isNull();
+          });
+    }
+  }
+
+  @Test
+  public void getCallingActivity_isSet() {
+    try (ActivityScenario<TranscriptActivity> activityScenario =
+        ActivityScenario.launchActivityForResult(TranscriptActivity.class)) {
+      activityScenario.onActivity(
+          activity -> {
+            assertThat(activity.getCallingActivity().getPackageName())
+                .isEqualTo("org.robolectric.integrationtests.axt");
+          });
+    }
+  }
 }

robolectric/src/main/java/org/robolectric/android/internal/LocalActivityInvoker.java

@@ -45,6 +45,8 @@ public void startActivity(Intent intent) {
   public void startActivityForResult(Intent intent, @Nullable Bundle activityOptions) {
     isActivityLaunchedForResult = true;
     controller = getInstrumentation().startActivitySyncInternal(intent, activityOptions);
+    ShadowActivity shadowActivity = Shadow.extract(controller.get());
+    shadowActivity.setCallingPackage(getInstrumentation().getContext().getPackageName());
   }
 
   @Override