1 files changed, 16 insertions, 7 deletions
diff --git a/tests/unit/src/com/android/car/calendar/common/EventDescriptionsTest.java b/tests/unit/src/com/android/car/calendar/common/EventDescriptionsTest.java
index 358e9cf..bea029f 100644
--- a/tests/unit/src/com/android/car/calendar/common/EventDescriptionsTest.java
+++ b/tests/unit/src/com/android/car/calendar/common/EventDescriptionsTest.java
@@ -18,28 +18,36 @@ package com.android.car.calendar.common;
 
 import static com.google.common.truth.Truth.assertThat;
 
+import static org.mockito.Mockito.when;
+
 import android.net.Uri;
+import android.telephony.TelephonyManager;
 
 import com.google.common.collect.Iterables;
 
 import org.junit.Before;
 import org.junit.Test;
+import org.mockito.Mockito;
 
 import java.io.UnsupportedEncodingException;
 import java.util.List;
 import java.util.Locale;
 
 public class EventDescriptionsTest {
-
     private static final String BASE_NUMBER = "30 303986300";
     private static final String LOCAL_NUMBER = "0" + BASE_NUMBER;
     private static final String INTERNATIONAL_NUMBER = "+49 " + BASE_NUMBER;
     private static final String ACCESS = ",,12;3*45#";
+    private static final String COUNTRY_ISO_CODE = "DE";
+
     private EventDescriptions mEventDescriptions;
+    private TelephonyManager mMockTelephonyManager;
 
     @Before
     public void setUp() {
-        mEventDescriptions = new EventDescriptions(Locale.GERMANY);
+        mMockTelephonyManager = Mockito.mock(TelephonyManager.class);
+        when(mMockTelephonyManager.getNetworkCountryIso()).thenReturn(COUNTRY_ISO_CODE);
+        mEventDescriptions = new EventDescriptions(Locale.GERMANY, mMockTelephonyManager);
     }
 
     @Test
@@ -48,7 +56,7 @@ public class EventDescriptionsTest {
                 mEventDescriptions.extractNumberAndPins(LOCAL_NUMBER);
         assertThat(numberAndAccesses).isNotEmpty();
         Dialer.NumberAndAccess numberAndAccess = Iterables.getFirst(numberAndAccesses, null);
-        assertThat(numberAndAccess.getNumber()).isEqualTo(INTERNATIONAL_NUMBER);
+        assertThat(numberAndAccess.getNumber()).isEqualTo(LOCAL_NUMBER);
     }
 
     @Test
@@ -62,9 +70,10 @@ public class EventDescriptionsTest {
 
     @Test
     public void extractNumberAndPin_internationalNumberWithDifferentLocale_resultIsInternational() {
-        mEventDescriptions = new EventDescriptions(Locale.FRANCE);
+        EventDescriptions eventDescriptions =
+                new EventDescriptions(Locale.FRANCE, mMockTelephonyManager);
         List<Dialer.NumberAndAccess> numberAndAccesses =
-                mEventDescriptions.extractNumberAndPins(INTERNATIONAL_NUMBER);
+                eventDescriptions.extractNumberAndPins(INTERNATIONAL_NUMBER);
         assertThat(numberAndAccesses).isNotEmpty();
         Dialer.NumberAndAccess numberAndAccess = Iterables.getFirst(numberAndAccesses, null);
         assertThat(numberAndAccess.getNumber()).isEqualTo(INTERNATIONAL_NUMBER);
@@ -105,8 +114,8 @@ public class EventDescriptionsTest {
         List<Dialer.NumberAndAccess> numberAndAccesses =
                 mEventDescriptions.extractNumberAndPins(input);
 
-        // The local number is valid but repeated so only included once.
-        assertThat(numberAndAccesses).hasSize(1);
+        // Keep all variations of a base number.
+        assertThat(numberAndAccesses).hasSize(3);
     }
 
     @Test