diff --git a/tests/cts/net/src/android/net/cts/NetworkAgentTest.kt b/tests/cts/net/src/android/net/cts/NetworkAgentTest.kt index 85c94e7db2..2fdd5fb201 100644 --- a/tests/cts/net/src/android/net/cts/NetworkAgentTest.kt +++ b/tests/cts/net/src/android/net/cts/NetworkAgentTest.kt @@ -24,6 +24,7 @@ import android.net.NetworkAgentConfig import android.net.NetworkCapabilities import android.net.NetworkProvider import android.net.NetworkRequest +import android.net.cts.NetworkAgentTest.TestableNetworkAgent.CallbackEntry.OnBandwidthUpdateRequested import android.net.cts.NetworkAgentTest.TestableNetworkAgent.CallbackEntry.OnNetworkUnwanted import android.os.Build import android.os.HandlerThread @@ -88,9 +89,15 @@ class NetworkAgentTest { private val history = ArrayTrackRecord().newReadHead() sealed class CallbackEntry { + object OnBandwidthUpdateRequested : CallbackEntry() object OnNetworkUnwanted : CallbackEntry() } + override fun onBandwidthUpdateRequested() { + super.onBandwidthUpdateRequested() + history.add(OnBandwidthUpdateRequested) + } + override fun onNetworkUnwanted() { super.onNetworkUnwanted() history.add(OnNetworkUnwanted) @@ -139,4 +146,13 @@ class NetworkAgentTest { agent.register() } } + + @Test + fun testOnBandwidthUpdateRequested() { + val (agent, callback) = createConnectedNetworkAgent() + callback.expectAvailableThenValidatedCallbacks(agent.network) + mCM.requestBandwidthUpdate(agent.network) + agent.expectCallback() + agent.unregister() + } }