From e83f0519ba3b45f8773c607d657790019d1b2992 Mon Sep 17 00:00:00 2001 From: yuanjiahsu Date: Sat, 19 Feb 2022 06:16:54 +0800 Subject: [PATCH] Fix flaky test case The callback is called from executor, so verify it with timeout time. Bug: 220358587 Test: atest FrameworksCoreTests:android.companion.virtual, run over 30 times locally and it all passed Change-Id: Idf598396bfb0b09fea8f755e170fea0b5f4eb98e --- .../companion/virtual/audio/VirtualAudioSessionTest.java | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/core/tests/coretests/src/android/companion/virtual/audio/VirtualAudioSessionTest.java b/core/tests/coretests/src/android/companion/virtual/audio/VirtualAudioSessionTest.java index d66cb712b1ca2..0e09d563c8849 100644 --- a/core/tests/coretests/src/android/companion/virtual/audio/VirtualAudioSessionTest.java +++ b/core/tests/coretests/src/android/companion/virtual/audio/VirtualAudioSessionTest.java @@ -22,6 +22,7 @@ import static android.media.AudioFormat.ENCODING_PCM_16BIT; import static com.google.common.truth.Truth.assertThat; +import static org.mockito.Mockito.timeout; import static org.mockito.Mockito.verify; import static org.testng.Assert.assertThrows; @@ -168,7 +169,7 @@ public class VirtualAudioSessionTest { mVirtualAudioSession.onPlaybackConfigChanged(configs); - verify(mCallback).onPlaybackConfigChanged(configs); + verify(mCallback, timeout(2000)).onPlaybackConfigChanged(configs); } @Test @@ -177,6 +178,6 @@ public class VirtualAudioSessionTest { mVirtualAudioSession.onRecordingConfigChanged(configs); - verify(mCallback).onRecordingConfigChanged(configs); + verify(mCallback, timeout(2000)).onRecordingConfigChanged(configs); } }