Skip to content

Commit 5d903cf

Browse files
committed
update tests
1 parent 089be65 commit 5d903cf

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

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

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -88,6 +88,7 @@ dependencies {
8888
annotationProcessor "com.google.auto.value:auto-value:1.6.5"
8989
annotationProcessor project(":encoders:firebase-encoders-processor")
9090

91+
testImplementation(project(":integ-testing"))
9192
testImplementation 'androidx.test:core:1.3.0'
9293
testImplementation 'com.github.tomakehurst:wiremock-standalone:2.26.3'
9394
testImplementation "com.google.truth:truth:$googleTruthVersion"

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

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,7 @@
3737
import com.google.firebase.FirebaseApp;
3838
import com.google.firebase.FirebaseOptions;
3939
import com.google.firebase.FirebaseOptions.Builder;
40+
import com.google.firebase.concurrent.TestOnlyExecutors;
4041
import com.google.firebase.ml.modeldownloader.internal.CustomModelDownloadService;
4142
import com.google.firebase.ml.modeldownloader.internal.FirebaseMlLogEvent.ModelDownloadLogEvent.DownloadStatus;
4243
import com.google.firebase.ml.modeldownloader.internal.FirebaseMlLogEvent.ModelDownloadLogEvent.ErrorCode;
@@ -137,8 +138,8 @@ public void setUp() throws Exception {
137138
mockModelDownloadService,
138139
mockFileManager,
139140
mockEventLogger,
140-
executor,
141-
executor,
141+
TestOnlyExecutors.background(),
142+
TestOnlyExecutors.blocking(),
142143
modelFactory);
143144
setUpTestingFiles(app);
144145

0 commit comments

Comments
 (0)