Merge "Convert DataUsageLib to Kotlin" into main
This commit is contained in:
Chaohui Wang
Android (Google) Code Review
@@ -1,115 +0,0 @@
|
|||||||
/*
|
|
||||||
* Copyright (C) 2020 The Android Open Source Project
|
|
||||||
*
|
|
||||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
||||||
* you may not use this file except in compliance with the License.
|
|
||||||
* You may obtain a copy of the License at
|
|
||||||
*
|
|
||||||
* http://www.apache.org/licenses/LICENSE-2.0
|
|
||||||
*
|
|
||||||
* Unless required by applicable law or agreed to in writing, software
|
|
||||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
||||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
||||||
* See the License for the specific language governing permissions and
|
|
||||||
* limitations under the License.
|
|
||||||
*/
|
|
||||||
|
|
||||||
package com.android.settings.datausage.lib;
|
|
||||||
|
|
||||||
import android.content.Context;
|
|
||||||
import android.net.NetworkStats;
|
|
||||||
import android.net.NetworkTemplate;
|
|
||||||
import android.telephony.SubscriptionInfo;
|
|
||||||
import android.telephony.SubscriptionManager;
|
|
||||||
import android.telephony.TelephonyManager;
|
|
||||||
import android.util.ArraySet;
|
|
||||||
import android.util.Log;
|
|
||||||
|
|
||||||
import androidx.annotation.NonNull;
|
|
||||||
|
|
||||||
import com.android.internal.util.ArrayUtils;
|
|
||||||
|
|
||||||
import java.util.Arrays;
|
|
||||||
import java.util.List;
|
|
||||||
import java.util.Set;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Lib class for data usage
|
|
||||||
*/
|
|
||||||
public class DataUsageLib {
|
|
||||||
private static final String TAG = "DataUsageLib";
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Return mobile NetworkTemplate based on {@code subId}
|
|
||||||
*/
|
|
||||||
public static NetworkTemplate getMobileTemplate(Context context, int subId) {
|
|
||||||
final TelephonyManager telephonyManager = context.getSystemService(TelephonyManager.class);
|
|
||||||
final int mobileDefaultSubId = telephonyManager.getSubscriptionId();
|
|
||||||
|
|
||||||
final SubscriptionManager subscriptionManager =
|
|
||||||
context.getSystemService(SubscriptionManager.class);
|
|
||||||
final List<SubscriptionInfo> subInfoList =
|
|
||||||
subscriptionManager.getAvailableSubscriptionInfoList();
|
|
||||||
if (subInfoList == null) {
|
|
||||||
Log.i(TAG, "Subscription is not inited: " + subId);
|
|
||||||
return getMobileTemplateForSubId(telephonyManager, mobileDefaultSubId);
|
|
||||||
}
|
|
||||||
|
|
||||||
for (SubscriptionInfo subInfo : subInfoList) {
|
|
||||||
if ((subInfo != null) && (subInfo.getSubscriptionId() == subId)) {
|
|
||||||
return getNormalizedMobileTemplate(telephonyManager, subId);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Log.i(TAG, "Subscription is not active: " + subId);
|
|
||||||
return getMobileTemplateForSubId(telephonyManager, mobileDefaultSubId);
|
|
||||||
}
|
|
||||||
|
|
||||||
private static NetworkTemplate getNormalizedMobileTemplate(
|
|
||||||
TelephonyManager telephonyManager, int subId) {
|
|
||||||
final NetworkTemplate mobileTemplate = getMobileTemplateForSubId(telephonyManager, subId);
|
|
||||||
final String[] mergedSubscriberIds = telephonyManager
|
|
||||||
.createForSubscriptionId(subId).getMergedImsisFromGroup();
|
|
||||||
if (ArrayUtils.isEmpty(mergedSubscriberIds)) {
|
|
||||||
Log.i(TAG, "mergedSubscriberIds is null.");
|
|
||||||
return mobileTemplate;
|
|
||||||
}
|
|
||||||
|
|
||||||
return normalizeMobileTemplate(mobileTemplate, mergedSubscriberIds);
|
|
||||||
}
|
|
||||||
|
|
||||||
private static NetworkTemplate normalizeMobileTemplate(
|
|
||||||
@NonNull NetworkTemplate template, @NonNull String[] merged) {
|
|
||||||
if (template.getSubscriberIds().isEmpty()) return template;
|
|
||||||
// The input template should have at most 1 subscriberId.
|
|
||||||
final String subscriberId = template.getSubscriberIds().iterator().next();
|
|
||||||
// In some rare cases (e.g. b/243015487), merged subscriberId list might contain
|
|
||||||
// duplicated items. Deduplication for better error handling.
|
|
||||||
final ArraySet mergedSet = new ArraySet(merged);
|
|
||||||
if (mergedSet.size() != merged.length) {
|
|
||||||
Log.wtf(TAG, "Duplicated merged list detected: " + Arrays.toString(merged));
|
|
||||||
}
|
|
||||||
if (mergedSet.contains(subscriberId)) {
|
|
||||||
// Requested template subscriber is part of the merge group; return
|
|
||||||
// a template that matches all merged subscribers.
|
|
||||||
return new NetworkTemplate.Builder(template.getMatchRule())
|
|
||||||
.setSubscriberIds(mergedSet)
|
|
||||||
.setMeteredness(template.getMeteredness()).build();
|
|
||||||
}
|
|
||||||
|
|
||||||
return template;
|
|
||||||
}
|
|
||||||
|
|
||||||
public static NetworkTemplate getMobileTemplateForSubId(
|
|
||||||
TelephonyManager telephonyManager, int subId) {
|
|
||||||
// Create template that matches any mobile network when the subscriberId is null.
|
|
||||||
String subscriberId = telephonyManager.getSubscriberId(subId);
|
|
||||||
return subscriberId != null
|
|
||||||
? new NetworkTemplate.Builder(NetworkTemplate.MATCH_CARRIER)
|
|
||||||
.setSubscriberIds(Set.of(subscriberId))
|
|
||||||
.setMeteredness(NetworkStats.METERED_YES)
|
|
||||||
.build()
|
|
||||||
: new NetworkTemplate.Builder(NetworkTemplate.MATCH_MOBILE)
|
|
||||||
.setMeteredness(NetworkStats.METERED_YES)
|
|
||||||
.build();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
99
src/com/android/settings/datausage/lib/DataUsageLib.kt
Normal file
99
src/com/android/settings/datausage/lib/DataUsageLib.kt
Normal file
@@ -0,0 +1,99 @@
|
|||||||
|
/*
|
||||||
|
* Copyright (C) 2023 The Android Open Source Project
|
||||||
|
*
|
||||||
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
* you may not use this file except in compliance with the License.
|
||||||
|
* You may obtain a copy of the License at
|
||||||
|
*
|
||||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
*
|
||||||
|
* Unless required by applicable law or agreed to in writing, software
|
||||||
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
* See the License for the specific language governing permissions and
|
||||||
|
* limitations under the License.
|
||||||
|
*/
|
||||||
|
|
||||||
|
package com.android.settings.datausage.lib
|
||||||
|
|
||||||
|
import android.content.Context
|
||||||
|
import android.net.NetworkStats
|
||||||
|
import android.net.NetworkTemplate
|
||||||
|
import android.telephony.SubscriptionManager
|
||||||
|
import android.telephony.TelephonyManager
|
||||||
|
import android.util.Log
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Lib class for data usage
|
||||||
|
*/
|
||||||
|
object DataUsageLib {
|
||||||
|
private const val TAG = "DataUsageLib"
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Return mobile NetworkTemplate based on `subId`
|
||||||
|
*/
|
||||||
|
@JvmStatic
|
||||||
|
fun getMobileTemplate(context: Context, subId: Int): NetworkTemplate {
|
||||||
|
val telephonyManager = context.getSystemService(TelephonyManager::class.java)!!
|
||||||
|
val mobileDefaultSubId = telephonyManager.subscriptionId
|
||||||
|
val subscriptionManager = context.getSystemService(SubscriptionManager::class.java)!!
|
||||||
|
val subInfoList = subscriptionManager.availableSubscriptionInfoList
|
||||||
|
if (subInfoList == null) {
|
||||||
|
Log.i(TAG, "Subscription is not inited: $subId")
|
||||||
|
return getMobileTemplateForSubId(telephonyManager, mobileDefaultSubId)
|
||||||
|
}
|
||||||
|
for (subInfo in subInfoList) {
|
||||||
|
if (subInfo?.subscriptionId == subId) {
|
||||||
|
return getNormalizedMobileTemplate(telephonyManager, subId)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Log.i(TAG, "Subscription is not active: $subId")
|
||||||
|
return getMobileTemplateForSubId(telephonyManager, mobileDefaultSubId)
|
||||||
|
}
|
||||||
|
|
||||||
|
private fun getNormalizedMobileTemplate(
|
||||||
|
telephonyManager: TelephonyManager,
|
||||||
|
subId: Int,
|
||||||
|
): NetworkTemplate {
|
||||||
|
val mobileTemplate = getMobileTemplateForSubId(telephonyManager, subId)
|
||||||
|
val mergedSubscriberIds =
|
||||||
|
telephonyManager.createForSubscriptionId(subId).mergedImsisFromGroup
|
||||||
|
if (mergedSubscriberIds.isNullOrEmpty()) {
|
||||||
|
Log.i(TAG, "mergedSubscriberIds is empty.")
|
||||||
|
return mobileTemplate
|
||||||
|
}
|
||||||
|
return normalizeMobileTemplate(mobileTemplate, mergedSubscriberIds)
|
||||||
|
}
|
||||||
|
|
||||||
|
private fun normalizeMobileTemplate(
|
||||||
|
template: NetworkTemplate,
|
||||||
|
merged: Array<String?>,
|
||||||
|
): NetworkTemplate {
|
||||||
|
val subscriberId = template.subscriberIds.firstOrNull() ?: return template
|
||||||
|
// In some rare cases (e.g. b/243015487), merged subscriberId list might contain
|
||||||
|
// duplicated items. Deduplication for better error handling.
|
||||||
|
val mergedSet = merged.toSet()
|
||||||
|
if (mergedSet.size != merged.size) {
|
||||||
|
Log.wtf(TAG, "Duplicated merged list detected: " + merged.contentToString())
|
||||||
|
}
|
||||||
|
return if (mergedSet.contains(subscriberId)) {
|
||||||
|
// Requested template subscriber is part of the merge group; return
|
||||||
|
// a template that matches all merged subscribers.
|
||||||
|
NetworkTemplate.Builder(template.matchRule)
|
||||||
|
.setSubscriberIds(mergedSet)
|
||||||
|
.setMeteredness(template.meteredness)
|
||||||
|
.build()
|
||||||
|
} else template
|
||||||
|
}
|
||||||
|
|
||||||
|
@JvmStatic
|
||||||
|
fun getMobileTemplateForSubId(telephonyManager: TelephonyManager, subId: Int): NetworkTemplate {
|
||||||
|
// Create template that matches any mobile network when the subscriberId is null.
|
||||||
|
val subscriberId = telephonyManager.getSubscriberId(subId)
|
||||||
|
return when (subscriberId) {
|
||||||
|
null -> NetworkTemplate.Builder(NetworkTemplate.MATCH_MOBILE)
|
||||||
|
else -> NetworkTemplate.Builder(NetworkTemplate.MATCH_CARRIER)
|
||||||
|
.setSubscriberIds(setOf(subscriberId))
|
||||||
|
}.setMeteredness(NetworkStats.METERED_YES).build()
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -1,116 +0,0 @@
|
|||||||
/*
|
|
||||||
* Copyright (C) 2020 The Android Open Source Project
|
|
||||||
*
|
|
||||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
||||||
* you may not use this file except in compliance with the License.
|
|
||||||
* You may obtain a copy of the License at
|
|
||||||
*
|
|
||||||
* http://www.apache.org/licenses/LICENSE-2.0
|
|
||||||
*
|
|
||||||
* Unless required by applicable law or agreed to in writing, software
|
|
||||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
||||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
||||||
* See the License for the specific language governing permissions and
|
|
||||||
* limitations under the License.
|
|
||||||
*/
|
|
||||||
|
|
||||||
package com.android.settings.datausage.lib;
|
|
||||||
|
|
||||||
import static com.google.common.truth.Truth.assertThat;
|
|
||||||
|
|
||||||
import static org.mockito.Mockito.anyInt;
|
|
||||||
import static org.mockito.Mockito.spy;
|
|
||||||
import static org.mockito.Mockito.when;
|
|
||||||
|
|
||||||
import android.content.Context;
|
|
||||||
import android.net.NetworkTemplate;
|
|
||||||
import android.os.ParcelUuid;
|
|
||||||
import android.os.RemoteException;
|
|
||||||
import android.telephony.SubscriptionInfo;
|
|
||||||
import android.telephony.SubscriptionManager;
|
|
||||||
import android.telephony.TelephonyManager;
|
|
||||||
|
|
||||||
import org.junit.Before;
|
|
||||||
import org.junit.Ignore;
|
|
||||||
import org.junit.Test;
|
|
||||||
import org.junit.runner.RunWith;
|
|
||||||
import org.mockito.Mock;
|
|
||||||
import org.mockito.MockitoAnnotations;
|
|
||||||
import org.robolectric.RobolectricTestRunner;
|
|
||||||
import org.robolectric.RuntimeEnvironment;
|
|
||||||
|
|
||||||
import java.util.List;
|
|
||||||
|
|
||||||
@RunWith(RobolectricTestRunner.class)
|
|
||||||
public class DataUsageLibTest {
|
|
||||||
|
|
||||||
private static final int SUB_ID = 1;
|
|
||||||
private static final int SUB_ID_2 = 2;
|
|
||||||
private static final String SUBSCRIBER_ID = "Test Subscriber";
|
|
||||||
private static final String SUBSCRIBER_ID_2 = "Test Subscriber 2";
|
|
||||||
|
|
||||||
@Mock
|
|
||||||
private TelephonyManager mTelephonyManager;
|
|
||||||
@Mock
|
|
||||||
private SubscriptionManager mSubscriptionManager;
|
|
||||||
@Mock
|
|
||||||
private SubscriptionInfo mInfo1;
|
|
||||||
@Mock
|
|
||||||
private SubscriptionInfo mInfo2;
|
|
||||||
@Mock
|
|
||||||
private ParcelUuid mParcelUuid;
|
|
||||||
private Context mContext;
|
|
||||||
private List<SubscriptionInfo> mInfos;
|
|
||||||
|
|
||||||
@Before
|
|
||||||
public void setUp() throws RemoteException {
|
|
||||||
MockitoAnnotations.initMocks(this);
|
|
||||||
|
|
||||||
mContext = spy(RuntimeEnvironment.application);
|
|
||||||
when(mContext.getSystemService(TelephonyManager.class)).thenReturn(mTelephonyManager);
|
|
||||||
when(mContext.getSystemService(SubscriptionManager.class)).thenReturn(mSubscriptionManager);
|
|
||||||
when(mTelephonyManager.getSubscriberId(SUB_ID)).thenReturn(SUBSCRIBER_ID);
|
|
||||||
when(mTelephonyManager.getSubscriberId(SUB_ID_2)).thenReturn(SUBSCRIBER_ID_2);
|
|
||||||
when(mTelephonyManager.createForSubscriptionId(anyInt())).thenReturn(mTelephonyManager);
|
|
||||||
when(mSubscriptionManager.isActiveSubscriptionId(anyInt())).thenReturn(true);
|
|
||||||
}
|
|
||||||
|
|
||||||
@Test
|
|
||||||
@Ignore
|
|
||||||
public void getMobileTemplate_infoNull_returnMobileAll() {
|
|
||||||
when(mSubscriptionManager.isActiveSubscriptionId(SUB_ID)).thenReturn(false);
|
|
||||||
|
|
||||||
final NetworkTemplate networkTemplate = DataUsageLib.getMobileTemplate(mContext, SUB_ID);
|
|
||||||
assertThat(networkTemplate.getSubscriberIds().contains(SUBSCRIBER_ID)).isTrue();
|
|
||||||
assertThat(networkTemplate.getSubscriberIds().contains(SUBSCRIBER_ID_2)).isFalse();
|
|
||||||
}
|
|
||||||
|
|
||||||
@Test
|
|
||||||
@Ignore
|
|
||||||
public void getMobileTemplate_groupUuidNull_returnMobileAll() {
|
|
||||||
when(mSubscriptionManager.getActiveSubscriptionInfo(SUB_ID)).thenReturn(mInfo1);
|
|
||||||
when(mInfo1.getGroupUuid()).thenReturn(null);
|
|
||||||
when(mTelephonyManager.getMergedImsisFromGroup())
|
|
||||||
.thenReturn(new String[] {SUBSCRIBER_ID});
|
|
||||||
|
|
||||||
final NetworkTemplate networkTemplate = DataUsageLib.getMobileTemplate(mContext, SUB_ID);
|
|
||||||
assertThat(networkTemplate.getSubscriberIds().contains(SUBSCRIBER_ID)).isTrue();
|
|
||||||
assertThat(networkTemplate.getSubscriberIds().contains(SUBSCRIBER_ID_2)).isFalse();
|
|
||||||
}
|
|
||||||
|
|
||||||
@Test
|
|
||||||
@Ignore
|
|
||||||
public void getMobileTemplate_groupUuidExist_returnMobileMerged() {
|
|
||||||
when(mSubscriptionManager.getActiveSubscriptionInfo(SUB_ID)).thenReturn(mInfo1);
|
|
||||||
when(mInfo1.getGroupUuid()).thenReturn(mParcelUuid);
|
|
||||||
// In some rare cases (e.g. b/243015487), merged subscriberId list might contain
|
|
||||||
// duplicated items. The implementation should perform deduplication.
|
|
||||||
when(mTelephonyManager.getMergedImsisFromGroup())
|
|
||||||
.thenReturn(new String[] {SUBSCRIBER_ID, SUBSCRIBER_ID, SUBSCRIBER_ID_2});
|
|
||||||
|
|
||||||
final NetworkTemplate networkTemplate = DataUsageLib.getMobileTemplate(mContext, SUB_ID);
|
|
||||||
assertThat(networkTemplate.getSubscriberIds().contains(SUBSCRIBER_ID)).isTrue();
|
|
||||||
assertThat(networkTemplate.getSubscriberIds().contains(SUBSCRIBER_ID_2)).isTrue();
|
|
||||||
assertThat(networkTemplate.getSubscriberIds().size() == 2).isTrue();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -0,0 +1,171 @@
|
|||||||
|
/*
|
||||||
|
* Copyright (C) 2023 The Android Open Source Project
|
||||||
|
*
|
||||||
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
* you may not use this file except in compliance with the License.
|
||||||
|
* You may obtain a copy of the License at
|
||||||
|
*
|
||||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
*
|
||||||
|
* Unless required by applicable law or agreed to in writing, software
|
||||||
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
* See the License for the specific language governing permissions and
|
||||||
|
* limitations under the License.
|
||||||
|
*/
|
||||||
|
|
||||||
|
package com.android.settings.datausage.lib
|
||||||
|
|
||||||
|
import android.content.Context
|
||||||
|
import android.net.NetworkStats
|
||||||
|
import android.net.NetworkTemplate
|
||||||
|
import android.telephony.SubscriptionInfo
|
||||||
|
import android.telephony.SubscriptionManager
|
||||||
|
import android.telephony.TelephonyManager
|
||||||
|
import androidx.test.core.app.ApplicationProvider
|
||||||
|
import androidx.test.ext.junit.runners.AndroidJUnit4
|
||||||
|
import com.google.common.truth.Truth.assertThat
|
||||||
|
import org.junit.Before
|
||||||
|
import org.junit.Rule
|
||||||
|
import org.junit.Test
|
||||||
|
import org.junit.runner.RunWith
|
||||||
|
import org.mockito.Mock
|
||||||
|
import org.mockito.Spy
|
||||||
|
import org.mockito.junit.MockitoJUnit
|
||||||
|
import org.mockito.junit.MockitoRule
|
||||||
|
import org.mockito.Mockito.`when` as whenever
|
||||||
|
|
||||||
|
@RunWith(AndroidJUnit4::class)
|
||||||
|
class DataUsageLibTest {
|
||||||
|
|
||||||
|
@get:Rule
|
||||||
|
val mockito: MockitoRule = MockitoJUnit.rule()
|
||||||
|
|
||||||
|
@Spy
|
||||||
|
private val context: Context = ApplicationProvider.getApplicationContext()
|
||||||
|
|
||||||
|
@Mock
|
||||||
|
private lateinit var telephonyManager: TelephonyManager
|
||||||
|
|
||||||
|
@Mock
|
||||||
|
private lateinit var subscriptionManager: SubscriptionManager
|
||||||
|
|
||||||
|
@Before
|
||||||
|
fun setUp() {
|
||||||
|
whenever(context.getSystemService(TelephonyManager::class.java))
|
||||||
|
.thenReturn(telephonyManager)
|
||||||
|
whenever(context.getSystemService(SubscriptionManager::class.java))
|
||||||
|
.thenReturn(subscriptionManager)
|
||||||
|
|
||||||
|
whenever(telephonyManager.subscriptionId).thenReturn(DEFAULT_SUB_ID)
|
||||||
|
whenever(telephonyManager.getSubscriberId(SUB_ID)).thenReturn(SUBSCRIBER_ID)
|
||||||
|
whenever(telephonyManager.getSubscriberId(DEFAULT_SUB_ID)).thenReturn(DEFAULT_SUBSCRIBER_ID)
|
||||||
|
whenever(telephonyManager.createForSubscriptionId(SUB_ID)).thenReturn(telephonyManager)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun getMobileTemplate_availableSubscriptionInfoListIsNull_returnDefaultSub() {
|
||||||
|
whenever(subscriptionManager.availableSubscriptionInfoList).thenReturn(null)
|
||||||
|
|
||||||
|
val mobileTemplate = DataUsageLib.getMobileTemplate(context, SUB_ID)
|
||||||
|
|
||||||
|
assertThat(mobileTemplate.matchRule).isEqualTo(NetworkTemplate.MATCH_CARRIER)
|
||||||
|
assertThat(mobileTemplate.subscriberIds).containsExactly(DEFAULT_SUBSCRIBER_ID)
|
||||||
|
assertThat(mobileTemplate.meteredness).isEqualTo(NetworkStats.METERED_YES)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun getMobileTemplate_subscriptionNotActive_returnDefaultSub() {
|
||||||
|
whenever(subscriptionManager.availableSubscriptionInfoList).thenReturn(listOf(null))
|
||||||
|
|
||||||
|
val mobileTemplate = DataUsageLib.getMobileTemplate(context, SUB_ID)
|
||||||
|
|
||||||
|
assertThat(mobileTemplate.matchRule).isEqualTo(NetworkTemplate.MATCH_CARRIER)
|
||||||
|
assertThat(mobileTemplate.subscriberIds).containsExactly(DEFAULT_SUBSCRIBER_ID)
|
||||||
|
assertThat(mobileTemplate.meteredness).isEqualTo(NetworkStats.METERED_YES)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun getMobileTemplate_mergedImsisFromGroupNull_returnRequestedSub() {
|
||||||
|
whenever(subscriptionManager.availableSubscriptionInfoList)
|
||||||
|
.thenReturn(listOf(SUBSCRIBER_INFO))
|
||||||
|
whenever(telephonyManager.mergedImsisFromGroup).thenReturn(null)
|
||||||
|
|
||||||
|
val mobileTemplate = DataUsageLib.getMobileTemplate(context, SUB_ID)
|
||||||
|
|
||||||
|
assertThat(mobileTemplate.matchRule).isEqualTo(NetworkTemplate.MATCH_CARRIER)
|
||||||
|
assertThat(mobileTemplate.subscriberIds).containsExactly(SUBSCRIBER_ID)
|
||||||
|
assertThat(mobileTemplate.meteredness).isEqualTo(NetworkStats.METERED_YES)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun getMobileTemplate_mergedImsisFromGroupEmpty_returnRequestedSub() {
|
||||||
|
whenever(subscriptionManager.availableSubscriptionInfoList)
|
||||||
|
.thenReturn(listOf(SUBSCRIBER_INFO))
|
||||||
|
whenever(telephonyManager.mergedImsisFromGroup).thenReturn(emptyArray())
|
||||||
|
|
||||||
|
val mobileTemplate = DataUsageLib.getMobileTemplate(context, SUB_ID)
|
||||||
|
|
||||||
|
assertThat(mobileTemplate.matchRule).isEqualTo(NetworkTemplate.MATCH_CARRIER)
|
||||||
|
assertThat(mobileTemplate.subscriberIds).containsExactly(SUBSCRIBER_ID)
|
||||||
|
assertThat(mobileTemplate.meteredness).isEqualTo(NetworkStats.METERED_YES)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun getMobileTemplate_mergedImsisFromGroupNotContainSub_returnRequestedSub() {
|
||||||
|
whenever(subscriptionManager.availableSubscriptionInfoList)
|
||||||
|
.thenReturn(listOf(SUBSCRIBER_INFO))
|
||||||
|
whenever(telephonyManager.mergedImsisFromGroup).thenReturn(arrayOf(DEFAULT_SUBSCRIBER_ID))
|
||||||
|
|
||||||
|
val mobileTemplate = DataUsageLib.getMobileTemplate(context, SUB_ID)
|
||||||
|
|
||||||
|
assertThat(mobileTemplate.matchRule).isEqualTo(NetworkTemplate.MATCH_CARRIER)
|
||||||
|
assertThat(mobileTemplate.subscriberIds).containsExactly(SUBSCRIBER_ID)
|
||||||
|
assertThat(mobileTemplate.meteredness).isEqualTo(NetworkStats.METERED_YES)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun getMobileTemplate_mergedImsisFromGroupContainSub_returnRequestedSub() {
|
||||||
|
whenever(subscriptionManager.availableSubscriptionInfoList)
|
||||||
|
.thenReturn(listOf(SUBSCRIBER_INFO))
|
||||||
|
whenever(telephonyManager.mergedImsisFromGroup)
|
||||||
|
.thenReturn(arrayOf(DEFAULT_SUBSCRIBER_ID, SUBSCRIBER_ID))
|
||||||
|
|
||||||
|
val mobileTemplate = DataUsageLib.getMobileTemplate(context, SUB_ID)
|
||||||
|
|
||||||
|
assertThat(mobileTemplate.matchRule).isEqualTo(NetworkTemplate.MATCH_CARRIER)
|
||||||
|
assertThat(mobileTemplate.subscriberIds)
|
||||||
|
.containsExactly(SUBSCRIBER_ID, DEFAULT_SUBSCRIBER_ID)
|
||||||
|
assertThat(mobileTemplate.meteredness).isEqualTo(NetworkStats.METERED_YES)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun getMobileTemplateForSubId_subscriberIdNotNull() {
|
||||||
|
whenever(telephonyManager.getSubscriberId(SUB_ID)).thenReturn(SUBSCRIBER_ID)
|
||||||
|
|
||||||
|
val mobileTemplate = DataUsageLib.getMobileTemplateForSubId(telephonyManager, SUB_ID)
|
||||||
|
|
||||||
|
assertThat(mobileTemplate.matchRule).isEqualTo(NetworkTemplate.MATCH_CARRIER)
|
||||||
|
assertThat(mobileTemplate.subscriberIds).containsExactly(SUBSCRIBER_ID)
|
||||||
|
assertThat(mobileTemplate.meteredness).isEqualTo(NetworkStats.METERED_YES)
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun getMobileTemplateForSubId_subscriberIdIsNull() {
|
||||||
|
whenever(telephonyManager.getSubscriberId(SUB_ID)).thenReturn(null)
|
||||||
|
|
||||||
|
val mobileTemplate = DataUsageLib.getMobileTemplateForSubId(telephonyManager, SUB_ID)
|
||||||
|
|
||||||
|
assertThat(mobileTemplate.matchRule).isEqualTo(NetworkTemplate.MATCH_MOBILE)
|
||||||
|
assertThat(mobileTemplate.subscriberIds).isEmpty()
|
||||||
|
assertThat(mobileTemplate.meteredness).isEqualTo(NetworkStats.METERED_YES)
|
||||||
|
}
|
||||||
|
|
||||||
|
private companion object {
|
||||||
|
const val DEFAULT_SUB_ID = 0
|
||||||
|
const val SUB_ID = 1
|
||||||
|
const val DEFAULT_SUBSCRIBER_ID = "Default Test Subscriber"
|
||||||
|
const val SUBSCRIBER_ID = "Test Subscriber"
|
||||||
|
val SUBSCRIBER_INFO: SubscriptionInfo = SubscriptionInfo.Builder().setId(SUB_ID).build()
|
||||||
|
}
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user