packages/SystemUI/tests/src/com/android/systemui/qs/QuickStatusBarHeaderControllerTest.kt

/*
 * Copyright (C) 2020 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 com.android.systemui.qs

import android.content.Context
import android.testing.AndroidTestingRunner
import android.view.View
import androidx.test.filters.SmallTest
import com.android.systemui.R
import com.android.systemui.SysuiTestCase
import com.android.systemui.battery.BatteryMeterViewController
import com.android.systemui.colorextraction.SysuiColorExtractor
import com.android.systemui.demomode.DemoModeController
import com.android.systemui.flags.FeatureFlags
import com.android.systemui.qs.carrier.QSCarrierGroup
import com.android.systemui.qs.carrier.QSCarrierGroupController
import com.android.systemui.statusbar.phone.StatusBarContentInsetsProvider
import com.android.systemui.statusbar.phone.StatusBarIconController
import com.android.systemui.statusbar.phone.StatusIconContainer
import com.android.systemui.statusbar.policy.Clock
import com.android.systemui.statusbar.policy.VariableDateView
import com.android.systemui.statusbar.policy.VariableDateViewController
import com.android.systemui.util.mockito.any
import com.android.systemui.util.mockito.argumentCaptor
import com.android.systemui.util.mockito.capture
import com.google.common.truth.Truth.assertThat
import org.junit.After
import org.junit.Before
import org.junit.Test
import org.junit.runner.RunWith
import org.mockito.Answers
import org.mockito.ArgumentMatchers.anyInt
import org.mockito.Mock
import org.mockito.Mockito.anyBoolean
import org.mockito.Mockito.reset
import org.mockito.Mockito.verify
import org.mockito.Mockito.`when`
import org.mockito.MockitoAnnotations

@SmallTest
@RunWith(AndroidTestingRunner::class)
class QuickStatusBarHeaderControllerTest : SysuiTestCase() {

    @Mock
    private lateinit var view: QuickStatusBarHeader
    @Mock
    private lateinit var privacyIconsController: HeaderPrivacyIconsController
    @Mock
    private lateinit var statusBarIconController: StatusBarIconController
    @Mock
    private lateinit var demoModeController: DemoModeController
    @Mock
    private lateinit var quickQSPanelController: QuickQSPanelController
    @Mock(answer = Answers.RETURNS_SELF)
    private lateinit var qsCarrierGroupControllerBuilder: QSCarrierGroupController.Builder
    @Mock
    private lateinit var qsCarrierGroupController: QSCarrierGroupController
    @Mock
    private lateinit var colorExtractor: SysuiColorExtractor
    @Mock
    private lateinit var iconContainer: StatusIconContainer
    @Mock
    private lateinit var qsCarrierGroup: QSCarrierGroup
    @Mock
    private lateinit var variableDateViewControllerFactory: VariableDateViewController.Factory
    @Mock
    private lateinit var variableDateViewController: VariableDateViewController
    @Mock
    private lateinit var batteryMeterViewController: BatteryMeterViewController
    @Mock
    private lateinit var clock: Clock
    @Mock
    private lateinit var variableDateView: VariableDateView
    @Mock
    private lateinit var mockView: View
    @Mock(answer = Answers.RETURNS_DEEP_STUBS)
    private lateinit var context: Context
    @Mock
    private lateinit var featureFlags: FeatureFlags
    @Mock
    private lateinit var insetsProvider: StatusBarContentInsetsProvider
    @Mock
    private lateinit var iconManagerFactory: StatusBarIconController.TintedIconManager.Factory
    @Mock
    private lateinit var iconManager: StatusBarIconController.TintedIconManager

    private val qsExpansionPathInterpolator = QSExpansionPathInterpolator()

    private lateinit var controller: QuickStatusBarHeaderController

    @Before
    fun setUp() {
        MockitoAnnotations.initMocks(this)
        stubViews()
        `when`(iconContainer.context).thenReturn(context)
        `when`(qsCarrierGroupControllerBuilder.build()).thenReturn(qsCarrierGroupController)
        `when`(variableDateViewControllerFactory.create(any()))
                .thenReturn(variableDateViewController)
        `when`(iconManagerFactory.create(any())).thenReturn(iconManager)
        `when`(view.resources).thenReturn(mContext.resources)
        `when`(view.isAttachedToWindow).thenReturn(true)
        `when`(view.context).thenReturn(context)

        controller = QuickStatusBarHeaderController(
                view,
                privacyIconsController,
                statusBarIconController,
                demoModeController,
                quickQSPanelController,
                qsCarrierGroupControllerBuilder,
                colorExtractor,
                qsExpansionPathInterpolator,
                featureFlags,
                variableDateViewControllerFactory,
                batteryMeterViewController,
                insetsProvider,
                iconManagerFactory,
        )
    }

    @After
    fun tearDown() {
        controller.onViewDetached()
    }

    @Test
    fun testClockNotClickable() {
        assertThat(clock.isClickable).isFalse()
    }

    @Test
    fun testSingleCarrierListenerAttachedOnInit() {
        controller.init()

        verify(qsCarrierGroupController).setOnSingleCarrierChangedListener(any())
    }

    @Test
    fun testSingleCarrierSetOnViewOnInit_false() {
        `when`(qsCarrierGroupController.isSingleCarrier).thenReturn(false)
        controller.init()

        verify(view).setIsSingleCarrier(false)
    }

    @Test
    fun testSingleCarrierSetOnViewOnInit_true() {
        `when`(qsCarrierGroupController.isSingleCarrier).thenReturn(true)
        controller.init()

        verify(view).setIsSingleCarrier(true)
    }

    @Test
    fun testRSSISlot_notCombined() {
        controller.init()

        val captor = argumentCaptor<List<String>>()
        verify(view).onAttach(any(), any(), capture(captor), any(), anyBoolean())

        assertThat(captor.value).containsExactly(
            mContext.getString(com.android.internal.R.string.status_bar_mobile)
        )
    }

    @Test
    fun testSingleCarrierCallback() {
        controller.init()
        reset(view)

        val captor = argumentCaptor<QSCarrierGroupController.OnSingleCarrierChangedListener>()
        verify(qsCarrierGroupController).setOnSingleCarrierChangedListener(capture(captor))

        captor.value.onSingleCarrierChanged(true)
        verify(view).setIsSingleCarrier(true)

        captor.value.onSingleCarrierChanged(false)
        verify(view).setIsSingleCarrier(false)
    }

    @Test
    fun testAlarmIconIgnored() {
        controller.init()

        verify(iconContainer).addIgnoredSlot(
                mContext.getString(com.android.internal.R.string.status_bar_alarm_clock))
    }

    private fun stubViews() {
        `when`(view.findViewById<View>(anyInt())).thenReturn(mockView)
        `when`(view.findViewById<QSCarrierGroup>(R.id.carrier_group)).thenReturn(qsCarrierGroup)
        `when`(view.findViewById<StatusIconContainer>(R.id.statusIcons)).thenReturn(iconContainer)
        `when`(view.findViewById<Clock>(R.id.clock)).thenReturn(clock)
        `when`(view.requireViewById<VariableDateView>(R.id.date)).thenReturn(variableDateView)
        `when`(view.requireViewById<VariableDateView>(R.id.date_clock)).thenReturn(variableDateView)
    }
}