134 lines
4.9 KiB
Kotlin
134 lines
4.9 KiB
Kotlin
/* 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.collections
|
|
|
|
import io.mockk.MockKAnnotations
|
|
import io.mockk.every
|
|
import io.mockk.impl.annotations.MockK
|
|
import kotlinx.coroutines.CompletableDeferred
|
|
import kotlinx.coroutines.ExperimentalCoroutinesApi
|
|
import mozilla.components.browser.state.state.BrowserState
|
|
import mozilla.components.browser.state.state.ReaderState
|
|
import mozilla.components.browser.state.state.createTab
|
|
import mozilla.components.feature.tab.collections.Tab
|
|
import mozilla.components.lib.publicsuffixlist.PublicSuffixList
|
|
import mozilla.components.support.test.robolectric.createAddedTestFragment
|
|
import mozilla.components.support.test.robolectric.testContext
|
|
import org.junit.Assert.assertEquals
|
|
import org.junit.Assert.assertNotNull
|
|
import org.junit.Assert.assertNull
|
|
import org.junit.Assert.assertTrue
|
|
import org.junit.Before
|
|
import org.junit.Test
|
|
import org.junit.runner.RunWith
|
|
import org.mozilla.fenix.ext.components
|
|
import org.mozilla.fenix.helpers.FenixRobolectricTestRunner
|
|
|
|
private const val URL_MOZILLA = "www.mozilla.org"
|
|
private const val SESSION_ID_MOZILLA = "0"
|
|
private const val URL_BCC = "www.bcc.co.uk"
|
|
private const val SESSION_ID_BCC = "1"
|
|
|
|
private const val SESSION_ID_BAD_1 = "not a real session id"
|
|
private const val SESSION_ID_BAD_2 = "definitely not a real session id"
|
|
|
|
@ExperimentalCoroutinesApi
|
|
@RunWith(FenixRobolectricTestRunner::class)
|
|
class CollectionCreationFragmentTest {
|
|
|
|
@MockK(relaxed = true) private lateinit var publicSuffixList: PublicSuffixList
|
|
|
|
private val sessionMozilla = createTab(URL_MOZILLA, id = SESSION_ID_MOZILLA)
|
|
private val sessionBcc = createTab(URL_BCC, id = SESSION_ID_BCC)
|
|
private val state = BrowserState(
|
|
tabs = listOf(sessionMozilla, sessionBcc)
|
|
)
|
|
|
|
@Before
|
|
fun before() {
|
|
MockKAnnotations.init(this)
|
|
every { publicSuffixList.stripPublicSuffix(URL_MOZILLA) } returns CompletableDeferred(URL_MOZILLA)
|
|
every { publicSuffixList.stripPublicSuffix(URL_BCC) } returns CompletableDeferred(URL_BCC)
|
|
}
|
|
|
|
@Test
|
|
fun `creation dialog shows and can be dismissed`() {
|
|
val store = testContext.components.core.store
|
|
every { store.state } returns state
|
|
|
|
val fragment = createAddedTestFragment {
|
|
CollectionCreationFragment().apply {
|
|
arguments = CollectionCreationFragmentArgs(
|
|
saveCollectionStep = SaveCollectionStep.SelectTabs
|
|
).toBundle()
|
|
}
|
|
}
|
|
|
|
assertNotNull(fragment.dialog)
|
|
assertTrue(fragment.requireDialog().isShowing)
|
|
fragment.dismiss()
|
|
assertNull(fragment.dialog)
|
|
}
|
|
|
|
@Test
|
|
fun `GIVEN tabs are present in state WHEN getTabs is called THEN tabs will be returned`() {
|
|
val tabs = state.getTabs(arrayOf(SESSION_ID_MOZILLA, SESSION_ID_BCC), publicSuffixList)
|
|
|
|
val hosts = tabs.map { it.hostname }
|
|
|
|
assertEquals(URL_MOZILLA, hosts[0])
|
|
assertEquals(URL_BCC, hosts[1])
|
|
}
|
|
|
|
@Test
|
|
fun `GIVEN some tabs are present in state WHEN getTabs is called THEN only valid tabs will be returned`() {
|
|
val tabs = state.getTabs(arrayOf(SESSION_ID_MOZILLA, SESSION_ID_BAD_1), publicSuffixList)
|
|
|
|
val hosts = tabs.map { it.hostname }
|
|
|
|
assertEquals(URL_MOZILLA, hosts[0])
|
|
assertEquals(1, hosts.size)
|
|
}
|
|
|
|
@Test
|
|
fun `GIVEN tabs are not present in state WHEN getTabs is called THEN an empty list will be returned`() {
|
|
val tabs = state.getTabs(arrayOf(SESSION_ID_BAD_1, SESSION_ID_BAD_2), publicSuffixList)
|
|
|
|
assertEquals(emptyList<Tab>(), tabs)
|
|
}
|
|
|
|
@Test
|
|
fun `WHEN getTabs is called will null tabIds THEN an empty list will be returned`() {
|
|
val tabs = state.getTabs(null, publicSuffixList)
|
|
|
|
assertEquals(emptyList<Tab>(), tabs)
|
|
}
|
|
|
|
@Test
|
|
fun `toTab uses active reader URL`() {
|
|
val tabWithoutReaderState = createTab(url = "https://example.com", id = "1")
|
|
|
|
val tabWithInactiveReaderState = createTab(url = "https://blog.mozilla.org", id = "2",
|
|
readerState = ReaderState(active = false, activeUrl = null)
|
|
)
|
|
|
|
val tabWithActiveReaderState = createTab(url = "moz-extension://123", id = "3",
|
|
readerState = ReaderState(active = true, activeUrl = "https://blog.mozilla.org/123")
|
|
)
|
|
|
|
val state = BrowserState(
|
|
tabs = listOf(tabWithoutReaderState, tabWithInactiveReaderState, tabWithActiveReaderState)
|
|
)
|
|
val tabs = state.getTabs(
|
|
arrayOf(tabWithoutReaderState.id, tabWithInactiveReaderState.id, tabWithActiveReaderState.id),
|
|
publicSuffixList
|
|
)
|
|
|
|
assertEquals(tabWithoutReaderState.content.url, tabs[0].url)
|
|
assertEquals(tabWithInactiveReaderState.content.url, tabs[1].url)
|
|
assertEquals("https://blog.mozilla.org/123", tabs[2].url)
|
|
}
|
|
}
|