Closes issue #16944 Refactor TrackingProtectionPanelDialogFragment to observe session via store.
Arturo Mejia
4 years ago
parent
9bee4797de
commit
743ba6918f
@ -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