/* 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 androidx.test.ext.junit.runners.AndroidJUnit4
import assertk.assertThat
import assertk.assertions.isNotNull
import assertk.assertions.isNull
import assertk.assertions.isTrue
import mozilla.components.support.test.robolectric.createAddedTestFragment
import io.mockk.MockKAnnotations
import io.mockk.every
import io.mockk.impl.annotations.MockK
import kotlinx.coroutines.ExperimentalCoroutinesApi
import kotlinx.coroutines.GlobalScope
import kotlinx.coroutines.ObsoleteCoroutinesApi
import kotlinx.coroutines.async
import mozilla.components.browser.session.Session
import mozilla.components.browser.session.SessionManager
import mozilla.components.feature.tab.collections.Tab
import mozilla.components.lib.publicsuffixlist.PublicSuffixList
import org.junit.Assert.assertEquals
import org.junit.Before
import org.junit.Test
import org.junit.runner.RunWith
import org.mozilla.fenix.TestApplication
import org.robolectric.annotation.Config

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
@ObsoleteCoroutinesApi
@RunWith(AndroidJUnit4::class)
@Config(application = TestApplication::class)
class CollectionCreationFragmentTest {

    @MockK private lateinit var sessionManager: SessionManager
    @MockK private lateinit var publicSuffixList: PublicSuffixList

    private val sessionMozilla = Session(initialUrl = URL_MOZILLA, id = SESSION_ID_MOZILLA)
    private val sessionBcc = Session(initialUrl = URL_BCC, id = SESSION_ID_BCC)

    @Before
    fun before() {
        MockKAnnotations.init(this)
        every { sessionManager.findSessionById(SESSION_ID_MOZILLA) } answers { sessionMozilla }
        every { sessionManager.findSessionById(SESSION_ID_BCC) } answers { sessionBcc }
        every { sessionManager.findSessionById(SESSION_ID_BAD_1) } answers { null }
        every { sessionManager.findSessionById(SESSION_ID_BAD_2) } answers { null }
        every { publicSuffixList.stripPublicSuffix(URL_MOZILLA) } answers { GlobalScope.async { URL_MOZILLA } }
        every { publicSuffixList.stripPublicSuffix(URL_BCC) } answers { GlobalScope.async { URL_BCC } }
    }

    @Test
    fun `creation dialog shows and can be dismissed`() {
        val fragment = createAddedTestFragment {
            CollectionCreationFragment().apply {
                arguments = CollectionCreationFragmentArgs(
                    // Fragment crashes if navArgs is null
                    previousFragmentId = 0,
                    saveCollectionStep = SaveCollectionStep.SelectTabs
                ).toBundle()
            }
        }

        assertThat(fragment.dialog).isNotNull()
        assertThat(fragment.requireDialog().isShowing).isTrue()
        fragment.dismiss()
        assertThat(fragment.dialog).isNull()
    }

    @Test
    fun `GIVEN tabs are present in session manager WHEN getTabs is called THEN tabs will be returned`() {
        val tabs = sessionManager
            .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 session manager WHEN getTabs is called THEN only valid tabs will be returned`() {
        val tabs = sessionManager
            .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 session manager WHEN getTabs is called THEN an empty list will be returned`() {
        val tabs = sessionManager
            .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 = sessionManager
            .getTabs(null, publicSuffixList)

        assertEquals(emptyList<Tab>(), tabs)
    }
}