a

mozilla-mobile · Apr 16, 2020 · 05f8313 · 05f8313
1 parent a5cdf61
commit 05f8313
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 5 deletions.
diff --git a/...ddons/src/main/java/mozilla/components/feature/addons/migration/SupportedAddonsChecker.kt b/...ddons/src/main/java/mozilla/components/feature/addons/migration/SupportedAddonsChecker.kt
@@ -158,8 +158,10 @@ internal class SupportedAddonsWorker(
             }
 
             withContext(Dispatchers.Main) {
-                newSupportedAddons.map {
-                    addonManager.enableAddon(it, source = EnableSource.APP_SUPPORT)
+                newSupportedAddons.forEach {
+                    addonManager.enableAddon(it, source = EnableSource.APP_SUPPORT, onError = { error ->
+                        GlobalAddonDependencyProvider.onCrash?.invoke(error)
+                    })
                 }
             }
 

diff --git a/...ns/src/test/java/mozilla/components/feature/addons/migration/SupportedAddonsWorkerTest.kt b/...ns/src/test/java/mozilla/components/feature/addons/migration/SupportedAddonsWorkerTest.kt
@@ -23,19 +23,24 @@ import mozilla.components.feature.addons.ui.translatedName
 import mozilla.components.support.base.ids.SharedIdsHelper
 import mozilla.components.support.ktx.android.content.appName
 import mozilla.components.support.test.mock
+import mozilla.components.support.test.eq
+import mozilla.components.support.test.any
+import mozilla.components.support.test.argumentCaptor
 import mozilla.components.support.test.robolectric.testContext
 import mozilla.components.support.test.rule.MainCoroutineRule
 import mozilla.components.support.test.whenever
 import org.junit.After
 import org.junit.Assert.assertEquals
 import org.junit.Assert.assertTrue
+import org.junit.Assert.assertNotNull
 import org.junit.Assert.assertFalse
 import org.junit.Before
 import org.junit.Rule
 import org.junit.Test
 import org.junit.runner.RunWith
 import org.mockito.Mockito.verify
 import java.io.IOException
+import java.lang.Exception
 
 @RunWith(AndroidJUnit4::class)
 class SupportedAddonsWorkerTest {
@@ -58,15 +63,19 @@ class SupportedAddonsWorkerTest {
     fun `doWork - will return Result_success and create a notification when a new supported add-on is found`() {
         val addonManager = mock<AddonManager>()
         val worker = TestListenableWorkerBuilder<SupportedAddonsWorker>(testContext).build()
-
+        var throwable: Throwable? = null
+        val onCrash = { caught: Throwable ->
+            throwable = caught
+        }
         val unsupportedAddon = mock<Addon> {
             whenever(translatableName).thenReturn(mapOf(Addon.DEFAULT_LOCALE to "one"))
             whenever(isSupported()).thenReturn(true)
             whenever(isDisabledAsUnsupported()).thenReturn(true)
             whenever(defaultLocale).thenReturn(Addon.DEFAULT_LOCALE)
         }
 
-        GlobalAddonDependencyProvider.initialize(addonManager, mock())
+        GlobalAddonDependencyProvider.initialize(addonManager, mock(), onCrash)
+        val onErrorCaptor = argumentCaptor<((Throwable) -> Unit)>()
 
         runBlocking {
             whenever(addonManager.getAddons()).thenReturn(listOf(unsupportedAddon))
@@ -76,7 +85,10 @@ class SupportedAddonsWorkerTest {
 
             val notificationId = SharedIdsHelper.getIdForTag(testContext, NOTIFICATION_TAG)
             assertTrue(isNotificationVisible(notificationId))
-            verify(addonManager).enableAddon(unsupportedAddon, source = EnableSource.APP_SUPPORT)
+            verify(addonManager).enableAddon(eq(unsupportedAddon), source = eq(EnableSource.APP_SUPPORT), onSuccess = any(), onError = onErrorCaptor.capture())
+
+            onErrorCaptor.value.invoke(Exception())
+            assertNotNull(throwable!!)
         }
     }