/*
 * Copyright (C) 2023 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package android.tools.traces.events

import android.tools.Timestamps
import android.tools.utils.CleanFlickerEnvironmentRule
import com.google.common.truth.Truth
import org.junit.ClassRule
import org.junit.Test

/** Tests for [CujTrace]. Run with `atest FlickerLibTest:CujTraceTest` */
class CujTraceTest {
    @Test
    fun canCreateFromListOfCujEvents() {
        val trace =
            CujTrace.from(
                listOf(
                    createCujEvent(
                        1,
                        CujType.CUJ_LAUNCHER_ALL_APPS_SCROLL,
                        CujEvent.JANK_CUJ_BEGIN_TAG
                    ),
                    createCujEvent(
                        2,
                        CujType.CUJ_LAUNCHER_ALL_APPS_SCROLL,
                        CujEvent.JANK_CUJ_END_TAG
                    ),
                )
            )

        Truth.assertThat(trace.entries).hasSize(1)
        Truth.assertThat(trace.entries.first().cuj).isEqualTo(CujType.CUJ_LAUNCHER_ALL_APPS_SCROLL)
        Truth.assertThat(trace.entries.first().startTimestamp.unixNanos).isEqualTo(1)
        Truth.assertThat(trace.entries.first().endTimestamp.unixNanos).isEqualTo(2)
        Truth.assertThat(trace.entries.first().canceled).isFalse()
    }

    @Test
    fun canCreateFromListOfCujEventsWithTags() {
        val trace =
            CujTrace.from(
                listOf(
                    createCujEvent(
                        1,
                        CujType.CUJ_LAUNCHER_ALL_APPS_SCROLL,
                        CujEvent.JANK_CUJ_BEGIN_TAG,
                        tag = "MySubType"
                    ),
                    createCujEvent(
                        2,
                        CujType.CUJ_LAUNCHER_ALL_APPS_SCROLL,
                        CujEvent.JANK_CUJ_END_TAG
                    ),
                )
            )

        Truth.assertThat(trace.entries).hasSize(1)
        Truth.assertThat(trace.entries.first().cuj).isEqualTo(CujType.CUJ_LAUNCHER_ALL_APPS_SCROLL)
        Truth.assertThat(trace.entries.first().tag).isEqualTo("MySubType")
        Truth.assertThat(trace.entries.first().startTimestamp.unixNanos).isEqualTo(1)
        Truth.assertThat(trace.entries.first().endTimestamp.unixNanos).isEqualTo(2)
        Truth.assertThat(trace.entries.first().canceled).isFalse()
    }

    @Test
    fun canCreateCanceledCujsFromListOfCujEvents() {
        val trace =
            CujTrace.from(
                listOf(
                    createCujEvent(
                        1,
                        CujType.CUJ_LAUNCHER_ALL_APPS_SCROLL,
                        CujEvent.JANK_CUJ_BEGIN_TAG
                    ),
                    createCujEvent(
                        2,
                        CujType.CUJ_LAUNCHER_ALL_APPS_SCROLL,
                        CujEvent.JANK_CUJ_CANCEL_TAG
                    ),
                )
            )

        Truth.assertThat(trace.entries).hasSize(1)
        Truth.assertThat(trace.entries.first().cuj).isEqualTo(CujType.CUJ_LAUNCHER_ALL_APPS_SCROLL)
        Truth.assertThat(trace.entries.first().startTimestamp.unixNanos).isEqualTo(1)
        Truth.assertThat(trace.entries.first().endTimestamp.unixNanos).isEqualTo(2)
        Truth.assertThat(trace.entries.first().canceled).isTrue()
    }

    @Test
    fun canHandleIncompleteCujs() {
        val trace =
            CujTrace.from(
                listOf(
                    createCujEvent(
                        1,
                        CujType.CUJ_LAUNCHER_ALL_APPS_SCROLL,
                        CujEvent.JANK_CUJ_CANCEL_TAG
                    ),
                    createCujEvent(
                        2,
                        CujType.CUJ_BIOMETRIC_PROMPT_TRANSITION,
                        CujEvent.JANK_CUJ_END_TAG
                    ),
                    createCujEvent(
                        3,
                        CujType.CUJ_LAUNCHER_APP_CLOSE_TO_HOME,
                        CujEvent.JANK_CUJ_BEGIN_TAG
                    ),
                )
            )

        Truth.assertThat(trace.entries).isEmpty()
    }

    @Test
    fun canHandleOutOfOrderEntries() {
        val trace =
            CujTrace.from(
                listOf(
                    createCujEvent(
                        2,
                        CujType.CUJ_LAUNCHER_ALL_APPS_SCROLL,
                        CujEvent.JANK_CUJ_END_TAG
                    ),
                    createCujEvent(
                        1,
                        CujType.CUJ_LAUNCHER_ALL_APPS_SCROLL,
                        CujEvent.JANK_CUJ_BEGIN_TAG
                    ),
                )
            )

        Truth.assertThat(trace.entries).hasSize(1)
        Truth.assertThat(trace.entries.first().cuj).isEqualTo(CujType.CUJ_LAUNCHER_ALL_APPS_SCROLL)
        Truth.assertThat(trace.entries.first().startTimestamp.unixNanos).isEqualTo(1)
        Truth.assertThat(trace.entries.first().endTimestamp.unixNanos).isEqualTo(2)
        Truth.assertThat(trace.entries.first().canceled).isFalse()
    }

    @Test
    fun canHandleMissingStartAndEnds() {
        val trace =
            CujTrace.from(
                listOf(
                    createCujEvent(
                        1,
                        CujType.CUJ_LAUNCHER_ALL_APPS_SCROLL,
                        CujEvent.JANK_CUJ_END_TAG
                    ),
                    createCujEvent(
                        2,
                        CujType.CUJ_LAUNCHER_ALL_APPS_SCROLL,
                        CujEvent.JANK_CUJ_BEGIN_TAG
                    ),
                )
            )

        Truth.assertThat(trace.entries).isEmpty()
    }

    private fun createCujEvent(
        timestamp: Long,
        cuj: CujType,
        type: String,
        tag: String? = null
    ): CujEvent {
        return CujEvent(
            Timestamps.from(unixNanos = timestamp),
            cuj,
            0,
            "root",
            0,
            type,
            tag,
        )
    }

    companion object {
        @ClassRule @JvmField val ENV_CLEANUP = CleanFlickerEnvironmentRule()
    }
}