From 524f93613fdb5f8ad180681be593133cf77474fb Mon Sep 17 00:00:00 2001
From: Chalard Jean <jchalard@google.com>
Date: Sun, 12 Apr 2020 14:53:45 +0000
Subject: [PATCH] Test onBandwidthUpdateRequested

Test: this
Bug: 139268426
Change-Id: I5f4b42dc68fdd13f26e59b4e2217d39dcee8f2a1
Merged-In: I427ae6ac2c8910683e47f503ba71a05e35507571
(cherry picked from commit bbe53cd710440468520d5d5713eaa503b6c9d8b9, aosp/1258136)
---
 .../net/src/android/net/cts/NetworkAgentTest.kt  | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

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<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()
+    }
 }