Merge "Test onBandwidthUpdateRequested" into rvc-dev

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<CallbackEntry>().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<OnBandwidthUpdateRequested>()
+        agent.unregister()
+    }
 }