Fix the failing IncomingCallFragmentTest.

Test: mma
Fixes: 141759990
Change-Id: I0d9ed59cbc0496ccdeb46d4fdf2ea32dfe23225e
diff --git a/tests/robotests/src/com/android/car/dialer/ui/activecall/IncomingCallFragmentTest.java b/tests/robotests/src/com/android/car/dialer/ui/activecall/IncomingCallFragmentTest.java
index bc4ac60..3d8b690 100644
--- a/tests/robotests/src/com/android/car/dialer/ui/activecall/IncomingCallFragmentTest.java
+++ b/tests/robotests/src/com/android/car/dialer/ui/activecall/IncomingCallFragmentTest.java
@@ -23,12 +23,11 @@
 import android.telecom.Call;
 import android.widget.TextView;
 
-import androidx.lifecycle.MutableLiveData;
-
 import com.android.car.dialer.CarDialerRobolectricTestRunner;
 import com.android.car.dialer.FragmentTestActivity;
 import com.android.car.dialer.R;
-import com.android.car.dialer.testutils.ShadowAndroidViewModelFactory;
+import com.android.car.dialer.TestDialerApplication;
+import com.android.car.dialer.telecom.InCallServiceImpl;
 
 import org.junit.Before;
 import org.junit.Test;
@@ -36,26 +35,26 @@
 import org.mockito.Mock;
 import org.mockito.MockitoAnnotations;
 import org.robolectric.Robolectric;
-import org.robolectric.annotation.Config;
+import org.robolectric.RuntimeEnvironment;
 
-@Config(shadows = {ShadowAndroidViewModelFactory.class})
+import java.util.Collections;
+
 @RunWith(CarDialerRobolectricTestRunner.class)
 public class IncomingCallFragmentTest {
     private IncomingCallFragment mIncomingCallFragment;
     @Mock
     private Call mMockCall;
     @Mock
-    private InCallViewModel mMockInCallViewModel;
+    InCallServiceImpl mMockInCallServiceImpl;
 
     @Before
     public void setUp() {
         MockitoAnnotations.initMocks(this);
 
-        ShadowAndroidViewModelFactory.add(InCallViewModel.class, mMockInCallViewModel);
-
-        MutableLiveData<Call> callLiveData = new MutableLiveData<>();
-        callLiveData.setValue(mMockCall);
-        when(mMockInCallViewModel.getIncomingCall()).thenReturn(callLiveData);
+        when(mMockCall.getState()).thenReturn(Call.STATE_RINGING);
+        when(mMockInCallServiceImpl.getCalls()).thenReturn(Collections.singletonList(mMockCall));
+        ((TestDialerApplication) RuntimeEnvironment.application).setupInCallServiceImpl(
+                mMockInCallServiceImpl);
 
         FragmentTestActivity fragmentTestActivity = Robolectric.buildActivity(
                 FragmentTestActivity.class).create().start().resume().get();