/* 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 mozilla.components.browser.storage.sync

import androidx.test.ext.junit.runners.AndroidJUnit4
import kotlinx.coroutines.ExperimentalCoroutinesApi
import kotlinx.coroutines.test.runTest
import mozilla.appservices.remotetabs.ClientRemoteTabs
import mozilla.appservices.remotetabs.DeviceType
import mozilla.appservices.remotetabs.RemoteTab
import mozilla.components.concept.base.crash.CrashReporting
import mozilla.components.support.test.any
import mozilla.components.support.test.mock
import mozilla.components.support.test.robolectric.testContext
import org.junit.After
import org.junit.Assert.assertEquals
import org.junit.Before
import org.junit.Test
import org.junit.runner.RunWith
import org.mockito.Mockito.`when`
import org.mockito.Mockito.mock
import org.mockito.Mockito.spy
import org.mockito.Mockito.times
import org.mockito.Mockito.verify
import mozilla.appservices.remotetabs.InternalException as RemoteTabProviderException
import mozilla.appservices.remotetabs.TabsStore as RemoteTabsProvider

@ExperimentalCoroutinesApi
@RunWith(AndroidJUnit4::class)
class RemoteTabsStorageTest {
    private lateinit var remoteTabs: RemoteTabsStorage
    private lateinit var apiMock: RemoteTabsProvider
    private lateinit var crashReporter: CrashReporting

    @Before
    fun setup() {
        crashReporter = mock()
        remoteTabs = spy(RemoteTabsStorage(testContext, crashReporter))
        apiMock = mock(RemoteTabsProvider::class.java)
        `when`(remoteTabs.api).thenReturn(apiMock)
    }

    @After
    fun cleanup() {
        remoteTabs.cleanup()
    }

    @Test
    fun `store() translates tabs to rust format`() = runTest {
        remoteTabs.store(
            listOf(
                Tab(
                    listOf(
                        TabEntry("Bar", "https://bar", null)
                    ),
                    0, 1574458165555
                ),
                Tab(
                    listOf(
                        TabEntry("Foo bar", "https://foo.bar", null),
                        TabEntry("Foo bar 1", "https://foo.bar/1", null),
                        TabEntry("Foo bar 2", "https://foo.bar/2", null)
                    ),
                    2, 0
                ),
                Tab(
                    listOf(
                        TabEntry("Foo 1", "https://foo", "https://foo/icon"),
                        TabEntry("Foo 2", "https://foo/1", "https://foo/icon2"),
                        TabEntry("Foo 3", "https://foo/1/1", "https://foo/icon3")
                    ),
                    1, 1574457405635
                )
            )
        )

        verify(apiMock).setLocalTabs(
            listOf(
                RemoteTab("Bar", listOf("https://bar"), null, 1574458165555),
                RemoteTab("Foo bar 2", listOf("https://foo.bar/2", "https://foo.bar/1", "https://foo.bar"), null, 0),
                RemoteTab("Foo 2", listOf("https://foo/1", "https://foo"), "https://foo/icon2", 1574457405635)
            )
        )
    }

    @Test
    fun `getAll() translates tabs to our format`() = runTest {
        `when`(apiMock.getAll()).thenReturn(
            listOf(
                ClientRemoteTabs(
                    "client1", "", DeviceType.MOBILE,
                    listOf(
                        RemoteTab("Foo", listOf("https://foo/1/1", "https://foo/1", "https://foo"), "https://foo/icon", 1574457405635)
                    )
                ),
                ClientRemoteTabs(
                    "client2", "", DeviceType.MOBILE,
                    listOf(
                        RemoteTab("Bar", listOf("https://bar"), null, 1574458165555),
                        RemoteTab("Foo Bar", listOf("https://foo.bar"), "https://foo.bar/icon", 0)
                    )
                )
            )
        )

        assertEquals(
            mapOf(
                SyncClient("client1") to listOf(
                    Tab(
                        listOf(
                            TabEntry("Foo", "https://foo", "https://foo/icon"),
                            TabEntry("Foo", "https://foo/1", "https://foo/icon"),
                            TabEntry("Foo", "https://foo/1/1", "https://foo/icon")
                        ),
                        2, 1574457405635
                    )
                ),
                SyncClient("client2") to listOf(
                    Tab(
                        listOf(
                            TabEntry("Bar", "https://bar", null)
                        ),
                        0, 1574458165555
                    ),
                    Tab(
                        listOf(
                            TabEntry("Foo Bar", "https://foo.bar", "https://foo.bar/icon")
                        ),
                        0, 0
                    )
                )
            ),
            remoteTabs.getAll()
        )
    }

    @Test
    fun `exceptions from getAll are propagated to the crash reporter`() = runTest {
        val throwable = RemoteTabProviderException("test")
        `when`(apiMock.getAll()).thenAnswer { throw throwable }

        remoteTabs.getAll()

        verify(crashReporter).submitCaughtException(throwable)

        `when`(apiMock.setLocalTabs(any())).thenAnswer { throw throwable }

        remoteTabs.store(emptyList())

        verify(crashReporter, times(2)).submitCaughtException(throwable)

        Unit
    }
}
