Skip to content

Commit 05001be

Browse files
committed
fix merge error
1 parent 7802907 commit 05001be

File tree

1 file changed

+5
-7
lines changed

1 file changed

+5
-7
lines changed

firebase-ml-modeldownloader/src/test/java/com/google/firebase/ml/modeldownloader/internal/ModelFileDownloadServiceTest.java

Lines changed: 5 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -786,7 +786,8 @@ public void loadNewlyDownloadedModelFile_successFilePresent()
786786
}
787787

788788
@Test
789-
public void loadNewlyDownloadedModelFile_successNoFile() throws FileNotFoundException {
789+
public void loadNewlyDownloadedModelFile_successNoFile()
790+
throws FileNotFoundException, FirebaseMlException {
790791
// Not found
791792
assertNull(modelFileDownloadService.getDownloadingModelStatusCode(0L));
792793
matrixCursor.addRow(new Integer[] {DownloadManager.STATUS_SUCCESSFUL});
@@ -809,7 +810,7 @@ public void loadNewlyDownloadedModelFile_successNoFile() throws FileNotFoundExce
809810
}
810811

811812
@Test
812-
public void loadNewlyDownloadedModelFile_Running() {
813+
public void loadNewlyDownloadedModelFile_Running() throws FirebaseMlException {
813814
// Not found
814815
assertNull(modelFileDownloadService.getDownloadingModelStatusCode(0L));
815816
matrixCursor.addRow(new Integer[] {DownloadManager.STATUS_RUNNING});
@@ -823,7 +824,7 @@ public void loadNewlyDownloadedModelFile_Running() {
823824
}
824825

825826
@Test
826-
public void loadNewlyDownloadedModelFile_Failed() {
827+
public void loadNewlyDownloadedModelFile_Failed() throws FirebaseMlException {
827828
// Not found
828829
assertNull(modelFileDownloadService.getDownloadingModelStatusCode(0L));
829830
matrixCursor =
@@ -871,10 +872,7 @@ public void loadNewlyDownloadedModelFile_initialLoad_successFilePresent()
871872
assertEquals(retrievedModel, customModelDownloadComplete);
872873
verify(mockDownloadManager, times(2)).remove(anyLong());
873874
verify(mockFileManager, times(1)).deleteNonLatestCustomModels();
874-
875-
verify(mockStatsLogger, never())
876-
.logDownloadEventWithErrorCode(any(), anyBoolean(), any(), any());
877-
verify(mockStatsLogger, times(1))
875+
verify(mockStatsLogger, times(2))
878876
.logDownloadEventWithErrorCode(
879877
eq(CUSTOM_MODEL_DOWNLOADING),
880878
eq(true),

0 commit comments

Comments
 (0)