FirebaseModelDownloaderTest.java: fix flake due to race condition in getModel_updateBackground_localExists_UpdateFound

Author: dconeybe

firebase-ml-modeldownloader/src/test/java/com/google/firebase/ml/modeldownloader/FirebaseModelDownloaderTest.java

@@ -25,6 +25,7 @@
 import static org.mockito.Mockito.doNothing;
 import static org.mockito.Mockito.mock;
 import static org.mockito.Mockito.never;
+import static org.mockito.Mockito.timeout;
 import static org.mockito.Mockito.times;
 import static org.mockito.Mockito.verify;
 import static org.mockito.Mockito.when;
@@ -540,7 +541,8 @@ public void getModel_updateBackground_localExists_UpdateFound() throws Exception
     verify(mockPrefs, times(2)).getCustomModelDetails(eq(MODEL_NAME));
     assertThat(task.isComplete()).isTrue();
     assertEquals(customModel, customModelLoaded);
-    verify(mockEventLogger)
+
+    verify(mockEventLogger, timeout(5000L))
         .logDownloadEventWithErrorCode(
             UPDATE_CUSTOM_MODEL_URL, false, DownloadStatus.UPDATE_AVAILABLE, ErrorCode.NO_ERROR);
   }