From ca13280e564454d89c12a1cc77d6453b6063321e Mon Sep 17 00:00:00 2001
From: Weng Su <wengsu@google.com>
Date: Fri, 7 Jun 2024 05:32:59 +0800
Subject: [PATCH] Fix Wi-Fi calling test case

Fix: 337417499
Flag: TEST_ONLY
Test: atest passed
atest -c NetworkProviderWifiCallingGroupTest

Change-Id: I2d57287a85075685b0bdb35a59dc8b147e90d38c
---
 .../telephony/NetworkProviderWifiCallingGroupTest.java       | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/tests/unit/src/com/android/settings/network/telephony/NetworkProviderWifiCallingGroupTest.java b/tests/unit/src/com/android/settings/network/telephony/NetworkProviderWifiCallingGroupTest.java
index 2165bc9a088..3adcbdb8b23 100644
--- a/tests/unit/src/com/android/settings/network/telephony/NetworkProviderWifiCallingGroupTest.java
+++ b/tests/unit/src/com/android/settings/network/telephony/NetworkProviderWifiCallingGroupTest.java
@@ -48,7 +48,6 @@ import com.android.settings.network.ims.MockWfcQueryImsState;
 import com.android.settingslib.core.lifecycle.Lifecycle;
 
 import org.junit.Before;
-import org.junit.Ignore;
 import org.junit.Test;
 import org.junit.runner.RunWith;
 import org.mockito.Mock;
@@ -115,7 +114,7 @@ public class NetworkProviderWifiCallingGroupTest {
         mCarrierConfig.putBoolean(CarrierConfigManager.KEY_CARRIER_WFC_IMS_AVAILABLE_BOOL, true);
         when(mTelecomManager.getSimCallManagerForSubscription(SUB_ID))
                 .thenReturn(mPhoneAccountHandle);
-        mMockQueryWfcState = new MockWfcQueryImsState(mContext, SUB_ID);
+        mMockQueryWfcState = spy(new MockWfcQueryImsState(mContext, SUB_ID));
 
         if (Looper.myLooper() == null) {
             Looper.prepare();
@@ -177,11 +176,11 @@ public class NetworkProviderWifiCallingGroupTest {
     }
 
     @Test
-    @Ignore("b/337417499")
     public void
     shouldShowWifiCallingForSub_wifiCallingEnabledWithActivityHandleIntent_returnTrue() {
         buildPhoneAccountConfigureIntent(true);
         doReturn(mMockQueryWfcState).when(mNetworkProviderWifiCallingGroup).queryImsState(SUB_ID);
+        doReturn(true).when(mMockQueryWfcState).isReadyToWifiCalling();
         doReturn(mPhoneAccountHandle).when(mNetworkProviderWifiCallingGroup)
                 .getPhoneAccountHandleForSubscriptionId(SUB_ID);