Fix TabDataRepositoryTest

Author: 0nko

app/src/test/java/com/duckduckgo/tabs/model/TabDataRepositoryTest.kt

@@ -21,9 +21,11 @@ import androidx.lifecycle.MutableLiveData
 import androidx.room.Room
 import androidx.test.ext.junit.runners.AndroidJUnit4
 import androidx.test.platform.app.InstrumentationRegistry
+import com.duckduckgo.adclick.api.AdClickManager
 import com.duckduckgo.app.browser.DuckDuckGoUrlDetector
 import com.duckduckgo.app.browser.certificates.BypassedSSLCertificatesRepository
 import com.duckduckgo.app.browser.favicon.FaviconManager
+import com.duckduckgo.app.browser.session.WebViewSessionStorage
 import com.duckduckgo.app.browser.tabpreview.WebViewPreviewPersister
 import com.duckduckgo.app.global.db.AppDatabase
 import com.duckduckgo.app.global.model.SiteFactoryImpl
@@ -92,6 +94,10 @@ class TabDataRepositoryTest {
 
     private val tabManagerFeatureFlags = FakeFeatureToggleFactory.create(TabManagerFeatureFlags::class.java)
 
+    private val mockWebViewSessionStorage: WebViewSessionStorage = mock()
+
+    private val mockAdClickManager: AdClickManager = mock()
+
     @Before
     fun before() {
         tabManagerFeatureFlags.multiSelection().setRawStoredState(State(enable = false))
@@ -635,7 +641,7 @@ class TabDataRepositoryTest {
 
         testee.undoDeletable(tabIds)
 
-        verify(mockDao).undoDeletableTabs(tabIds)
+        verify(mockDao).undoDeletableTabs(tabIds, false)
     }
 
     @Test
@@ -660,6 +666,24 @@ class TabDataRepositoryTest {
         verify(mockDao).deleteTabsAndUpdateSelection(tabIds)
         tabIds.forEach { tabId ->
             assertNull(testee.retrieveSiteData(tabId).value)
+            verify(mockWebViewSessionStorage).deleteSession(tabId)
+            verify(mockAdClickManager).clearTabId(tabId)
+        }
+    }
+
+    @Test
+    fun whenPurgeDeletableTabsThenPurgeDeletableTabsAndClearData() = runTest {
+        val testee = tabDataRepository()
+        val tabIds = listOf("tabid1", "tabid2")
+        whenever(mockDao.getDeletableTabIds()).thenReturn(tabIds)
+
+        testee.purgeDeletableTabs()
+
+        verify(mockDao).purgeDeletableTabsAndUpdateSelection()
+        tabIds.forEach { tabId ->
+            assertNull(testee.retrieveSiteData(tabId).value)
+            verify(mockWebViewSessionStorage).deleteSession(tabId)
+            verify(mockAdClickManager).clearTabId(tabId)
         }
     }
 
@@ -693,7 +717,8 @@ class TabDataRepositoryTest {
             timeProvider,
             coroutinesTestRule.testScope,
             coroutinesTestRule.testDispatcherProvider,
-            tabManagerFeatureFlags,
+            mockAdClickManager,
+            mockWebViewSessionStorage,
         )
     }