package io.oxygenhub.android.ui

import io.oxygenhub.android.FakeHubApi

import io.oxygenhub.android.MainDispatcherRule

import io.oxygenhub.android.api.FlowJob

import io.oxygenhub.android.api.FlowRun

import io.oxygenhub.android.api.LogLine

import io.oxygenhub.android.ui.common.Load

import io.oxygenhub.android.ui.flow.JobLogViewModel

import io.oxygenhub.android.ui.flow.RunViewModel

import io.oxygenhub.android.ui.flow.RunsViewModel

import kotlinx.coroutines.test.runTest

import org.junit.Assert.assertEquals

import org.junit.Assert.assertFalse

import org.junit.Assert.assertTrue

import org.junit.Rule

import org.junit.Test

class FlowViewModelsTest {

    @get:Rule val main = MainDispatcherRule()

    private val api = FakeHubApi().apply {

        runList = listOf(FlowRun(56, "main", "passed"), FlowRun(55, "main", "failed"))

        runStore[56] = FlowRun(56, "main", "running", jobs = listOf(FlowJob("count", "running", runner = "android-pixel-10-pro-fold")))

        logs["count"] = mutableListOf(LogLine(1, 0, "flow", "fetching"), LogLine(2, 1800, "stdout", "step 1 of 120"))

    }

    @Test

    fun listsTheRepositorysRuns() {

        val runs = (RunsViewModel(api, "andymccall", "phone-runner-demo").state.value as Load.Ready).value

        assertEquals(listOf(56L, 55L), runs.map { it.number })

    }

    @Test

    fun aRunFollowsItsJobsAndCanBeCancelled() = runTest {

        val vm = RunViewModel(api, "andymccall", "phone-runner-demo", 56, pollMs = 0)

        assertTrue((vm.state.value as Load.Ready).value.live)

        api.runStore[56] = api.runStore.getValue(56).copy(status = "passed", jobs = listOf(FlowJob("count", "passed", runner = "plex-1")))

        vm.poll()

        assertEquals("plex-1", (vm.state.value as Load.Ready).value.jobs[0].runner)

        vm.cancel()

        assertTrue("cancel 56" in api.calls)

        vm.rerunJob("count")

        assertTrue("rerun 56 count" in api.calls)

    }

    @Test

    fun aGatedJobIsApprovedOrRejectedAndARunExpedited() = runTest {

        val vm = RunViewModel(api, "andymccall", "phone-runner-demo", 56, pollMs = 0)

        vm.decide("deploy", approve = true)

        vm.decide("deploy", approve = false)

        vm.expedite("  the fix for the outage  ")

        assertEquals(listOf("approve 56 deploy", "reject 56 deploy", "expedite 56 the fix for the outage"), api.calls.filter { it.startsWith("approve") || it.startsWith("reject") || it.startsWith("expedite") })

        val gated = io.oxygenhub.android.api.FlowJob("deploy", status = "queued", approval = "waiting")

        assertTrue(gated.waitsForPerson)

        assertTrue(!gated.copy(approval = "approved").waitsForPerson && !gated.copy(status = "running").waitsForPerson)

    }

    @Test

    fun aLogGrowsFromTheLastLineHeldUntilTheJobEnds() = runTest {

        val vm = JobLogViewModel(api, "andymccall", "phone-runner-demo", 56, "count", pollMs = 0)

        var log = (vm.state.value as Load.Ready).value

        assertEquals(listOf("fetching", "step 1 of 120"), log.lines.map { it.text })

        assertTrue(log.live)

        api.logs.getValue("count") += LogLine(3, 2500, "flow", "── handed back by android-pixel-10-pro-fold: off Wi-Fi")

        api.runStore[56] = api.runStore.getValue(56).copy(status = "running", jobs = listOf(FlowJob("count", "queued")))

        vm.next()

        log = (vm.state.value as Load.Ready).value

        assertEquals(3, log.lines.size)

        assertTrue("log count after 2" in api.calls)

        api.runStore[56] = api.runStore.getValue(56).copy(status = "passed", jobs = listOf(FlowJob("count", "passed")))

        vm.next()

        assertFalse((vm.state.value as Load.Ready).value.live)

        assertEquals("passed", (vm.state.value as Load.Ready).value.status)

    }

}