87 lines
2.7 KiB
Kotlin
87 lines
2.7 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.ext
|
|
|
|
import io.mockk.every
|
|
import io.mockk.mockk
|
|
import mozilla.components.browser.session.Session
|
|
import mozilla.components.browser.session.SessionManager
|
|
import org.junit.Assert.assertEquals
|
|
import org.junit.Test
|
|
|
|
class SessionManagerTest {
|
|
|
|
@Test
|
|
fun `returns all normal sessions`() {
|
|
val sessions = listOf(
|
|
Session("https://example.com", private = false),
|
|
Session("https://mozilla.org", private = true),
|
|
Session("https://github.com", private = false)
|
|
)
|
|
val sessionManager = mockSessionManager(sessions)
|
|
|
|
assertEquals(
|
|
listOf(sessions[0], sessions[2]),
|
|
sessionManager.sessionsOfType(private = false).toList()
|
|
)
|
|
}
|
|
|
|
@Test
|
|
fun `returns all private sessions`() {
|
|
val sessions = listOf(
|
|
Session("https://example.com", private = false),
|
|
Session("https://mozilla.org", private = true),
|
|
Session("https://github.com", private = false)
|
|
)
|
|
val sessionManager = mockSessionManager(sessions)
|
|
|
|
assertEquals(
|
|
listOf(sessions[1]),
|
|
sessionManager.sessionsOfType(private = true).toList()
|
|
)
|
|
}
|
|
|
|
@Test
|
|
fun `normalSessionSize only counts non-private non-custom sessions`() {
|
|
val normal1 = mockSession()
|
|
val normal2 = mockSession()
|
|
val normal3 = mockSession()
|
|
|
|
val private1 = mockSession(isPrivate = true)
|
|
val private2 = mockSession(isPrivate = true)
|
|
|
|
val custom1 = mockSession(isCustom = true)
|
|
val custom2 = mockSession(isCustom = true)
|
|
val custom3 = mockSession(isCustom = true)
|
|
|
|
val privateCustom = mockSession(isPrivate = true, isCustom = true)
|
|
|
|
val sessionManager = mockSessionManager(
|
|
listOf(
|
|
normal1, private1, private2, custom1,
|
|
normal2, normal3, custom2, custom3, privateCustom
|
|
)
|
|
)
|
|
|
|
assertEquals(3, sessionManager.normalSessionSize())
|
|
}
|
|
|
|
private fun mockSessionManager(sessions: List<Session>): SessionManager {
|
|
val sessionManager: SessionManager = mockk()
|
|
every { sessionManager.sessions } returns sessions
|
|
return sessionManager
|
|
}
|
|
|
|
private fun mockSession(
|
|
sessionId: String? = null,
|
|
isPrivate: Boolean = false,
|
|
isCustom: Boolean = false
|
|
) = mockk<Session> {
|
|
sessionId?.let { every { id } returns it }
|
|
every { private } returns isPrivate
|
|
every { isCustomTabSession() } returns isCustom
|
|
}
|
|
}
|