commit ab9f1262702da216e9fa6cf798192ad1fb15a006
author Chalard Jean <jchalard@google.com> Fri Mar 13 21:08:39 2020 +0900
committer Chalard Jean <jchalard@google.com> Fri Apr 10 11:20:40 2020 +0900
tree 4b8ecb2f1faa6b468d2ef8d7ec8c075d6ca3b261
parent a2b1460f895169a284d1461cd329b8337811bcfe

Test onBandwidthUpdateRequested

Test: this
Bug: 139268426
Change-Id: I427ae6ac2c8910683e47f503ba71a05e35507571

tests/cts/net/src/android/net/cts/NetworkAgentTest.kt

diff --git a/tests/cts/net/src/android/net/cts/NetworkAgentTest.kt b/tests/cts/net/src/android/net/cts/NetworkAgentTest.kt
index 85c94e7..2fdd5fb 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.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 @@
         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 @@
             agent.register()
         }
     }
+
+    @Test
+    fun testOnBandwidthUpdateRequested() {
+        val (agent, callback) = createConnectedNetworkAgent()
+        callback.expectAvailableThenValidatedCallbacks(agent.network)
+        mCM.requestBandwidthUpdate(agent.network)
+        agent.expectCallback<OnBandwidthUpdateRequested>()
+        agent.unregister()
+    }
 }