Bug 1876567 - Fix Middleware Issue with Processing Translations States

This patch corrects an issue where `isTranslateProcessing` and
`isRestoreProcessing` states were not updating due to not sending along
pre-process actions in `EngineDelegateMiddleware`.

android-components/components/browser/state/src/main/java/mozilla/components/browser/state/engine/middleware/EngineDelegateMiddleware.kt

@@ -44,8 +44,8 @@ internal class EngineDelegateMiddleware(
             is EngineAction.ExitFullScreenModeAction -> exitFullScreen(context.store, action)
             is EngineAction.SaveToPdfAction -> saveToPdf(context.store, action)
             is EngineAction.PrintContentAction -> printContent(context.store, action)
-            is TranslationsAction.TranslateAction -> translate(context.store, action)
-            is TranslationsAction.TranslateRestoreAction -> translateRestoreOriginal(context.store, action)
+            is TranslationsAction.TranslateAction -> translate(context.store, next, action)
+            is TranslationsAction.TranslateRestoreAction -> translateRestoreOriginal(context.store, next, action)
             is EngineAction.ClearDataAction -> clearData(context.store, action)
             is EngineAction.PurgeHistoryAction -> purgeHistory(context.state)
             else -> next(action)
@@ -156,16 +156,20 @@ internal class EngineDelegateMiddleware(
 
     private fun translate(
         store: Store<BrowserState, BrowserAction>,
+        next: (BrowserAction) -> Unit,
         action: TranslationsAction.TranslateAction,
     ) = scope.launch {
+        next(action)
         getEngineSessionOrDispatch(store, action)
             ?.requestTranslate(action.fromLanguage, action.toLanguage, action.options)
     }
 
     private fun translateRestoreOriginal(
         store: Store<BrowserState, BrowserAction>,
+        next: (BrowserAction) -> Unit,
         action: TranslationsAction.TranslateRestoreAction,
     ) = scope.launch {
+        next(action)
         getEngineSessionOrDispatch(store, action)
             ?.requestTranslationRestore()
     }

android-components/components/browser/state/src/test/java/mozilla/components/browser/state/engine/middleware/EngineDelegateMiddlewareTest.kt

@@ -5,14 +5,17 @@
 package mozilla.components.browser.state.engine.middleware
 
 import mozilla.components.browser.state.action.EngineAction
+import mozilla.components.browser.state.action.TranslationsAction
 import mozilla.components.browser.state.engine.EngineMiddleware
+import mozilla.components.browser.state.selector.findTab
 import mozilla.components.browser.state.state.BrowserState
 import mozilla.components.browser.state.state.EngineState
 import mozilla.components.browser.state.state.createCustomTab
 import mozilla.components.browser.state.state.createTab
 import mozilla.components.browser.state.store.BrowserStore
 import mozilla.components.concept.engine.Engine
 import mozilla.components.concept.engine.EngineSession
+import mozilla.components.support.test.any
 import mozilla.components.support.test.ext.joinBlocking
 import mozilla.components.support.test.libstate.ext.waitUntilIdle
 import mozilla.components.support.test.mock
@@ -742,4 +745,67 @@ class EngineDelegateMiddlewareTest {
         verify(engineSession1).purgeHistory()
         verify(engineSession2).purgeHistory()
     }
+
+    @Test
+    fun `TranslateAction correctly sets progress state AND begins a translation`() {
+        val tab = createTab("https://www.mozilla.org")
+        val engineSession: EngineSession = mock()
+        val engine: Engine = mock()
+        doReturn(engineSession).`when`(engine).createSession()
+        val store = BrowserStore(
+            middleware = EngineMiddleware.create(
+                engine = engine,
+                scope = scope,
+            ),
+            initialState = BrowserState(
+                tabs = listOf(tab),
+            ),
+        )
+
+        assertEquals(false, store.state.findTab(tab.id)?.translationsState?.isTranslateProcessing)
+
+        store.dispatch(
+            TranslationsAction.TranslateAction(
+                tabId = tab.id,
+                fromLanguage = "es",
+                toLanguage = "en",
+                options = null,
+            ),
+        ).joinBlocking()
+
+        dispatcher.scheduler.advanceUntilIdle()
+        store.waitUntilIdle()
+
+        verify(engineSession).requestTranslate(any(), any(), any())
+        assertEquals(true, store.state.findTab(tab.id)?.translationsState?.isTranslateProcessing)
+    }
+
+    @Test
+    fun `TranslateRestoreAction correctly sets progress state AND begins a restore`() {
+        val tab = createTab("https://www.mozilla.org")
+        val engineSession: EngineSession = mock()
+        val engine: Engine = mock()
+        doReturn(engineSession).`when`(engine).createSession()
+        val store = BrowserStore(
+            middleware = EngineMiddleware.create(
+                engine = engine,
+                scope = scope,
+            ),
+            initialState = BrowserState(
+                tabs = listOf(tab),
+            ),
+        )
+
+        assertEquals(false, store.state.findTab(tab.id)?.translationsState?.isRestoreProcessing)
+
+        store.dispatch(
+            TranslationsAction.TranslateRestoreAction(tabId = tab.id),
+        ).joinBlocking()
+
+        dispatcher.scheduler.advanceUntilIdle()
+        store.waitUntilIdle()
+
+        verify(engineSession).requestTranslationRestore()
+        assertEquals(true, store.state.findTab(tab.id)?.translationsState?.isRestoreProcessing)
+    }
 }