commit 7f0d18537d76d03b49e4ddf661bb1bcbb0cbbd80
author Aaron Huang <huangaaron@google.com> Tue Feb 08 23:17:39 2022 +0800
committer Aaron Huang <huangaaron@google.com> Wed Feb 09 14:58:20 2022 +0800
tree dc24ff4fd24561325775ab8f3c808a72759842ee
parent d97e5a42d436f35e2d07b1d7cafad4e3cefc1efe

Add test for NetworkTemplate.Builder#setRoaming(int)

Ignore-AOSP-First: Required API is not in downstream yet
Bug: 215435701
Test: CtsNetTestCases
Change-Id: Ie000244a954585241304adef95b4156e64d33f41

tests/common/java/android/net/netstats/NetworkTemplateTest.kt

diff --git a/tests/common/java/android/net/netstats/NetworkTemplateTest.kt b/tests/common/java/android/net/netstats/NetworkTemplateTest.kt
index eb2df0d..192694b 100644
--- a/tests/common/java/android/net/netstats/NetworkTemplateTest.kt
+++ b/tests/common/java/android/net/netstats/NetworkTemplateTest.kt
@@ -19,6 +19,7 @@
 import android.net.NetworkStats.DEFAULT_NETWORK_ALL
 import android.net.NetworkStats.METERED_ALL
 import android.net.NetworkStats.METERED_YES
+import android.net.NetworkStats.ROAMING_YES
 import android.net.NetworkStats.ROAMING_ALL
 import android.net.NetworkTemplate
 import android.net.NetworkTemplate.MATCH_BLUETOOTH
@@ -84,6 +85,19 @@
                     }
         }
 
+        // Verify template which matches roaming cellular and carrier networks with
+        // the given IMSI.
+        listOf(MATCH_MOBILE, MATCH_CARRIER).forEach { matchRule ->
+            NetworkTemplate.Builder(matchRule).setSubscriberIds(setOf(TEST_IMSI1))
+                    .setRoaming(ROAMING_YES).setMeteredness(METERED_YES).build().let {
+                        val expectedTemplate = NetworkTemplate(matchRule, TEST_IMSI1,
+                                arrayOf(TEST_IMSI1), arrayOf<String>(), METERED_YES,
+                                ROAMING_YES, DEFAULT_NETWORK_ALL, NETWORK_TYPE_ALL,
+                                OEM_MANAGED_ALL, SUBSCRIBER_ID_MATCH_RULE_EXACT)
+                        assertEquals(expectedTemplate, it)
+                    }
+        }
+
         // Verify carrier template cannot be created without IMSI.
         assertFailsWith<IllegalArgumentException> {
             NetworkTemplate.Builder(MATCH_CARRIER).build()