fenix/app/src/test/java/org/mozilla/fenix/utils/ToolbarPopupWindowTest.kt

71 lines
3.1 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.utils
import kotlinx.coroutines.test.TestCoroutineDispatcher
import mozilla.components.browser.state.action.ContentAction
import mozilla.components.browser.state.state.BrowserState
import mozilla.components.browser.state.state.ReaderState
import mozilla.components.browser.state.state.createCustomTab
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.Assert.assertEquals
import org.junit.Rule
import org.junit.Test
import org.junit.runner.RunWith
import org.mozilla.fenix.helpers.FenixRobolectricTestRunner
@RunWith(FenixRobolectricTestRunner::class)
class ToolbarPopupWindowTest {
private val testDispatcher = TestCoroutineDispatcher()
@get:Rule
val coroutinesTestRule = MainCoroutineRule(testDispatcher)
@Test
fun `getUrlForClipboard should get the right URL`() {
// Custom tab
val customTabSession = createCustomTab("https://mozilla.org")
var store = BrowserStore(BrowserState(customTabs = listOf(customTabSession)))
assertEquals(
"https://mozilla.org",
ToolbarPopupWindow.getUrlForClipboard(store, customTabSession.id)
)
// Regular tab
val regularTab = createTab(url = "http://firefox.com")
store = BrowserStore(BrowserState(tabs = listOf(regularTab), selectedTabId = regularTab.id))
assertEquals("http://firefox.com", ToolbarPopupWindow.getUrlForClipboard(store))
// Reader Tab
val readerTab = createTab(
url = "moz-extension://1234",
readerState = ReaderState(active = true, activeUrl = "https://blog.mozilla.org/123")
)
store = BrowserStore(BrowserState(tabs = listOf(readerTab), selectedTabId = readerTab.id))
assertEquals("https://blog.mozilla.org/123", ToolbarPopupWindow.getUrlForClipboard(store))
}
@Test
fun `getUrlForClipboard should get the updated URL`() {
// Custom tab
val customTabSession = createCustomTab("https://mozilla.org")
var store = BrowserStore(BrowserState(customTabs = listOf(customTabSession)))
store.dispatch(ContentAction.UpdateUrlAction(customTabSession.id, "https://firefox.com")).joinBlocking()
assertEquals(
"https://firefox.com",
ToolbarPopupWindow.getUrlForClipboard(store, customTabSession.id)
)
// Regular tab
val regularTab = createTab(url = "http://firefox.com")
store = BrowserStore(BrowserState(tabs = listOf(regularTab), selectedTabId = regularTab.id))
store.dispatch(ContentAction.UpdateUrlAction(regularTab.id, "https://mozilla.org")).joinBlocking()
assertEquals("https://mozilla.org", ToolbarPopupWindow.getUrlForClipboard(store))
}
}