commit 391f37f41df54c8cb853f7b04ffe19170547dca4
author SongFerng Wang <songferngwang@google.com> Wed Jan 24 13:41:28 2024 +0000
committer SongFerng Wang <songferngwang@google.com> Wed Jan 24 13:45:58 2024 +0000
tree bc748954b20218de6db8bfe8d04cf70da06dca18
parent 37d2c195c0570a37aae232f314cf495cd305e725

Renew the testcase for using new api setDataEnabledForReason

Bug: 318310357
Change-Id: Ia800ab2727747e5f2d979051a3ffd7fe4ce92ed9
Test: test the MobileDataSliceTest

tests/robotests/src/com/android/settings/network/telephony/MobileDataSliceTest.java

diff --git a/tests/robotests/src/com/android/settings/network/telephony/MobileDataSliceTest.java b/tests/robotests/src/com/android/settings/network/telephony/MobileDataSliceTest.java
index ba37c30..8445fe2 100644
--- a/tests/robotests/src/com/android/settings/network/telephony/MobileDataSliceTest.java
+++ b/tests/robotests/src/com/android/settings/network/telephony/MobileDataSliceTest.java
@@ -114,7 +114,8 @@
 
         mMobileDataSlice.onNotifyChange(intent);
 
-        verify(mTelephonyManager).setDataEnabled(true);
+        verify(mTelephonyManager)
+                .setDataEnabledForReason(TelephonyManager.DATA_ENABLED_REASON_USER, true);
     }
 
     @Test
@@ -126,7 +127,8 @@
 
         mMobileDataSlice.onNotifyChange(intent);
 
-        verify(mTelephonyManager).setDataEnabled(false);
+        verify(mTelephonyManager)
+                .setDataEnabledForReason(TelephonyManager.DATA_ENABLED_REASON_USER, false);
     }
 
     @Test
@@ -138,7 +140,8 @@
 
         mMobileDataSlice.onNotifyChange(intent);
 
-        verify(mTelephonyManager, times(0)).setDataEnabled(true);
+        verify(mTelephonyManager, times(0))
+                .setDataEnabledForReason(TelephonyManager.DATA_ENABLED_REASON_USER, true);
     }
 
     @Test