KtIrc/src/test/kotlin/com/dmdirc/ktirc/events/UserStateHandlerTest.kt

package com.dmdirc.ktirc.events

import com.dmdirc.ktirc.IrcClient
import com.dmdirc.ktirc.TestConstants
import com.dmdirc.ktirc.io.CaseMapping
import com.dmdirc.ktirc.model.*
import com.nhaarman.mockitokotlin2.argForWhich
import com.nhaarman.mockitokotlin2.doReturn
import com.nhaarman.mockitokotlin2.mock
import kotlinx.coroutines.runBlocking
import org.junit.jupiter.api.Assertions.*
import org.junit.jupiter.api.BeforeEach
import org.junit.jupiter.api.Test

internal class UserStateHandlerTest {

    private val serverState = ServerState("", "")
    private val userState = UserState { CaseMapping.Rfc }

    private val ircClient = mock<IrcClient> {
        on { serverState } doReturn serverState
        on { userState } doReturn userState
        on { isLocalUser(argForWhich<User> { nickname == "zeroCool" }) } doReturn true
        on { isLocalUser("zeroCool") } doReturn true
    }

    private val handler = UserStateHandler()

    @BeforeEach
    fun setUp() {
        serverState.features[ServerFeature.ModePrefixes] = ModePrefixMapping("ov", "@+")
    }

    @Test
    fun `adds channel to user on join`() {
        runBlocking {
            userState += User("acidBurn")

            handler.processEvent(ircClient, ChannelJoined(TestConstants.time, User("acidBurn", "libby", "root.localhost"), "#thegibson"))

            assertEquals(listOf("#thegibson"), userState["acidBurn"]?.channels?.toList())
        }
    }

    @Test
    fun `updates user info on join`() {
        runBlocking {
            userState += User("acidBurn")

            handler.processEvent(ircClient, ChannelJoined(TestConstants.time, User("acidBurn", "libby", "root.localhost"), "#thegibson"))

            val details = userState["acidBurn"]?.details!!
            assertEquals("libby", details.ident)
            assertEquals("root.localhost", details.hostname)
        }
    }

    @Test
    fun `removes channel from user on part`() {
        runBlocking {
            userState += User("acidBurn")
            userState.addToChannel(User("acidBurn"), "#thegibson")
            userState.addToChannel(User("acidBurn"), "#dumpsterdiving")

            handler.processEvent(ircClient, ChannelParted(TestConstants.time, User("acidBurn", "libby", "root.localhost"), "#dumpsterdiving"))

            assertEquals(listOf("#thegibson"), userState["acidBurn"]?.channels?.toList())
        }
    }

    @Test
    fun `removes user on part from last channel`() {
        runBlocking {
            userState += User("acidBurn")
            userState.addToChannel(User("acidBurn"), "#dumpsterdiving")

            handler.processEvent(ircClient, ChannelParted(TestConstants.time, User("acidBurn", "libby", "root.localhost"), "#dumpsterdiving"))

            assertNull(userState["acidBurn"])
        }
    }

    @Test
    fun `removes channel from all users on local part`() {
        runBlocking {
            userState += User("acidBurn")
            userState.addToChannel(User("acidBurn"), "#dumpsterdiving")
            userState.addToChannel(User("acidBurn"), "#thegibson")

            userState += User("zeroCool")
            userState.addToChannel(User("zeroCool"), "#dumpsterdiving")
            userState.addToChannel(User("zeroCool"), "#thegibson")

            handler.processEvent(ircClient, ChannelParted(TestConstants.time, User("zeroCool", "dade", "root.localhost"), "#dumpsterdiving"))

            assertEquals(listOf("#thegibson"), userState["acidBurn"]?.channels?.toList())
            assertEquals(listOf("#thegibson"), userState["zeroCool"]?.channels?.toList())
        }
    }

    @Test
    fun `removes remote users with no remaining channels on local part`() {
        runBlocking {
            userState += User("acidBurn")
            userState.addToChannel(User("acidBurn"), "#dumpsterdiving")

            handler.processEvent(ircClient, ChannelParted(TestConstants.time, User("zeroCool", "dade", "root.localhost"), "#dumpsterdiving"))

            assertNull(userState["acidBurn"])
        }
    }

    @Test
    fun `keeps local user with no remaining channels after local part`() {
        runBlocking {
            userState += User("zeroCool")
            userState.addToChannel(User("zeroCool"), "#dumpsterdiving")

            handler.processEvent(ircClient, ChannelParted(TestConstants.time, User("zeroCool", "dade", "root.localhost"), "#dumpsterdiving"))

            assertNotNull(userState["zeroCool"])
        }
    }


    @Test
    fun `removes channel from user on kick`() {
        runBlocking {
            userState += User("acidBurn")
            userState.addToChannel(User("acidBurn"), "#thegibson")
            userState.addToChannel(User("acidBurn"), "#dumpsterdiving")

            handler.processEvent(ircClient, ChannelUserKicked(TestConstants.time, User("thePlague"), "#dumpsterdiving", "acidBurn"))

            assertEquals(listOf("#thegibson"), userState["acidBurn"]?.channels?.toList())
        }
    }

    @Test
    fun `removes user on kick from last channel`() {
        runBlocking {
            userState += User("acidBurn")
            userState.addToChannel(User("acidBurn"), "#dumpsterdiving")

            handler.processEvent(ircClient, ChannelUserKicked(TestConstants.time, User("thePlague"), "#dumpsterdiving", "acidBurn"))

            assertNull(userState["acidBurn"])
        }
    }

    @Test
    fun `removes channel from all users on local kick`() {
        runBlocking {
            userState += User("acidBurn")
            userState.addToChannel(User("acidBurn"), "#dumpsterdiving")
            userState.addToChannel(User("acidBurn"), "#thegibson")

            userState += User("zeroCool")
            userState.addToChannel(User("zeroCool"), "#dumpsterdiving")
            userState.addToChannel(User("zeroCool"), "#thegibson")

            handler.processEvent(ircClient, ChannelUserKicked(TestConstants.time, User("thePlague"), "#dumpsterdiving", "zeroCool"))

            assertEquals(listOf("#thegibson"), userState["acidBurn"]?.channels?.toList())
            assertEquals(listOf("#thegibson"), userState["zeroCool"]?.channels?.toList())
        }
    }

    @Test
    fun `removes remote users with no remaining channels on local kick`() {
        runBlocking {
            userState += User("acidBurn")
            userState.addToChannel(User("acidBurn"), "#dumpsterdiving")

            handler.processEvent(ircClient, ChannelUserKicked(TestConstants.time, User("thePlague"), "#dumpsterdiving", "zeroCool"))

            assertNull(userState["acidBurn"])
        }
    }

    @Test
    fun `keeps local user with no remaining channels after local kick`() {
        runBlocking {
            userState += User("zeroCool")
            userState.addToChannel(User("zeroCool"), "#dumpsterdiving")

            handler.processEvent(ircClient, ChannelUserKicked(TestConstants.time, User("thePlague"), "#dumpsterdiving", "zeroCool"))

            assertNotNull(userState["zeroCool"])
        }
    }

    @Test
    fun `removes user entirely on quit`() {
        runBlocking {
            userState += User("acidBurn")
            userState.addToChannel(User("acidBurn"), "#dumpsterdiving")

            handler.processEvent(ircClient, UserQuit(TestConstants.time, User("acidBurn", "libby", "root.localhost")))

            assertNull(userState["acidBurn"])
        }
    }

    @Test
    fun `adds users to channels on names received`() {
        runBlocking {
            userState += User("acidBurn")
            userState.addToChannel(User("acidBurn"), "#dumpsterdiving")

            handler.processEvent(ircClient, ChannelNamesReceived(TestConstants.time, "#thegibson", listOf("@acidBurn")))

            assertEquals(listOf("#dumpsterdiving", "#thegibson"), userState["acidBurn"]?.channels?.toList())
        }
    }

    @Test
    fun `updates user details on names received`() {
        runBlocking {
            userState += User("acidBurn")
            userState.addToChannel(User("acidBurn"), "#dumpsterdiving")

            handler.processEvent(ircClient, ChannelNamesReceived(TestConstants.time, "#thegibson", listOf("@acidBurn!libby@root.localhost")))

            val details = userState["acidBurn"]?.details!!
            assertEquals("libby", details.ident)
            assertEquals("root.localhost", details.hostname)
        }
    }

    @Test
    fun `updates user info on account change`() {
        runBlocking {
            userState += User("acidBurn")

            handler.processEvent(ircClient, UserAccountChanged(TestConstants.time, User("acidBurn", "libby", "root.localhost"), "acidBurn"))

            val details = userState["acidBurn"]?.details!!
            assertEquals("acidBurn", details.account)
        }
    }
}