Add support for getSubscriptionId(phoneAccountHandle) in shadowTeleph…

…onyManager.

PiperOrigin-RevId: 538023398

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

@@ -1137,4 +1137,16 @@ public void getEmergencyNumbersList_wasSet_returnsCorrectList() throws Exception
         ImmutableMap.of(0, ImmutableList.of(emergencyNumber)));
     assertThat(telephonyManager.getEmergencyNumberList().get(0)).containsExactly(emergencyNumber);
   }
+
+  @Test
+  @Config(minSdk = R)
+  public void getSubscriptionIdForPhoneAccountHandle() {
+    int subscriptionId = 123;
+    PhoneAccountHandle phoneAccountHandle =
+        new PhoneAccountHandle(
+            new ComponentName(ApplicationProvider.getApplicationContext(), Object.class), "handle");
+    shadowOf(telephonyManager)
+        .setPhoneAccountHandleSubscriptionId(phoneAccountHandle, subscriptionId);
+    assertEquals(subscriptionId, telephonyManager.getSubscriptionId(phoneAccountHandle));
+  }
 }

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

@@ -95,6 +95,7 @@ public class ShadowTelephonyManager {
   private final Map<PhoneAccountHandle, Uri> voicemailRingtoneUriMap = new HashMap<>();
   private final Map<PhoneAccountHandle, TelephonyManager> phoneAccountToTelephonyManagers =
       new HashMap<>();
+  private final Map<PhoneAccountHandle, Integer> phoneAccountHandleSubscriptionId = new HashMap<>();
 
   private PhoneStateListener lastListener;
   private /*TelephonyCallback*/ Object lastTelephonyCallback;
@@ -1095,6 +1096,16 @@ public void setSubscriberId(String subscriberId) {
     this.subscriberId = subscriberId;
   }
 
+  @Implementation(minSdk = R)
+  protected int getSubscriptionId(PhoneAccountHandle handle) {
+    checkReadPhoneStatePermission();
+    return phoneAccountHandleSubscriptionId.get(handle);
+  }
+
+  public void setPhoneAccountHandleSubscriptionId(PhoneAccountHandle handle, int subscriptionId) {
+    phoneAccountHandleSubscriptionId.put(handle, subscriptionId);
+  }
+
   /** Returns the value set by {@link #setVisualVoicemailPackageName(String)}. */
   @Implementation(minSdk = O)
   protected String getVisualVoicemailPackageName() {