[fenix] Closes issue https://github.com/mozilla-mobile/fenix/issues/16944 Refactor TrackingProtectionPanelDialogFragment to observe session via store.
parent
91fb839701
commit
72e6e7e69e
@ -0,0 +1,149 @@
|
|||||||
|
/* This Source Code Form is subject to the terms of the Mozilla Public
|
||||||
|
License, v. 2.0. If a copy of the MPL was not distributed with this
|
||||||
|
file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
||||||
|
|
||||||
|
package org.mozilla.fenix.trackingprotection
|
||||||
|
|
||||||
|
import androidx.lifecycle.Lifecycle
|
||||||
|
import androidx.lifecycle.LifecycleOwner
|
||||||
|
import androidx.lifecycle.LifecycleRegistry
|
||||||
|
import io.mockk.every
|
||||||
|
import io.mockk.mockk
|
||||||
|
import io.mockk.spyk
|
||||||
|
import io.mockk.verify
|
||||||
|
import junit.framework.TestCase.assertNotSame
|
||||||
|
import kotlinx.coroutines.ExperimentalCoroutinesApi
|
||||||
|
import kotlinx.coroutines.test.TestCoroutineDispatcher
|
||||||
|
import mozilla.components.browser.state.action.ContentAction
|
||||||
|
import mozilla.components.browser.state.action.TabListAction
|
||||||
|
import mozilla.components.browser.state.action.TrackingProtectionAction.TrackerBlockedAction
|
||||||
|
import mozilla.components.browser.state.action.TrackingProtectionAction.TrackerLoadedAction
|
||||||
|
import mozilla.components.browser.state.selector.findTab
|
||||||
|
import mozilla.components.browser.state.state.TabSessionState
|
||||||
|
import mozilla.components.browser.state.state.createTab
|
||||||
|
import mozilla.components.browser.state.store.BrowserStore
|
||||||
|
import mozilla.components.support.test.ext.joinBlocking
|
||||||
|
import mozilla.components.support.test.rule.MainCoroutineRule
|
||||||
|
import org.junit.After
|
||||||
|
import org.junit.Before
|
||||||
|
import org.junit.Rule
|
||||||
|
import org.junit.Test
|
||||||
|
import org.junit.runner.RunWith
|
||||||
|
import org.mozilla.fenix.helpers.FenixRobolectricTestRunner
|
||||||
|
|
||||||
|
@ExperimentalCoroutinesApi
|
||||||
|
@RunWith(FenixRobolectricTestRunner::class)
|
||||||
|
class TrackingProtectionPanelDialogFragmentTest {
|
||||||
|
|
||||||
|
private val testDispatcher = TestCoroutineDispatcher()
|
||||||
|
|
||||||
|
@get:Rule
|
||||||
|
val coroutinesTestRule = MainCoroutineRule(testDispatcher)
|
||||||
|
private lateinit var lifecycleOwner: MockedLifecycleOwner
|
||||||
|
private lateinit var fragment: TrackingProtectionPanelDialogFragment
|
||||||
|
private lateinit var store: BrowserStore
|
||||||
|
|
||||||
|
@Before
|
||||||
|
fun setup() {
|
||||||
|
fragment = spyk(TrackingProtectionPanelDialogFragment())
|
||||||
|
lifecycleOwner = MockedLifecycleOwner(Lifecycle.State.STARTED)
|
||||||
|
|
||||||
|
store = BrowserStore()
|
||||||
|
every { fragment.view } returns mockk(relaxed = true)
|
||||||
|
every { fragment.lifecycle } returns lifecycleOwner.lifecycle
|
||||||
|
every { fragment.activity } returns mockk(relaxed = true)
|
||||||
|
}
|
||||||
|
|
||||||
|
@After
|
||||||
|
fun cleanUp() {
|
||||||
|
testDispatcher.cleanupTestCoroutines()
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun `WHEN the url is updated THEN the url view is updated`() {
|
||||||
|
val trackingProtectionStore: TrackingProtectionStore = mockk(relaxed = true)
|
||||||
|
val tab = createTab("mozilla.org")
|
||||||
|
|
||||||
|
every { fragment.trackingProtectionStore } returns trackingProtectionStore
|
||||||
|
every { fragment.provideTabId() } returns tab.id
|
||||||
|
|
||||||
|
fragment.observeUrlChange(store)
|
||||||
|
addAndSelectTab(tab)
|
||||||
|
|
||||||
|
verify(exactly = 1) {
|
||||||
|
trackingProtectionStore.dispatch(TrackingProtectionAction.UrlChange("mozilla.org"))
|
||||||
|
}
|
||||||
|
|
||||||
|
store.dispatch(ContentAction.UpdateUrlAction(tab.id, "wikipedia.org")).joinBlocking()
|
||||||
|
|
||||||
|
verify(exactly = 1) {
|
||||||
|
trackingProtectionStore.dispatch(TrackingProtectionAction.UrlChange("wikipedia.org"))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun `WHEN a tracker is loaded THEN trackers view is updated`() {
|
||||||
|
val trackingProtectionStore: TrackingProtectionStore = mockk(relaxed = true)
|
||||||
|
val tab = createTab("mozilla.org")
|
||||||
|
|
||||||
|
every { fragment.trackingProtectionStore } returns trackingProtectionStore
|
||||||
|
every { fragment.provideTabId() } returns tab.id
|
||||||
|
every { fragment.updateTrackers(any()) } returns Unit
|
||||||
|
|
||||||
|
fragment.observeTrackersChange(store)
|
||||||
|
addAndSelectTab(tab)
|
||||||
|
|
||||||
|
verify(exactly = 1) {
|
||||||
|
fragment.updateTrackers(tab)
|
||||||
|
}
|
||||||
|
|
||||||
|
store.dispatch(TrackerLoadedAction(tab.id, mockk())).joinBlocking()
|
||||||
|
|
||||||
|
val updatedTab = store.state.findTab(tab.id)!!
|
||||||
|
|
||||||
|
assertNotSame(updatedTab, tab)
|
||||||
|
|
||||||
|
verify(exactly = 1) {
|
||||||
|
fragment.updateTrackers(updatedTab)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
fun `WHEN a tracker is blocked THEN trackers view is updated`() {
|
||||||
|
val trackingProtectionStore: TrackingProtectionStore = mockk(relaxed = true)
|
||||||
|
val tab = createTab("mozilla.org")
|
||||||
|
|
||||||
|
every { fragment.trackingProtectionStore } returns trackingProtectionStore
|
||||||
|
every { fragment.provideTabId() } returns tab.id
|
||||||
|
every { fragment.updateTrackers(any()) } returns Unit
|
||||||
|
|
||||||
|
fragment.observeTrackersChange(store)
|
||||||
|
addAndSelectTab(tab)
|
||||||
|
|
||||||
|
verify(exactly = 1) {
|
||||||
|
fragment.updateTrackers(tab)
|
||||||
|
}
|
||||||
|
|
||||||
|
store.dispatch(TrackerBlockedAction(tab.id, mockk())).joinBlocking()
|
||||||
|
|
||||||
|
val updatedTab = store.state.findTab(tab.id)!!
|
||||||
|
|
||||||
|
assertNotSame(updatedTab, tab)
|
||||||
|
|
||||||
|
verify(exactly = 1) {
|
||||||
|
fragment.updateTrackers(tab)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
private fun addAndSelectTab(tab: TabSessionState) {
|
||||||
|
store.dispatch(TabListAction.AddTabAction(tab)).joinBlocking()
|
||||||
|
store.dispatch(TabListAction.SelectTabAction(tab.id)).joinBlocking()
|
||||||
|
}
|
||||||
|
|
||||||
|
internal class MockedLifecycleOwner(initialState: Lifecycle.State) : LifecycleOwner {
|
||||||
|
private val lifecycleRegistry = LifecycleRegistry(this).apply {
|
||||||
|
currentState = initialState
|
||||||
|
}
|
||||||
|
|
||||||
|
override fun getLifecycle(): Lifecycle = lifecycleRegistry
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in New Issue