Dynamic preferences for IMEI

 - Support N-sim devices for imei preference

Bug: 36458278
Test: make RunSettingsRoboTests -j40
Change-Id: I0dda78b44e0c64ddd69c9a30e90c49f03b2260f3
diff --git a/res/values/strings.xml b/res/values/strings.xml
index 4582bc4..675bfcd 100644
--- a/res/values/strings.xml
+++ b/res/values/strings.xml
@@ -2664,9 +2664,7 @@
     <!-- [CHAR LIMIT=100] Main settings screen item's summary for the SD card and storage settings -->
     <string name="storage_settings_summary" product="default">Unmount SD card, view available storage</string>
     <!-- About phone screen, title for IMEI for multi-sim devices -->
-    <string name="imei_multi_sim_slot_1">IMEI (sim slot 1)</string>
-    <!-- About phone screen, title for IMEI for multi-sim devices -->
-    <string name="imei_multi_sim_slot_2">IMEI (sim slot 2)</string>
+    <string name="imei_multi_sim">IMEI (sim slot %1$d)</string>
     <!-- Do not translate. About phone, status item title -->
     <string name="status_imei">IMEI</string>
     <!-- Do not translate. About phone, status item title -->
@@ -2688,9 +2686,7 @@
     <!-- About phone, status item title.  The phone PRL Version of the current device.-->
     <string name="status_prl_version">PRL version</string>
     <!-- About phone screen, title for MEID for multi-sim devices -->
-    <string name="meid_multi_sim_sim_slot_1">MEID (sim slot 1)</string>
-    <!-- About phone screen, title for MEID for multi-sim devices -->
-    <string name="meid_multi_sim_sim_slot_2">MEID (sim slot 2)</string>
+    <string name="meid_multi_sim">MEID (sim slot %1$d)</string>
     <!-- About phone, status item title.  The phone MEID number of the current LTE/CDMA device. [CHAR LIMIT=30] -->
     <string name="status_meid_number">MEID</string>
     <!-- About phone, status item title.  The ICCID of the current LTE device. [CHAR LIMIT=30] -->
diff --git a/res/xml/device_info_settings_v2.xml b/res/xml/device_info_settings_v2.xml
index 5957c1a..77ead2c 100644
--- a/res/xml/device_info_settings_v2.xml
+++ b/res/xml/device_info_settings_v2.xml
@@ -25,18 +25,21 @@
     <!-- Phone number -->
     <Preference
         android:key="phone_number"
+        android:order="0"
         android:title="@string/status_number"
         android:summary="@string/summary_placeholder"/>
 
     <!-- SIM status Sim Slot 1 -->
     <Preference
         android:key="sim_status_sim_1"
+        android:order="10"
         android:title="@string/sim_status_title"
         android:summary="@string/summary_placeholder"/>
 
     <!-- SIM status Sim Slot 2-->
     <Preference
         android:key="sim_status_sim_2"
+        android:order="11"
         android:title="@string/sim_status_title_sim_slot_2"
         android:summary="@string/summary_placeholder"/>
 
@@ -44,32 +47,29 @@
     <!-- Model & hardware -->
     <Preference
         android:key="device_model"
+        android:order="21"
         android:title="@string/hardware_info"
         android:summary="@string/summary_placeholder"/>
 
-    <!-- IMEI Sim Slot 1 -->
+    <!-- IMEI -->
     <Preference
-        android:key="imei_info_sim_slot_1"
+        android:key="imei_info"
+        android:order="22"
         android:title="@string/status_imei"
         settings:keywords="@string/keywords_imei_info"
         android:summary="@string/summary_placeholder"/>
 
-    <!-- IMEI Sim Slot 2 -->
-    <Preference
-        android:key="imei_info_sim_slot_2"
-        android:title="@string/imei_multi_sim_slot_2"
-        settings:keywords="@string/keywords_imei_info"
-        android:summary="@string/summary_placeholder"/>
-
     <!-- Android version -->
     <Preference
         android:key="firmware_version"
+        android:order="32"
         android:title="@string/firmware_version"
         android:summary="@string/summary_placeholder"/>
 
     <!--IP address -->
     <Preference
         android:key="wifi_ip_address"
+        android:order="33"
         android:title="@string/wifi_ip_address"
         android:summary="@string/summary_placeholder"
         settings:allowDividerAbove="true"/>
@@ -77,12 +77,14 @@
     <!-- Wi-Fi MAC address -->
     <Preference
         android:key="wifi_mac_address"
+        android:order="34"
         android:title="@string/status_wifi_mac_address"
         android:summary="@string/summary_placeholder"/>
 
     <!-- Bluetooth address -->
     <Preference
         android:key="bt_address"
+        android:order="35"
         android:title="@string/status_bt_address"
         android:summary="@string/summary_placeholder"/>
 
@@ -90,6 +92,7 @@
     <!-- Legal information -->
     <Preference
         android:key="legal_container"
+        android:order="36"
         android:title="@string/legal_information"
         android:fragment="com.android.settings.LegalSettings"
         settings:allowDividerAbove="true"/>
@@ -97,6 +100,7 @@
     <!-- Regulatory labels -->
     <Preference
         android:key="regulatory_info"
+        android:order="37"
         android:title="@string/regulatory_labels">
         <intent android:action="android.settings.SHOW_REGULATORY_INFO"/>
     </Preference>
@@ -104,6 +108,7 @@
     <!-- Safety & regulatory manual -->
     <Preference
         android:key="safety_info"
+        android:order="38"
         android:title="@string/safety_and_regulatory_info">
         <intent android:action="android.settings.SHOW_SAFETY_AND_REGULATORY_INFO"/>
     </Preference>
@@ -111,6 +116,7 @@
     <!-- Build number -->
     <Preference
         android:key="build_number"
+        android:order="39"
         android:title="@string/build_number"
         android:summary="@string/summary_placeholder"
         settings:allowDividerAbove="true"/>
diff --git a/src/com/android/settings/DeviceInfoSettings.java b/src/com/android/settings/DeviceInfoSettings.java
index 5c26645..f4939f6 100644
--- a/src/com/android/settings/DeviceInfoSettings.java
+++ b/src/com/android/settings/DeviceInfoSettings.java
@@ -34,17 +34,16 @@
 import com.android.settings.deviceinfo.FeedbackPreferenceController;
 import com.android.settings.deviceinfo.FirmwareVersionPreferenceController;
 import com.android.settings.deviceinfo.IpAddressPreferenceController;
-import com.android.settings.deviceinfo.WifiMacAddressPreferenceController;
-import com.android.settings.deviceinfo.imei.ImeiInfoDualSimPreferenceController;
-import com.android.settings.deviceinfo.imei.ImeiInfoPreferenceControllerV2;
 import com.android.settings.deviceinfo.KernelVersionPreferenceController;
 import com.android.settings.deviceinfo.ManualPreferenceController;
 import com.android.settings.deviceinfo.RegulatoryInfoPreferenceController;
 import com.android.settings.deviceinfo.SafetyInfoPreferenceController;
 import com.android.settings.deviceinfo.SecurityPatchPreferenceController;
+import com.android.settings.deviceinfo.WifiMacAddressPreferenceController;
+import com.android.settings.deviceinfo.firmwareversion.FirmwareVersionPreferenceControllerV2;
+import com.android.settings.deviceinfo.imei.ImeiInfoPreferenceControllerV2;
 import com.android.settings.deviceinfo.simstatus.SimStatusDualSimPreferenceController;
 import com.android.settings.deviceinfo.simstatus.SimStatusPreferenceControllerV2;
-import com.android.settings.deviceinfo.firmwareversion.FirmwareVersionPreferenceControllerV2;
 import com.android.settings.search.BaseSearchIndexProvider;
 import com.android.settings.search.Indexable;
 import com.android.settingslib.core.AbstractPreferenceController;
@@ -140,8 +139,6 @@
 
             controllers.add(new ImeiInfoPreferenceControllerV2(context, fragment));
 
-            controllers.add(new ImeiInfoDualSimPreferenceController(context, fragment));
-
             controllers.add(new FirmwareVersionPreferenceControllerV2(context, fragment));
 
             controllers.add(new IpAddressPreferenceController(context, lifecycle));
diff --git a/src/com/android/settings/deviceinfo/imei/AbstractImeiInfoPreferenceController.java b/src/com/android/settings/deviceinfo/imei/AbstractImeiInfoPreferenceController.java
deleted file mode 100644
index a9871a0..0000000
--- a/src/com/android/settings/deviceinfo/imei/AbstractImeiInfoPreferenceController.java
+++ /dev/null
@@ -1,99 +0,0 @@
-/*
- * Copyright (C) 2017 The Android Open Source Project
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- *      http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-
-package com.android.settings.deviceinfo.imei;
-
-import static android.telephony.TelephonyManager.PHONE_TYPE_CDMA;
-
-import android.app.Fragment;
-import android.content.Context;
-import android.support.annotation.VisibleForTesting;
-import android.support.v7.preference.Preference;
-import android.support.v7.preference.PreferenceScreen;
-import android.telephony.TelephonyManager;
-import android.text.TextUtils;
-
-import com.android.settings.core.PreferenceControllerMixin;
-import com.android.settingslib.deviceinfo.AbstractSimStatusImeiInfoPreferenceController;
-
-/**
- * Controller that manages preference for single and dual sim devices.
- */
-public abstract class AbstractImeiInfoPreferenceController extends
-        AbstractSimStatusImeiInfoPreferenceController implements PreferenceControllerMixin {
-
-    protected final boolean mIsMultiSim;
-    protected final TelephonyManager mTelephonyManager;
-
-    private Preference mPreference;
-    private Fragment mFragment;
-
-    public AbstractImeiInfoPreferenceController(Context context, Fragment fragment) {
-        super(context);
-
-        mFragment = fragment;
-        mTelephonyManager = (TelephonyManager) context.getSystemService(Context.TELEPHONY_SERVICE);
-        mIsMultiSim = mTelephonyManager.getPhoneCount() > 1;
-    }
-
-    @Override
-    public void displayPreference(PreferenceScreen screen) {
-        super.displayPreference(screen);
-        mPreference = screen.findPreference(getPreferenceKey());
-        if (mPreference == null) {
-            return;
-        }
-        final int phoneType = mTelephonyManager.getPhoneType();
-        if (phoneType == PHONE_TYPE_CDMA) {
-            mPreference.setTitle(getTitleForCdmaPhone());
-            mPreference.setSummary(getMeid());
-        } else {
-            // GSM phone
-            mPreference.setTitle(getTitleForGsmPhone());
-            mPreference.setSummary(mTelephonyManager.getImei(getSimSlot()));
-        }
-    }
-
-    @Override
-    public boolean handlePreferenceTreeClick(Preference preference) {
-        if (!TextUtils.equals(preference.getKey(), getPreferenceKey())) {
-            return false;
-        }
-
-        ImeiInfoDialogFragment.show(mFragment, getSimSlot(), mPreference.getTitle().toString());
-        return true;
-    }
-
-    /**
-     * @return The preference title for phones based on CDMA technology.
-     */
-    protected abstract String getTitleForCdmaPhone();
-
-    /**
-     * @return The preference title for phones based on GSM technology.
-     */
-    protected abstract String getTitleForGsmPhone();
-
-    /**
-     * @return The sim slot to retrieve IMEI/CDMA information about.
-     */
-    protected abstract int getSimSlot();
-
-    @VisibleForTesting
-    String getMeid() {
-        return mTelephonyManager.getMeid(getSimSlot());
-    }
-}
diff --git a/src/com/android/settings/deviceinfo/imei/ImeiInfoDualSimPreferenceController.java b/src/com/android/settings/deviceinfo/imei/ImeiInfoDualSimPreferenceController.java
deleted file mode 100644
index 61d99b9..0000000
--- a/src/com/android/settings/deviceinfo/imei/ImeiInfoDualSimPreferenceController.java
+++ /dev/null
@@ -1,57 +0,0 @@
-/*
- * Copyright (C) 2017 The Android Open Source Project
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- *      http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-
-package com.android.settings.deviceinfo.imei;
-
-import android.app.Fragment;
-import android.content.Context;
-
-import com.android.settings.R;
-
-public class ImeiInfoDualSimPreferenceController extends AbstractImeiInfoPreferenceController {
-
-    private static final String KEY_IMEI_INFO_DUAL_SIM = "imei_info_sim_slot_2";
-    private static final int SIM_SLOT = 1;
-
-    public ImeiInfoDualSimPreferenceController(Context context, Fragment fragment) {
-        super(context, fragment);
-    }
-
-    @Override
-    public boolean isAvailable() {
-        return super.isAvailable() && mIsMultiSim;
-    }
-
-    @Override
-    public String getPreferenceKey() {
-        return KEY_IMEI_INFO_DUAL_SIM;
-    }
-
-    @Override
-    protected String getTitleForCdmaPhone() {
-        return mContext.getResources().getString(R.string.meid_multi_sim_sim_slot_2);
-    }
-
-    @Override
-    protected String getTitleForGsmPhone() {
-        return mContext.getResources().getString(R.string.imei_multi_sim_slot_2);
-    }
-
-    @Override
-    protected int getSimSlot() {
-        return SIM_SLOT;
-    }
-}
diff --git a/src/com/android/settings/deviceinfo/imei/ImeiInfoPreferenceControllerV2.java b/src/com/android/settings/deviceinfo/imei/ImeiInfoPreferenceControllerV2.java
index 74448bb..dcc6db6 100644
--- a/src/com/android/settings/deviceinfo/imei/ImeiInfoPreferenceControllerV2.java
+++ b/src/com/android/settings/deviceinfo/imei/ImeiInfoPreferenceControllerV2.java
@@ -16,19 +16,41 @@
 
 package com.android.settings.deviceinfo.imei;
 
+import static android.telephony.TelephonyManager.PHONE_TYPE_CDMA;
+
 import android.app.Fragment;
 import android.content.Context;
+import android.support.annotation.VisibleForTesting;
+import android.support.v7.preference.Preference;
+import android.support.v7.preference.PreferenceScreen;
+import android.telephony.TelephonyManager;
 
 import com.android.settings.R;
+import com.android.settings.core.PreferenceControllerMixin;
+import com.android.settingslib.deviceinfo.AbstractSimStatusImeiInfoPreferenceController;
 
-public class ImeiInfoPreferenceControllerV2 extends AbstractImeiInfoPreferenceController {
+import java.util.ArrayList;
+import java.util.List;
 
-    public static final int SIM_SLOT = 0;
+/**
+ * Controller that manages preference for single and multi sim devices.
+ */
+public class ImeiInfoPreferenceControllerV2 extends
+        AbstractSimStatusImeiInfoPreferenceController implements PreferenceControllerMixin {
 
-    private static final String KEY_IMEI_INFO = "imei_info_sim_slot_1";
+    private static final String KEY_IMEI_INFO = "imei_info";
+
+    private final boolean mIsMultiSim;
+    private final TelephonyManager mTelephonyManager;
+    private final List<Preference> mPreferenceList = new ArrayList<>();
+    private final Fragment mFragment;
 
     public ImeiInfoPreferenceControllerV2(Context context, Fragment fragment) {
-        super(context, fragment);
+        super(context);
+
+        mFragment = fragment;
+        mTelephonyManager = (TelephonyManager) context.getSystemService(Context.TELEPHONY_SERVICE);
+        mIsMultiSim = mTelephonyManager.getPhoneCount() > 1;
     }
 
     @Override
@@ -37,19 +59,69 @@
     }
 
     @Override
-    protected String getTitleForCdmaPhone() {
-        return mIsMultiSim ? mContext.getResources().getString(R.string.meid_multi_sim_sim_slot_1)
-                : mContext.getResources().getString(R.string.status_meid_number);
+    public void displayPreference(PreferenceScreen screen) {
+        super.displayPreference(screen);
+        final Preference preference = screen.findPreference(getPreferenceKey());
+        if (!isAvailable() || preference == null || !preference.isVisible()) {
+            return;
+        }
+
+        mPreferenceList.add(preference);
+        updatePreference(preference, 0 /* sim slot */);
+
+        final int imeiPreferenceOrder = preference.getOrder();
+        // Add additional preferences for each sim in the device
+        for (int simSlotNumber = 1; simSlotNumber < mTelephonyManager.getPhoneCount();
+                simSlotNumber++) {
+            final Preference multiSimPreference = createNewPreference(screen.getContext());
+            multiSimPreference.setOrder(imeiPreferenceOrder + simSlotNumber);
+            multiSimPreference.setKey(KEY_IMEI_INFO + simSlotNumber);
+            screen.addPreference(multiSimPreference);
+            mPreferenceList.add(multiSimPreference);
+            updatePreference(multiSimPreference, simSlotNumber);
+        }
     }
 
     @Override
-    protected String getTitleForGsmPhone() {
-        return mIsMultiSim ? mContext.getResources().getString(R.string.imei_multi_sim_slot_1)
-                : mContext.getResources().getString(R.string.status_imei);
+    public boolean handlePreferenceTreeClick(Preference preference) {
+        final int simSlot = mPreferenceList.indexOf(preference);
+        if (simSlot == -1) {
+            return false;
+        }
+
+        ImeiInfoDialogFragment.show(mFragment, simSlot, preference.getTitle().toString());
+        return true;
     }
 
-    @Override
-    protected int getSimSlot() {
-        return SIM_SLOT;
+    private void updatePreference(Preference preference, int simSlot) {
+        final int phoneType = mTelephonyManager.getPhoneType();
+        if (phoneType == PHONE_TYPE_CDMA) {
+            preference.setTitle(getTitleForCdmaPhone(simSlot));
+            preference.setSummary(getMeid(simSlot));
+        } else {
+            // GSM phone
+            preference.setTitle(getTitleForGsmPhone(simSlot));
+            preference.setSummary(mTelephonyManager.getImei(simSlot));
+        }
+    }
+
+    private CharSequence getTitleForGsmPhone(int simSlot) {
+        return mIsMultiSim ? mContext.getString(R.string.imei_multi_sim, simSlot + 1)
+                : mContext.getString(R.string.status_imei);
+    }
+
+    private CharSequence getTitleForCdmaPhone(int simSlot) {
+        return mIsMultiSim ? mContext.getString(R.string.meid_multi_sim, simSlot + 1)
+                : mContext.getString(R.string.status_meid_number);
+    }
+
+    @VisibleForTesting
+    String getMeid(int simSlot) {
+        return mTelephonyManager.getMeid(simSlot);
+    }
+
+    @VisibleForTesting
+    Preference createNewPreference(Context context) {
+        return new Preference(context);
     }
 }
diff --git a/tests/robotests/src/com/android/settings/deviceinfo/imei/AbstractImeiInfoPreferenceControllerTest.java b/tests/robotests/src/com/android/settings/deviceinfo/imei/AbstractImeiInfoPreferenceControllerTest.java
deleted file mode 100644
index 51e6005..0000000
--- a/tests/robotests/src/com/android/settings/deviceinfo/imei/AbstractImeiInfoPreferenceControllerTest.java
+++ /dev/null
@@ -1,143 +0,0 @@
-/*
- * Copyright (C) 2017 The Android Open Source Project
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- *      http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-
-package com.android.settings.deviceinfo.imei;
-
-import static android.telephony.TelephonyManager.PHONE_TYPE_CDMA;
-import static android.telephony.TelephonyManager.PHONE_TYPE_GSM;
-
-import static org.mockito.ArgumentMatchers.anyInt;
-import static org.mockito.Mockito.doReturn;
-import static org.mockito.Mockito.mock;
-import static org.mockito.Mockito.spy;
-import static org.mockito.Mockito.verify;
-import static org.mockito.Mockito.when;
-
-import android.app.Fragment;
-import android.app.FragmentManager;
-import android.content.Context;
-import android.os.UserManager;
-import android.support.v7.preference.Preference;
-import android.support.v7.preference.PreferenceScreen;
-import android.telephony.TelephonyManager;
-
-import com.android.settings.TestConfig;
-import com.android.settings.testutils.SettingsRobolectricTestRunner;
-
-import org.junit.Before;
-import org.junit.Test;
-import org.junit.runner.RunWith;
-import org.mockito.Answers;
-import org.mockito.Mock;
-import org.mockito.MockitoAnnotations;
-import org.robolectric.RuntimeEnvironment;
-import org.robolectric.annotation.Config;
-import org.robolectric.util.ReflectionHelpers;
-
-@RunWith(SettingsRobolectricTestRunner.class)
-@Config(manifest = TestConfig.MANIFEST_PATH, sdk = TestConfig.SDK_VERSION)
-public class AbstractImeiInfoPreferenceControllerTest {
-
-    @Mock
-    private Preference mPreference;
-    @Mock
-    private PreferenceScreen mScreen;
-    @Mock
-    private TelephonyManager mTelephonyManager;
-    @Mock
-    private UserManager mUserManager;
-    @Mock
-    private Fragment mFragment;
-
-    private Context mContext;
-    private AbstractImeiInfoPreferenceControllerImpl mController;
-
-    @Before
-    public void setUp() {
-        MockitoAnnotations.initMocks(this);
-        mContext = spy(RuntimeEnvironment.application);
-        doReturn(mUserManager).when(mContext).getSystemService(UserManager.class);
-        mController = new AbstractImeiInfoPreferenceControllerImpl(mContext, mFragment);
-        ReflectionHelpers.setField(mController, "mTelephonyManager", mTelephonyManager);
-        when(mScreen.findPreference(mController.getPreferenceKey())).thenReturn(mPreference);
-        when(mPreference.getKey()).thenReturn(mController.getPreferenceKey());
-    }
-
-    @Test
-    public void displayPreference_cdmaPhone_shouldSetCdmaTitleAndMeid() {
-        mController = spy(mController);
-        final String meid = "125132215123";
-        when(mTelephonyManager.getPhoneType()).thenReturn(PHONE_TYPE_CDMA);
-        doReturn(meid).when(mController).getMeid();
-
-        mController.displayPreference(mScreen);
-
-        verify(mPreference).setTitle(mController.getTitleForCdmaPhone());
-        verify(mPreference).setSummary(meid);
-    }
-
-    @Test
-    public void displayPreference_gsmPhone_shouldSetGsmTitleAndImei() {
-        final String imei = "125132215123";
-        when(mTelephonyManager.getPhoneType()).thenReturn(PHONE_TYPE_GSM);
-        when(mTelephonyManager.getImei(anyInt())).thenReturn(imei);
-
-        mController.displayPreference(mScreen);
-
-        verify(mPreference).setTitle(mController.getTitleForGsmPhone());
-        verify(mPreference).setSummary(imei);
-    }
-
-    @Test
-    public void handlePreferenceTreeClick_shouldStartDialogFragment() {
-        when(mFragment.getChildFragmentManager()).thenReturn(
-                mock(FragmentManager.class, Answers.RETURNS_DEEP_STUBS));
-        when(mPreference.getTitle()).thenReturn("SomeTitle");
-        mController.displayPreference(mScreen);
-
-        mController.handlePreferenceTreeClick(mPreference);
-
-        verify(mFragment).getChildFragmentManager();
-    }
-
-    public class AbstractImeiInfoPreferenceControllerImpl extends
-            AbstractImeiInfoPreferenceController {
-
-        public AbstractImeiInfoPreferenceControllerImpl(Context context, Fragment fragment) {
-            super(context, fragment);
-        }
-
-        @Override
-        public String getPreferenceKey() {
-            return "foobar";
-        }
-
-        @Override
-        protected String getTitleForCdmaPhone() {
-            return "foo";
-        }
-
-        @Override
-        protected String getTitleForGsmPhone() {
-            return "bar";
-        }
-
-        @Override
-        protected int getSimSlot() {
-            return 0;
-        }
-    }
-}
diff --git a/tests/robotests/src/com/android/settings/deviceinfo/imei/ImeiInfoDualSimPreferenceControllerTest.java b/tests/robotests/src/com/android/settings/deviceinfo/imei/ImeiInfoDualSimPreferenceControllerTest.java
deleted file mode 100644
index ce99ee4..0000000
--- a/tests/robotests/src/com/android/settings/deviceinfo/imei/ImeiInfoDualSimPreferenceControllerTest.java
+++ /dev/null
@@ -1,89 +0,0 @@
-/*
- * Copyright (C) 2017 The Android Open Source Project
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- *      http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-
-package com.android.settings.deviceinfo.imei;
-
-import static com.google.common.truth.Truth.assertThat;
-
-import static org.mockito.Mockito.doReturn;
-import static org.mockito.Mockito.spy;
-import static org.mockito.Mockito.when;
-
-import android.app.Fragment;
-import android.content.Context;
-import android.os.UserManager;
-import android.support.v7.preference.Preference;
-import android.support.v7.preference.PreferenceScreen;
-import android.telephony.TelephonyManager;
-
-import com.android.settings.R;
-import com.android.settings.TestConfig;
-import com.android.settings.testutils.SettingsRobolectricTestRunner;
-
-import org.junit.Before;
-import org.junit.Test;
-import org.junit.runner.RunWith;
-import org.mockito.Mock;
-import org.mockito.MockitoAnnotations;
-import org.robolectric.RuntimeEnvironment;
-import org.robolectric.annotation.Config;
-import org.robolectric.util.ReflectionHelpers;
-
-@RunWith(SettingsRobolectricTestRunner.class)
-@Config(manifest = TestConfig.MANIFEST_PATH, sdk = TestConfig.SDK_VERSION)
-public class ImeiInfoDualSimPreferenceControllerTest {
-
-    @Mock
-    private Preference mPreference;
-    @Mock
-    private PreferenceScreen mScreen;
-    @Mock
-    private TelephonyManager mTelephonyManager;
-    @Mock
-    private UserManager mUserManager;
-    @Mock
-    private Fragment mFragment;
-
-    private Context mContext;
-    private ImeiInfoDualSimPreferenceController mController;
-
-    @Before
-    public void setUp() {
-        MockitoAnnotations.initMocks(this);
-        mContext = spy(RuntimeEnvironment.application);
-        doReturn(mUserManager).when(mContext).getSystemService(UserManager.class);
-        mController = new ImeiInfoDualSimPreferenceController(mContext, mFragment);
-        ReflectionHelpers.setField(mController, "mTelephonyManager", mTelephonyManager);
-        when(mScreen.findPreference(mController.getPreferenceKey())).thenReturn(mPreference);
-        when(mPreference.getKey()).thenReturn(mController.getPreferenceKey());
-    }
-
-    @Test
-    public void getTitleForCdmaPhone_shouldReturnCdmaMultiSimString() {
-        ReflectionHelpers.setField(mController, "mIsMultiSim", true);
-
-        assertThat(mController.getTitleForCdmaPhone()).isEqualTo(mContext.getResources().getString(
-                R.string.meid_multi_sim_sim_slot_2));
-    }
-
-    @Test
-    public void getTitleForGsmPhone_shouldReturnGsmMultiSimString() {
-        ReflectionHelpers.setField(mController, "mIsMultiSim", true);
-
-        assertThat(mController.getTitleForGsmPhone()).isEqualTo(mContext.getResources().getString(
-                R.string.imei_multi_sim_slot_2));
-    }
-}
diff --git a/tests/robotests/src/com/android/settings/deviceinfo/imei/ImeiInfoPreferenceControllerV2Test.java b/tests/robotests/src/com/android/settings/deviceinfo/imei/ImeiInfoPreferenceControllerV2Test.java
index 233e27e..3fed589 100644
--- a/tests/robotests/src/com/android/settings/deviceinfo/imei/ImeiInfoPreferenceControllerV2Test.java
+++ b/tests/robotests/src/com/android/settings/deviceinfo/imei/ImeiInfoPreferenceControllerV2Test.java
@@ -16,13 +16,19 @@
 
 package com.android.settings.deviceinfo.imei;
 
-import static com.google.common.truth.Truth.assertThat;
+import static android.telephony.TelephonyManager.PHONE_TYPE_CDMA;
+import static android.telephony.TelephonyManager.PHONE_TYPE_GSM;
 
+import static org.mockito.ArgumentMatchers.anyInt;
 import static org.mockito.Mockito.doReturn;
+import static org.mockito.Mockito.mock;
 import static org.mockito.Mockito.spy;
+import static org.mockito.Mockito.times;
+import static org.mockito.Mockito.verify;
 import static org.mockito.Mockito.when;
 
 import android.app.Fragment;
+import android.app.FragmentManager;
 import android.content.Context;
 import android.os.UserManager;
 import android.support.v7.preference.Preference;
@@ -36,6 +42,7 @@
 import org.junit.Before;
 import org.junit.Test;
 import org.junit.runner.RunWith;
+import org.mockito.Answers;
 import org.mockito.Mock;
 import org.mockito.MockitoAnnotations;
 import org.robolectric.RuntimeEnvironment;
@@ -49,6 +56,8 @@
     @Mock
     private Preference mPreference;
     @Mock
+    private Preference mSecondSimPreference;
+    @Mock
     private PreferenceScreen mScreen;
     @Mock
     private TelephonyManager mTelephonyManager;
@@ -65,41 +74,97 @@
         MockitoAnnotations.initMocks(this);
         mContext = spy(RuntimeEnvironment.application);
         doReturn(mUserManager).when(mContext).getSystemService(UserManager.class);
-        mController = new ImeiInfoPreferenceControllerV2(mContext, mFragment);
+        mController = spy(new ImeiInfoPreferenceControllerV2(mContext, mFragment));
+        doReturn(true).when(mController).isAvailable();
+        when(mScreen.getContext()).thenReturn(mContext);
+        doReturn(mSecondSimPreference).when(mController).createNewPreference(mContext);
         ReflectionHelpers.setField(mController, "mTelephonyManager", mTelephonyManager);
         when(mScreen.findPreference(mController.getPreferenceKey())).thenReturn(mPreference);
-        when(mPreference.getKey()).thenReturn(mController.getPreferenceKey());
+        final String prefKey = mController.getPreferenceKey();
+        when(mPreference.getKey()).thenReturn(prefKey);
+        when(mPreference.isVisible()).thenReturn(true);
     }
 
     @Test
-    public void getTitleForCdmaPhone_noMultiSim_shouldReturnCdmaNoMultiSimString() {
-        ReflectionHelpers.setField(mController, "mIsMultiSim", false);
-
-        assertThat(mController.getTitleForCdmaPhone()).isEqualTo(mContext.getResources().getString(
-                R.string.status_meid_number));
-    }
-
-    @Test
-    public void getTitleForCdmaPhone_multiSim_shouldReturnCdmaMultiSimString() {
+    public void displayPreference_multiSimGsm_shouldAddSecondPreference() {
         ReflectionHelpers.setField(mController, "mIsMultiSim", true);
+        when(mTelephonyManager.getPhoneCount()).thenReturn(2);
+        when(mTelephonyManager.getPhoneType()).thenReturn(PHONE_TYPE_GSM);
 
-        assertThat(mController.getTitleForCdmaPhone()).isEqualTo(mContext.getResources().getString(
-                R.string.meid_multi_sim_sim_slot_1));
+        mController.displayPreference(mScreen);
+
+        verify(mScreen).addPreference(mSecondSimPreference);
     }
 
     @Test
-    public void getTitleForGsmPhone_noMultiSim_shouldReturnGsmNoMultiSimString() {
+    public void displayPreference_singleSimCdmaPhone_shouldSetSingleSimCdmaTitleAndMeid() {
         ReflectionHelpers.setField(mController, "mIsMultiSim", false);
+        final String meid = "125132215123";
+        when(mTelephonyManager.getPhoneType()).thenReturn(PHONE_TYPE_CDMA);
+        doReturn(meid).when(mController).getMeid(anyInt());
 
-        assertThat(mController.getTitleForGsmPhone()).isEqualTo(mContext.getResources().getString(
-                R.string.status_imei));
+        mController.displayPreference(mScreen);
+
+        verify(mPreference).setTitle(mContext.getString(R.string.status_meid_number));
+        verify(mPreference).setSummary(meid);
     }
 
     @Test
-    public void getTitleForGsmPhone_multiSim_shouldReturnGsmMultiSimString() {
+    public void displayPreference_multiSimCdmaPhone_shouldSetMultiSimCdmaTitleAndMeid() {
         ReflectionHelpers.setField(mController, "mIsMultiSim", true);
+        final String meid = "125132215123";
+        when(mTelephonyManager.getPhoneCount()).thenReturn(2);
+        when(mTelephonyManager.getPhoneType()).thenReturn(PHONE_TYPE_CDMA);
+        doReturn(meid).when(mController).getMeid(anyInt());
 
-        assertThat(mController.getTitleForGsmPhone()).isEqualTo(mContext.getResources().getString(
-                R.string.imei_multi_sim_slot_1));
+        mController.displayPreference(mScreen);
+
+        verify(mPreference).setTitle(mContext.getString(R.string.meid_multi_sim, 1 /* sim slot */));
+        verify(mSecondSimPreference).setTitle(
+                mContext.getString(R.string.meid_multi_sim, 2 /* sim slot */));
+        verify(mPreference).setSummary(meid);
+        verify(mSecondSimPreference).setSummary(meid);
+    }
+
+    @Test
+    public void displayPreference_singleSimGsmPhone_shouldSetSingleSimGsmTitleAndImei() {
+        ReflectionHelpers.setField(mController, "mIsMultiSim", false);
+        final String imei = "125132215123";
+        when(mTelephonyManager.getPhoneType()).thenReturn(PHONE_TYPE_GSM);
+        when(mTelephonyManager.getImei(anyInt())).thenReturn(imei);
+
+        mController.displayPreference(mScreen);
+
+        verify(mPreference).setTitle(mContext.getString(R.string.status_imei));
+        verify(mPreference).setSummary(imei);
+    }
+
+    @Test
+    public void displayPreference_multiSimGsmPhone_shouldSetMultiSimGsmTitleAndImei() {
+        ReflectionHelpers.setField(mController, "mIsMultiSim", true);
+        final String imei = "125132215123";
+        when(mTelephonyManager.getPhoneCount()).thenReturn(2);
+        when(mTelephonyManager.getPhoneType()).thenReturn(PHONE_TYPE_GSM);
+        when(mTelephonyManager.getImei(anyInt())).thenReturn(imei);
+
+        mController.displayPreference(mScreen);
+
+        verify(mPreference).setTitle(mContext.getString(R.string.imei_multi_sim, 1 /* sim slot */));
+        verify(mSecondSimPreference).setTitle(
+                mContext.getString(R.string.imei_multi_sim, 2 /* sim slot */));
+        verify(mPreference).setSummary(imei);
+        verify(mSecondSimPreference).setSummary(imei);
+    }
+
+    @Test
+    public void handlePreferenceTreeClick_shouldStartDialogFragment() {
+        when(mFragment.getChildFragmentManager()).thenReturn(
+                mock(FragmentManager.class, Answers.RETURNS_DEEP_STUBS));
+        when(mPreference.getTitle()).thenReturn("SomeTitle");
+        mController.displayPreference(mScreen);
+
+        mController.handlePreferenceTreeClick(mPreference);
+
+        verify(mFragment).getChildFragmentManager();
     }
 }