Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit 5aea5bc6 authored by Junyu Lai's avatar Junyu Lai Committed by Automerger Merge Worker
Browse files

Merge "[MS51] Remove NetworkTemplate#buildTemplate* usages from SettingsLib" am: 69e3121b

Original change: https://android-review.googlesource.com/c/platform/frameworks/base/+/1944685

Change-Id: I07f6f57ff7b3bd56b7dc51fa46cbcbc9068baff0
parents 445203d3 69e3121b
Loading
Loading
Loading
Loading
+1 −2
Original line number Diff line number Diff line
@@ -102,8 +102,7 @@ public class DataUsageController {
    }

    public DataUsageInfo getWifiDataUsageInfo() {
        NetworkTemplate template = NetworkTemplate.buildTemplateWifi(
                NetworkTemplate.WIFI_NETWORKID_ALL, null);
        NetworkTemplate template = new NetworkTemplate.Builder(NetworkTemplate.MATCH_WIFI).build();
        return getDataUsageInfo(template);
    }

+11 −4
Original line number Diff line number Diff line
@@ -17,6 +17,7 @@
package com.android.settingslib.net;

import android.content.Context;
import android.net.NetworkStats;
import android.net.NetworkTemplate;
import android.telephony.SubscriptionInfo;
import android.telephony.SubscriptionManager;
@@ -26,6 +27,7 @@ import android.util.Log;
import com.android.internal.util.ArrayUtils;

import java.util.List;
import java.util.Set;

/**
 * Utils class for data usage
@@ -73,10 +75,15 @@ public class DataUsageUtils {

    private static NetworkTemplate getMobileTemplateForSubId(
            TelephonyManager telephonyManager, int subId) {
        // The null subscriberId means that no any mobile/carrier network will be matched.
        // Using old API: buildTemplateMobileAll for the null subscriberId to avoid NPE.
        // Create template that matches any mobile network when the subscriberId is null.
        String subscriberId = telephonyManager.getSubscriberId(subId);
        return subscriberId != null ? NetworkTemplate.buildTemplateCarrierMetered(subscriberId)
                : NetworkTemplate.buildTemplateMobileAll(subscriberId);
        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();
    }
}
+6 −1
Original line number Diff line number Diff line
@@ -20,6 +20,7 @@ import static junit.framework.Assert.assertEquals;

import android.net.NetworkPolicy;
import android.net.NetworkPolicyManager;
import android.net.NetworkStats;
import android.net.NetworkTemplate;

import androidx.test.InstrumentationRegistry;
@@ -32,6 +33,8 @@ import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;

import java.util.Set;

@RunWith(AndroidJUnit4.class)
@SmallTest
public class NetworkPolicyEditorTest {
@@ -44,7 +47,9 @@ public class NetworkPolicyEditorTest {

    @Before
    public void setUp() {
        mNetworkTemplate = NetworkTemplate.buildTemplateCarrierMetered("123456789123456");
        mNetworkTemplate = new NetworkTemplate.Builder(NetworkTemplate.MATCH_CARRIER)
                .setMeteredness(NetworkStats.METERED_YES)
                .setSubscriberIds(Set.of("123456789123456")).build();
        NetworkPolicyManager policyManager = NetworkPolicyManager.from(InstrumentationRegistry
                .getContext());
        mNetworkPolicyEditor = new NetworkPolicyEditor(policyManager);
+9 −4
Original line number Diff line number Diff line
@@ -42,6 +42,8 @@ import org.mockito.MockitoAnnotations;
import org.robolectric.RobolectricTestRunner;
import org.robolectric.shadows.ShadowSubscriptionManager;

import java.util.Set;

@RunWith(RobolectricTestRunner.class)
public class DataUsageControllerTest {

@@ -75,10 +77,13 @@ public class DataUsageControllerTest {
        ShadowSubscriptionManager.setDefaultDataSubscriptionId(mDefaultSubscriptionId);
        doReturn(SUB_ID).when(mTelephonyManager).getSubscriberId();

        mNetworkTemplate = NetworkTemplate.buildTemplateCarrierMetered(SUB_ID);
        mNetworkTemplate2 = NetworkTemplate.buildTemplateCarrierMetered(SUB_ID_2);
        mWifiNetworkTemplate = NetworkTemplate.buildTemplateWifi(
                NetworkTemplate.WIFI_NETWORKID_ALL, null);
        mNetworkTemplate = new NetworkTemplate.Builder(NetworkTemplate.MATCH_CARRIER)
                .setMeteredness(android.net.NetworkStats.METERED_YES)
                .setSubscriberIds(Set.of(SUB_ID)).build();
        mNetworkTemplate2 = new NetworkTemplate.Builder(NetworkTemplate.MATCH_CARRIER)
                .setMeteredness(android.net.NetworkStats.METERED_YES)
                .setSubscriberIds(Set.of(SUB_ID_2)).build();
        mWifiNetworkTemplate = new NetworkTemplate.Builder(NetworkTemplate.MATCH_WIFI).build();
    }

    @Test
+6 −1
Original line number Diff line number Diff line
@@ -30,6 +30,7 @@ import android.app.usage.NetworkStatsManager;
import android.content.Context;
import android.net.NetworkPolicy;
import android.net.NetworkPolicyManager;
import android.net.NetworkStats;
import android.net.NetworkTemplate;
import android.text.format.DateUtils;

@@ -40,6 +41,8 @@ import org.mockito.Mock;
import org.mockito.MockitoAnnotations;
import org.robolectric.RobolectricTestRunner;

import java.util.Set;

@RunWith(RobolectricTestRunner.class)
public class NetworkCycleDataForUidLoaderTest {
    private static final String SUB_ID = "Test Subscriber";
@@ -62,7 +65,9 @@ public class NetworkCycleDataForUidLoaderTest {
        when(mContext.getSystemService(Context.NETWORK_POLICY_SERVICE))
                .thenReturn(mNetworkPolicyManager);
        when(mNetworkPolicyManager.getNetworkPolicies()).thenReturn(new NetworkPolicy[0]);
        mNetworkTemplate = NetworkTemplate.buildTemplateCarrierMetered(SUB_ID);
        mNetworkTemplate = new NetworkTemplate.Builder(NetworkTemplate.MATCH_CARRIER)
                .setMeteredness(NetworkStats.METERED_YES)
                .setSubscriberIds(Set.of(SUB_ID)).build();
    }

    @Test