update tests

Author: VinayGuthal

firebase-ml-modeldownloader/firebase-ml-modeldownloader.gradle

@@ -88,6 +88,7 @@ dependencies {
     annotationProcessor "com.google.auto.value:auto-value:1.6.5"
     annotationProcessor project(":encoders:firebase-encoders-processor")
 
+    testImplementation(project(":integ-testing"))
     testImplementation 'androidx.test:core:1.3.0'
     testImplementation 'com.github.tomakehurst:wiremock-standalone:2.26.3'
     testImplementation "com.google.truth:truth:$googleTruthVersion"

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

@@ -37,6 +37,7 @@
 import com.google.firebase.FirebaseApp;
 import com.google.firebase.FirebaseOptions;
 import com.google.firebase.FirebaseOptions.Builder;
+import com.google.firebase.concurrent.TestOnlyExecutors;
 import com.google.firebase.ml.modeldownloader.internal.CustomModelDownloadService;
 import com.google.firebase.ml.modeldownloader.internal.FirebaseMlLogEvent.ModelDownloadLogEvent.DownloadStatus;
 import com.google.firebase.ml.modeldownloader.internal.FirebaseMlLogEvent.ModelDownloadLogEvent.ErrorCode;
@@ -137,8 +138,8 @@ public void setUp() throws Exception {
             mockModelDownloadService,
             mockFileManager,
             mockEventLogger,
-            executor,
-            executor,
+            TestOnlyExecutors.background(),
+            TestOnlyExecutors.blocking(),
             modelFactory);
     setUpTestingFiles(app);