Supports EuiccManager#getEid in ShadowEuiccManager

PiperOrigin-RevId: 236706886
robolectric · Mar 4, 2019 · e64f675 · e64f675
1 parent 2f7aa83
commit e64f675
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 7 deletions.
diff --git a/robolectric/src/test/java/org/robolectric/shadows/ShadowEuiccManagerTest.java b/robolectric/src/test/java/org/robolectric/shadows/ShadowEuiccManagerTest.java
@@ -2,40 +2,46 @@
 
 import static android.os.Build.VERSION_CODES.P;
 import static com.google.common.truth.Truth.assertThat;
-import static org.robolectric.RuntimeEnvironment.application;
+import static org.robolectric.Shadows.shadowOf;
 
 import android.telephony.euicc.EuiccManager;
+import androidx.test.core.app.ApplicationProvider;
 import androidx.test.ext.junit.runners.AndroidJUnit4;
 import org.junit.Before;
 import org.junit.Test;
 import org.junit.runner.RunWith;
-import org.robolectric.Shadows;
 import org.robolectric.annotation.Config;
 
 /** Junit test for {@link ShadowEuiccManager}. */
 @RunWith(AndroidJUnit4.class)
 @Config(minSdk = P)
 public class ShadowEuiccManagerTest {
   private EuiccManager euiccManager;
-  private ShadowEuiccManager shadowEuiccManager;
 
   @Before
   public void setUp() {
-    euiccManager = application.getSystemService(EuiccManager.class);
-    shadowEuiccManager = Shadows.shadowOf(euiccManager);
+    euiccManager = ApplicationProvider.getApplicationContext().getSystemService(EuiccManager.class);
   }
 
   @Test
   public void isEnabled() {
-    shadowEuiccManager.setIsEnabled(true);
+    shadowOf(euiccManager).setIsEnabled(true);
 
     assertThat(euiccManager.isEnabled()).isTrue();
   }
 
   @Test
   public void isEnabled_whenSetToFalse() {
-    shadowEuiccManager.setIsEnabled(false);
+    shadowOf(euiccManager).setIsEnabled(false);
 
     assertThat(euiccManager.isEnabled()).isFalse();
   }
+
+  @Test
+  public void getEid() {
+    String eid = "testEid";
+    shadowOf(euiccManager).setEid(eid);
+
+    assertThat(euiccManager.getEid()).isEqualTo(eid);
+  }
 }
diff --git a/shadows/framework/src/main/java/org/robolectric/shadows/ShadowEuiccManager.java b/shadows/framework/src/main/java/org/robolectric/shadows/ShadowEuiccManager.java
@@ -10,6 +10,7 @@
 public class ShadowEuiccManager {
 
   private boolean enabled;
+  private String eid;
 
   /** Returns {@code false}, or the value specified by calling {@link #setIsEnabled}. */
   @Implementation
@@ -21,4 +22,14 @@ protected boolean isEnabled() {
   public void setIsEnabled(boolean isEnabled) {
     enabled = isEnabled;
   }
+
+  @Implementation
+  protected String getEid() {
+    return eid;
+  }
+
+  /** Set the value to be returned by {@link EuiccManager#getEid}. */
+  public void setEid(String eid) {
+    this.eid = eid;
+  }
 }