Add support for testing onReserved callbacks

Test: TH
Change-Id: I0d35567dae5865cc2de9e5f6811f1b9a57a04cbd
This commit is contained in:
Patrick Rohr 2024-12-09 15:14:55 -08:00
commit fe3d77a042

View file

@ -32,6 +32,7 @@ import com.android.testutils.RecorderCallback.CallbackEntry.LinkPropertiesChange
import com.android.testutils.RecorderCallback.CallbackEntry.LocalInfoChanged
import com.android.testutils.RecorderCallback.CallbackEntry.Losing
import com.android.testutils.RecorderCallback.CallbackEntry.Lost
import com.android.testutils.RecorderCallback.CallbackEntry.Reserved
import com.android.testutils.RecorderCallback.CallbackEntry.Resumed
import com.android.testutils.RecorderCallback.CallbackEntry.Suspended
import com.android.testutils.RecorderCallback.CallbackEntry.Unavailable
@ -66,6 +67,12 @@ open class RecorderCallback private constructor(
// constructor by specifying override.
abstract val network: Network
data class Reserved private constructor(
override val network: Network,
val caps: NetworkCapabilities
): CallbackEntry() {
constructor(caps: NetworkCapabilities) : this(NULL_NETWORK, caps)
}
data class Available(override val network: Network) : CallbackEntry()
data class CapabilitiesChanged(
override val network: Network,
@ -99,6 +106,8 @@ open class RecorderCallback private constructor(
// Convenience constants for expecting a type
companion object {
@JvmField
val RESERVED = Reserved::class
@JvmField
val AVAILABLE = Available::class
@JvmField
@ -127,6 +136,11 @@ open class RecorderCallback private constructor(
val history = backingRecord.newReadHead()
val mark get() = history.mark
override fun onReserved(caps: NetworkCapabilities) {
Log.d(logTag, "onReserved $caps")
history.add(Reserved(caps))
}
override fun onAvailable(network: Network) {
Log.d(logTag, "onAvailable $network")
history.add(Available(network))