diff options
author | Android Build Coastguard Worker <android-build-coastguard-worker@google.com> | 2023-11-10 00:19:19 +0000 |
---|---|---|
committer | Android Build Coastguard Worker <android-build-coastguard-worker@google.com> | 2023-11-10 00:19:19 +0000 |
commit | 8acdd52c9dce7268d27610d334b95ed11f4c34d0 (patch) | |
tree | 36090690bd2e55712ba5e11334ae1ce6d87d3288 | |
parent | cd5d41dc972d50b68d85ae07661d3c61bb842272 (diff) | |
parent | 132bdfcb6e105cedece77a7c5a21910b95a41211 (diff) | |
download | ImsServiceEntitlement-android14-qpr2-s1-release.tar.gz |
Snap for 11078623 from 132bdfcb6e105cedece77a7c5a21910b95a41211 to 24Q1-releaseandroid-14.0.0_r33android-14.0.0_r32android-14.0.0_r31android-14.0.0_r30android-14.0.0_r29android14-qpr2-s3-releaseandroid14-qpr2-s2-releaseandroid14-qpr2-s1-releaseandroid14-qpr2-release
Change-Id: I5714cfbe44d9d1cfeaec18f8ec8b2e48b84f9416
-rw-r--r-- | tests/unittests/src/com/android/imsserviceentitlement/EntitlementUtilsTest.java | 27 |
1 files changed, 24 insertions, 3 deletions
diff --git a/tests/unittests/src/com/android/imsserviceentitlement/EntitlementUtilsTest.java b/tests/unittests/src/com/android/imsserviceentitlement/EntitlementUtilsTest.java index 04477c9..18c3cb0 100644 --- a/tests/unittests/src/com/android/imsserviceentitlement/EntitlementUtilsTest.java +++ b/tests/unittests/src/com/android/imsserviceentitlement/EntitlementUtilsTest.java @@ -20,11 +20,15 @@ import static org.mockito.Mockito.never; import static org.mockito.Mockito.verify; import static org.mockito.Mockito.when; +import android.content.Context; + import androidx.test.runner.AndroidJUnit4; import com.android.imsserviceentitlement.WfcActivationController.EntitlementResultCallback; +import com.android.imsserviceentitlement.entitlement.EntitlementConfiguration; import com.android.imsserviceentitlement.entitlement.EntitlementResult; import com.android.imsserviceentitlement.utils.Executors; +import com.android.libraries.entitlement.ServiceEntitlement; import org.junit.Before; import org.junit.Rule; @@ -35,6 +39,7 @@ import org.mockito.junit.MockitoJUnit; import org.mockito.junit.MockitoRule; import java.lang.reflect.Field; +import java.util.concurrent.CountDownLatch; @RunWith(AndroidJUnit4.class) public class EntitlementUtilsTest { @@ -42,6 +47,9 @@ public class EntitlementUtilsTest { @Mock private ImsEntitlementApi mMockImsEntitlementApi; @Mock private EntitlementResultCallback mEntitlementResultCallback; @Mock private EntitlementResult mEntitlementResult; + @Mock private Context mContext; + @Mock private ServiceEntitlement mServiceEntitlement; + @Mock private EntitlementConfiguration mEntitlementConfiguration; @Before public void setUp() throws Exception { @@ -69,10 +77,23 @@ public class EntitlementUtilsTest { @Test public void entitlementCheck_cancelEntitlementCheck_onFailure() throws Exception { useDirectExecutor(false); - when(mMockImsEntitlementApi.checkEntitlementStatus()).thenReturn(mEntitlementResult); - - EntitlementUtils.entitlementCheck(mMockImsEntitlementApi, mEntitlementResultCallback); + CountDownLatch entitlementCheckLatch = new CountDownLatch(1); + ImsEntitlementApi mockImsEntitlementApi = + new ImsEntitlementApi( + mContext, 1, true, mServiceEntitlement, mEntitlementConfiguration) { + @Override + public EntitlementResult checkEntitlementStatus() { + try { + entitlementCheckLatch.await(); + } catch (InterruptedException e) { + } + return mEntitlementResult; + } + }; + + EntitlementUtils.entitlementCheck(mockImsEntitlementApi, mEntitlementResultCallback); EntitlementUtils.cancelEntitlementCheck(); + entitlementCheckLatch.countDown(); verify(mEntitlementResultCallback, never()).onEntitlementResult(mEntitlementResult); } |