No Issue: Fix DownloadItemKtTest

4 years ago · ffb794751e
parent f3b870d356
commit ffb794751e
1 changed files with 2 additions and 1 deletions
--- a/app/src/test/java/org/mozilla/fenix/ext/DownloadItemKtTest.kt
+++ b/app/src/test/java/org/mozilla/fenix/ext/DownloadItemKtTest.kt
@ -4,6 +4,7 @@

 package org.mozilla.fenix.ext

+import mozilla.components.browser.state.state.content.DownloadState
 import org.junit.Test

 import org.junit.Assert.assertEquals
@ -13,7 +14,7 @@ import org.mozilla.fenix.library.downloads.DownloadItem
 class DownloadItemKtTest {
    @Test
    fun getIcon() {
-        val downloadItem = DownloadItem(0, "MyAwesomeFile", "", "", "image/png")
+        val downloadItem = DownloadItem("0", "MyAwesomeFile", "", "", "image/png", DownloadState.Status.COMPLETED)

        assertEquals(R.drawable.ic_file_type_image, downloadItem.getIcon())
        assertEquals(R.drawable.ic_file_type_audio_note, downloadItem.copy(contentType = "audio/mp3").getIcon())