LineageOS/android_packages_modules_Connectivity
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge "Add a CTS test for NetworkAgent#setUnderlyingNetworks."

Browse Source
This commit is contained in:
Lorenzo Colitti
2020-11-28 02:49:53 +00:00
committed by Gerrit Code Review
parent e74700dd5d 3657ce9cda
commit 7d04b2a26c
1 changed files with 62 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

66
tests/cts/net/src/android/net/cts/NetworkAgentTest.kt
View File

@@ -82,6 +82,7 @@ import org.mockito.Mockito.mock
import org.mockito.Mockito.verify import org.mockito.Mockito.verify
import java.net.InetAddress import java.net.InetAddress
import java.time.Duration import java.time.Duration
import java.util.Arrays
import java.util.UUID import java.util.UUID
import kotlin.test.assertEquals import kotlin.test.assertEquals
import kotlin.test.assertFailsWith import kotlin.test.assertFailsWith
@@ -312,9 +313,11 @@ class NetworkAgentTest {
private fun createNetworkAgent( private fun createNetworkAgent(
context: Context = realContext, context: Context = realContext,
name: String? = null name: String? = null,
nc: NetworkCapabilities = NetworkCapabilities(),
lp: LinkProperties = LinkProperties()
): TestableNetworkAgent { ): TestableNetworkAgent {
val nc = NetworkCapabilities().apply { nc.apply {
addTransportType(NetworkCapabilities.TRANSPORT_TEST) addTransportType(NetworkCapabilities.TRANSPORT_TEST)
removeCapability(NetworkCapabilities.NET_CAPABILITY_TRUSTED) removeCapability(NetworkCapabilities.NET_CAPABILITY_TRUSTED)
removeCapability(NetworkCapabilities.NET_CAPABILITY_INTERNET) removeCapability(NetworkCapabilities.NET_CAPABILITY_INTERNET)
@@ -325,7 +328,7 @@ class NetworkAgentTest {
setNetworkSpecifier(StringNetworkSpecifier(name)) setNetworkSpecifier(StringNetworkSpecifier(name))
} }
} }
val lp = LinkProperties().apply { lp.apply {
addLinkAddress(LinkAddress(LOCAL_IPV4_ADDRESS, 0)) addLinkAddress(LinkAddress(LOCAL_IPV4_ADDRESS, 0))
} }
val config = NetworkAgentConfig.Builder().build() val config = NetworkAgentConfig.Builder().build()
@@ -541,8 +544,63 @@ class NetworkAgentTest {
// tearDown() will unregister the requests and agents // tearDown() will unregister the requests and agents
} }
private fun hasAllTransports(nc: NetworkCapabilities?, transports: IntArray) =
nc != null && transports.all { nc.hasTransport(it) }
@Test @Test
@IgnoreUpTo(android.os.Build.VERSION_CODES.R) @IgnoreUpTo(Build.VERSION_CODES.R)
fun testSetUnderlyingNetworks() {
val request = NetworkRequest.Builder()
.addTransportType(NetworkCapabilities.TRANSPORT_TEST)
.addTransportType(NetworkCapabilities.TRANSPORT_VPN)
.removeCapability(NetworkCapabilities.NET_CAPABILITY_NOT_VPN)
.removeCapability(NetworkCapabilities.NET_CAPABILITY_TRUSTED) // TODO: add to VPN!
.build()
val callback = TestableNetworkCallback()
mCM.registerNetworkCallback(request, callback)
val nc = NetworkCapabilities().apply {
addTransportType(NetworkCapabilities.TRANSPORT_TEST)
addTransportType(NetworkCapabilities.TRANSPORT_VPN)
removeCapability(NetworkCapabilities.NET_CAPABILITY_NOT_VPN)
}
val defaultNetwork = mCM.activeNetwork
assertNotNull(defaultNetwork)
val defaultNetworkTransports = mCM.getNetworkCapabilities(defaultNetwork).transportTypes
val agent = createNetworkAgent(nc = nc)
agent.register()
agent.markConnected()
callback.expectAvailableThenValidatedCallbacks(agent.network!!)
var vpnNc = mCM.getNetworkCapabilities(agent.network)
assertNotNull(vpnNc)
assertTrue(NetworkCapabilities.TRANSPORT_VPN in vpnNc.transportTypes)
assertTrue(hasAllTransports(vpnNc, defaultNetworkTransports),
"VPN transports ${Arrays.toString(vpnNc.transportTypes)}" +
" lacking transports from ${Arrays.toString(defaultNetworkTransports)}")
agent.setUnderlyingNetworks(listOf<Network>())
callback.expectCapabilitiesThat(agent.network!!) {
it.transportTypes.size == 1 && it.hasTransport(NetworkCapabilities.TRANSPORT_VPN)
}
val expectedTransports = (defaultNetworkTransports.toSet() +
NetworkCapabilities.TRANSPORT_VPN).toIntArray()
agent.setUnderlyingNetworks(null)
callback.expectCapabilitiesThat(agent.network!!) {
it.transportTypes.size == expectedTransports.size &&
hasAllTransports(it, expectedTransports)
}
agent.unregister()
callback.expectCallback<Lost>(agent.network)
mCM.unregisterNetworkCallback(callback)
}
@Test
@IgnoreUpTo(Build.VERSION_CODES.R)
fun testAgentStartsInConnecting() { fun testAgentStartsInConnecting() {
val mockContext = mock(Context::class.java) val mockContext = mock(Context::class.java)
val mockCm = mock(ConnectivityManager::class.java) val mockCm = mock(ConnectivityManager::class.java)