[Settings] Replace ImsManager#isWfcEnabledByPlatform() am: c134ef08d4 am: 9c4d1190d9 am: 712c3aa944 am: 7fe5ab1aea am: 65697d3d0e

Change-Id: I0d5dcfc33b0bddacf059d1fb9af559646e32e68a

tests/robotests/src/com/android/settings/network/ims/MockWifiCallingQueryImsState.java

@@ -17,6 +17,7 @@
 package com.android.settings.network.ims;
 
 import android.content.Context;
+import android.telephony.ims.ImsException;
 
 import com.android.ims.ImsManager;
 
@@ -27,6 +28,7 @@ import com.android.ims.ImsManager;
 public class MockWifiCallingQueryImsState extends WifiCallingQueryImsState {
 
     private Boolean mIsTtyOnVolteEnabled;
+    private Boolean mIsEnabledOnPlatform;
     private Boolean mIsProvisionedOnDevice;
     private Boolean mIsEnabledByUser;
 
@@ -56,6 +58,20 @@ public class MockWifiCallingQueryImsState extends WifiCallingQueryImsState {
         return super.isTtyOnVolteEnabled(subId);
     }
 
+
+    public void setIsEnabledByPlatform(boolean isEnabled) {
+        mIsEnabledOnPlatform = isEnabled;
+    }
+
+    @Override
+    boolean isEnabledByPlatform(int subId) throws InterruptedException, ImsException,
+            IllegalArgumentException {
+        if (mIsEnabledOnPlatform != null) {
+            return mIsEnabledOnPlatform;
+        }
+        return super.isEnabledByPlatform(subId);
+    }
+
     public void setIsProvisionedOnDevice(boolean isProvisioned) {
         mIsProvisionedOnDevice = isProvisioned;
     }

tests/robotests/src/com/android/settings/wifi/calling/WifiCallingSettingsTest.java

@@ -78,7 +78,8 @@ public class WifiCallingSettingsTest {
         doReturn(true).when(mQueryImsState2).isEnabledByUser();
         doReturn(mImsManager).when(mQueryImsState1).getImsManager(anyInt());
         doReturn(mImsManager).when(mQueryImsState2).getImsManager(anyInt());
-        doReturn(true).when(mImsManager).isWfcEnabledByPlatform();
+        mQueryImsState1.setIsEnabledByPlatform(true);
+        mQueryImsState2.setIsEnabledByPlatform(true);
         mQueryImsState1.setIsProvisionedOnDevice(true);
         mQueryImsState2.setIsProvisionedOnDevice(true);
 
@@ -100,7 +101,7 @@ public class WifiCallingSettingsTest {
 
         SubscriptionUtil.setActiveSubscriptionsForTesting(new ArrayList<>(
                 Collections.singletonList(info)));
-        doReturn(true).when(mImsManager).isWfcEnabledByPlatform();
+        mQueryImsState1.setIsEnabledByPlatform(true);
         mQueryImsState1.setIsProvisionedOnDevice(true);
 
         final Intent intent = new Intent();