packages/SystemUI/tests/src/com/android/keyguard/KeyguardStatusViewTest.kt

package com.android.keyguard

import android.test.suitebuilder.annotation.SmallTest
import android.testing.AndroidTestingRunner
import android.testing.TestableLooper.RunWithLooper
import android.view.LayoutInflater
import android.view.View
import android.view.ViewGroup
import com.android.systemui.R
import com.android.systemui.SysuiTestCase
import com.android.systemui.util.children
import com.google.common.truth.Truth.assertThat
import org.junit.Before
import org.junit.Test
import org.junit.runner.RunWith

@SmallTest
@RunWith(AndroidTestingRunner::class)
@RunWithLooper(setAsMainLooper = true)
class KeyguardStatusViewTest : SysuiTestCase() {

    private lateinit var keyguardStatusView: KeyguardStatusView
    private val mediaView: View
        get() = keyguardStatusView.findViewById(R.id.status_view_media_container)
    private val statusViewContainer: ViewGroup
        get() = keyguardStatusView.findViewById(R.id.status_view_container)
    private val clockView: ViewGroup
        get() = keyguardStatusView.findViewById(R.id.keyguard_clock_container)
    private val childrenExcludingMedia
        get() = statusViewContainer.children.filter { it != mediaView }

    @Before
    fun setUp() {
        keyguardStatusView = LayoutInflater.from(context)
                .inflate(R.layout.keyguard_status_view, /* root= */ null) as KeyguardStatusView
    }

    @Test
    fun setChildrenTranslationYExcludingMediaView_mediaViewIsNotTranslated() {
        val translationY = 1234f

        keyguardStatusView.setChildrenTranslationY(translationY, /* excludeMedia= */true)

        assertThat(mediaView.translationY).isEqualTo(0)

        childrenExcludingMedia.forEach {
            assertThat(it.translationY).isEqualTo(translationY)
        }
    }

    @Test
    fun setChildrenTranslationYIncludeMediaView() {
        val translationY = 1234f

        keyguardStatusView.setChildrenTranslationY(translationY, /* excludeMedia= */false)

        statusViewContainer.children.forEach {
            assertThat(it.translationY).isEqualTo(translationY)
        }
    }

    @Test
    fun setAlphaExcludeClock() {
        keyguardStatusView.setAlpha(0.5f, /* excludeClock= */true)
        assertThat(statusViewContainer.alpha).isNotEqualTo(0.5f)
        assertThat(mediaView.alpha).isEqualTo(0.5f)
        assertThat(clockView.alpha).isNotEqualTo(0.5f)
    }
}