Snap for 6529395 from 86ab0405a8fed4e7c41657eaa1cfc9d0a40e4d59 to sc-release (3de6c640) · Commits · e / os / android_packages_modules_NetworkStack

tests/lib/src/com/android/testutils/TestableNetworkCallback.kt

+11 −1

Original line number	Diff line number	Diff line
		@@ -32,6 +32,7 @@ import com.android.testutils.RecorderCallback.CallbackEntry.Suspended
		import com.android.testutils.RecorderCallback.CallbackEntry.Unavailable
		import kotlin.reflect.KClass
		import kotlin.test.assertEquals
		import kotlin.test.assertNotNull
		import kotlin.test.assertTrue
		import kotlin.test.fail

		@@ -194,7 +195,16 @@ open class TestableNetworkCallback private constructor(
		timeoutMs: Long = defaultTimeoutMs,
		from: Int = mark,
		crossinline predicate: (T) -> Boolean = { true }
		) = history.poll(timeoutMs, from) { it is T && predicate(it) } as T
		): T = eventuallyExpectOrNull(timeoutMs, from, predicate).also {
		assertNotNull(it, "Callback ${T::class} not received within ${timeoutMs}ms")
		} as T

		// TODO (b/157405399) straighten and unify the method names
		inline fun <reified T : CallbackEntry> eventuallyExpectOrNull(
		timeoutMs: Long = defaultTimeoutMs,
		from: Int = mark,
		crossinline predicate: (T) -> Boolean = { true }
		) = history.poll(timeoutMs, from) { it is T && predicate(it) } as T?

		fun expectCallbackThat(
		timeoutMs: Long = defaultTimeoutMs,