Merge "Convert DataUsageLib to Kotlin" into main
This commit is contained in:
Chaohui Wang
Android (Google) Code Review
@@ -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