Fix flaky test: testNsdManager_registerOffloadEngine

Fix flaky test by only catch relevant the AddOrUpdateEvent and
RemoveEvent.

Bug: 294685851
Test: atest CtsNetTestCases
Change-Id: Ia12e51a6527f1e23610fdb0bea1068ebb5b80a73
This commit is contained in:
Yuyang Huang 2023-08-10 16:30:09 +09:00
commit 40f958b90d

View file

@ -910,12 +910,16 @@ class NsdManagerTest {
val record = NsdRegistrationRecord()
nsdManager.registerService(si, NsdManager.PROTOCOL_DNS_SD, record)
val addOrUpdateEvent = offloadEngine
.expectCallback<TestNsdOffloadEngine.OffloadEvent.AddOrUpdateEvent>()
.expectCallbackEventually<TestNsdOffloadEngine.OffloadEvent.AddOrUpdateEvent> {
it.info.key.serviceName == serviceName
}
checkOffloadServiceInfo(addOrUpdateEvent.info)
nsdManager.unregisterService(record)
val unregisterEvent = offloadEngine
.expectCallback<TestNsdOffloadEngine.OffloadEvent.RemoveEvent>()
.expectCallbackEventually<TestNsdOffloadEngine.OffloadEvent.RemoveEvent> {
it.info.key.serviceName == serviceName
}
checkOffloadServiceInfo(unregisterEvent.info)
runAsShell(NETWORK_SETTINGS) {