DialogRememberSystemUiControllerTest.kt

/*
 * Copyright 2022 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
 *
 *      https://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.google.accompanist.systemuicontroller

import android.os.Build
import android.view.View
import android.view.Window
import androidx.activity.ComponentActivity
import androidx.compose.runtime.SideEffect
import androidx.compose.ui.graphics.Color
import androidx.compose.ui.platform.LocalView
import androidx.compose.ui.test.junit4.createAndroidComposeRule
import androidx.compose.ui.window.Dialog
import androidx.compose.ui.window.DialogWindowProvider
import androidx.core.view.ViewCompat
import androidx.core.view.WindowCompat
import androidx.core.view.WindowInsetsCompat
import androidx.core.view.WindowInsetsControllerCompat
import androidx.test.ext.junit.runners.AndroidJUnit4
import androidx.test.filters.FlakyTest
import androidx.test.filters.SdkSuppress
import com.google.accompanist.internal.test.IgnoreOnRobolectric
import com.google.accompanist.internal.test.waitUntil
import com.google.accompanist.internal.test.withActivity
import com.google.accompanist.systemuicontroller.test.R
import com.google.common.truth.Truth.assertThat
import org.junit.Before
import org.junit.Rule
import org.junit.Test
import org.junit.experimental.categories.Category
import org.junit.runner.RunWith

@RunWith(AndroidJUnit4::class)
class DialogRememberSystemUiControllerTest {
    @get:Rule
    val rule = createAndroidComposeRule<ComponentActivity>()

    private lateinit var window: Window
    private lateinit var contentView: View

    @Before
    fun setup() {
        rule.activityRule.scenario.onActivity {
            it.setTheme(R.style.DialogSystemUiControllerTheme)
        }
    }

    @Test
    fun statusBarColor() {
        rule.setContent {
            Dialog(onDismissRequest = {}) {
                window = (LocalView.current.parent as DialogWindowProvider).window

                // Create an systemUiController and set the status bar color
                val systemUiController = rememberSystemUiController()
                SideEffect {
                    systemUiController.setStatusBarColor(Color.Blue, darkIcons = false)
                }
            }
        }

        // Assert that the color was set
        assertThat(Color(window.statusBarColor)).isEqualTo(Color.Blue)
    }

    @Test
    fun navigationBarColor() {
        rule.setContent {
            Dialog(onDismissRequest = {}) {
                window = (LocalView.current.parent as DialogWindowProvider).window

                // Now create an systemUiController and set the navigation bar color
                val systemUiController = rememberSystemUiController()
                SideEffect {
                    systemUiController.setNavigationBarColor(Color.Green, darkIcons = false)
                }
            }
        }

        assertThat(Color(window.navigationBarColor)).isEqualTo(Color.Green)
    }

    @Test
    fun systemBarColor() {
        rule.setContent {
            Dialog(onDismissRequest = {}) {
                window = (LocalView.current.parent as DialogWindowProvider).window

                // Now create an systemUiController and set the system bar colors
                val systemUiController = rememberSystemUiController()
                SideEffect {
                    systemUiController.setSystemBarsColor(Color.Red, darkIcons = false)
                }
            }
        }

        // Assert that the colors were set
        assertThat(Color(window.statusBarColor)).isEqualTo(Color.Red)
        assertThat(Color(window.navigationBarColor)).isEqualTo(Color.Red)
    }

    @Test
    @Category(IgnoreOnRobolectric::class) // Robolectric implements the new behavior from 23+
    @SdkSuppress(maxSdkVersion = 22)
    fun statusBarIcons_scrim() {
        // Now create an systemUiController and set the navigation bar with dark icons
        rule.setContent {
            Dialog(onDismissRequest = {}) {
                window = (LocalView.current.parent as DialogWindowProvider).window
                contentView = LocalView.current

                val systemUiController = rememberSystemUiController()
                SideEffect {
                    systemUiController.setStatusBarColor(Color.White, darkIcons = true) {
                        // Here we can provide custom logic to 'darken' the color to maintain contrast.
                        // We return red just to assert below.
                        Color.Red
                    }
                }
            }
        }

        // Assert that the colors were set to our 'darkened' color
        assertThat(Color(window.statusBarColor)).isEqualTo(Color.Red)

        // Assert that the system couldn't apply the native light icons
        rule.activityRule.scenario.onActivity {
            val windowInsetsController = WindowCompat.getInsetsController(window, contentView)
            assertThat(windowInsetsController.isAppearanceLightStatusBars).isFalse()
        }
    }

    @Test
    @Category(IgnoreOnRobolectric::class)
    @SdkSuppress(minSdkVersion = 23)
    fun statusBarIcons_native() {
        // Now create an systemUiController and set the status bar with dark icons
        rule.setContent {
            Dialog(onDismissRequest = {}) {
                window = (LocalView.current.parent as DialogWindowProvider).window
                contentView = LocalView.current

                val systemUiController = rememberSystemUiController()
                SideEffect {
                    systemUiController.setStatusBarColor(Color.White, darkIcons = true) {
                        // Here we can provide custom logic to 'darken' the color to maintain contrast.
                        // We return red just to assert below.
                        Color.Red
                    }
                }
            }
        }

        // Assert that the colors were darkened color is not used
        assertThat(Color(window.statusBarColor)).isEqualTo(Color.White)

        // Assert that the system applied the native light icons
        rule.activityRule.scenario.onActivity {
            val windowInsetsController = WindowCompat.getInsetsController(window, contentView)
            assertThat(windowInsetsController.isAppearanceLightStatusBars).isTrue()
        }
    }

    @Test
    @Category(IgnoreOnRobolectric::class) // Robolectric implements the new behavior from 25+
    @SdkSuppress(maxSdkVersion = 25)
    fun navigationBarIcons_scrim() {
        // Now create an systemUiController and set the navigation bar with dark icons
        rule.setContent {
            Dialog(onDismissRequest = {}) {
                window = (LocalView.current.parent as DialogWindowProvider).window
                contentView = LocalView.current

                val systemUiController = rememberSystemUiController()
                SideEffect {
                    systemUiController.setNavigationBarColor(Color.White, darkIcons = true) {
                        // Here we can provide custom logic to 'darken' the color to maintain contrast.
                        // We return red just to assert below.
                        Color.Red
                    }
                }
            }
        }

        // Assert that the colors were set to our 'darkened' color
        assertThat(Color(window.navigationBarColor)).isEqualTo(Color.Red)

        // Assert that the system couldn't apply the native light icons
        rule.activityRule.scenario.onActivity {
            val windowInsetsController = WindowCompat.getInsetsController(window, contentView)
            assertThat(windowInsetsController.isAppearanceLightNavigationBars).isFalse()
        }
    }

    @Test
    @Category(IgnoreOnRobolectric::class)
    @SdkSuppress(minSdkVersion = 26)
    fun navigationBar_native() {
        // Now create an systemUiController and set the navigation bar with dark icons
        rule.setContent {
            Dialog(onDismissRequest = {}) {
                window = (LocalView.current.parent as DialogWindowProvider).window
                contentView = LocalView.current

                val systemUiController = rememberSystemUiController()
                SideEffect {
                    systemUiController.setNavigationBarColor(Color.White, darkIcons = true) {
                        // Here we can provide custom logic to 'darken' the color to maintain contrast.
                        // We return red just to assert below.
                        Color.Red
                    }
                }
            }
        }

        // Assert that the colors were darkened color is not used
        assertThat(Color(window.navigationBarColor)).isEqualTo(Color.White)

        // Assert that the system applied the native light icons
        rule.activityRule.scenario.onActivity {
            val windowInsetsController = WindowCompat.getInsetsController(window, contentView)
            assertThat(windowInsetsController.isAppearanceLightNavigationBars).isTrue()
        }
    }

    @Test
    @SdkSuppress(minSdkVersion = 29)
    fun navigationBar_contrastEnforced() {
        lateinit var systemUiController: SystemUiController

        rule.setContent {
            Dialog(onDismissRequest = {}) {
                window = (LocalView.current.parent as DialogWindowProvider).window

                systemUiController = rememberSystemUiController()
            }
        }

        if (Build.VERSION.SDK_INT >= 29) {
            // On API 29+, the system can modify the bar colors to maintain contrast.
            // We disable that here to make it simple to assert expected values
            rule.activityRule.scenario.onActivity {
                window.apply {
                    isNavigationBarContrastEnforced = false
                    isStatusBarContrastEnforced = false
                }
            }
        }

        rule.activityRule.scenario.onActivity {
            // Assert that the contrast is not enforced initially
            assertThat(systemUiController.isNavigationBarContrastEnforced).isFalse()

            // and set the navigation bar with dark icons and enforce contrast
            systemUiController.setNavigationBarColor(
                Color.Transparent,
                darkIcons = true,
                navigationBarContrastEnforced = true
            ) {
                // Here we can provide custom logic to 'darken' the color to maintain contrast.
                // We return red just to assert below.
                Color.Red
            }

            // Assert that the colors were darkened color is not used
            assertThat(Color(window.navigationBarColor)).isEqualTo(Color.Transparent)

            // Assert that the system applied the contrast enforced property
            assertThat(window.isNavigationBarContrastEnforced).isTrue()

            // Assert that the controller reflects that the contrast is enforced
            assertThat(systemUiController.isNavigationBarContrastEnforced).isTrue()
        }
    }

    @Test
    @SdkSuppress(minSdkVersion = 30) // TODO: https://issuetracker.google.com/issues/189366125
    fun systemBarsBehavior_showBarsByTouch() {
        lateinit var systemUiController: SystemUiController

        rule.setContent {
            Dialog(onDismissRequest = {}) {
                window = (LocalView.current.parent as DialogWindowProvider).window
                contentView = LocalView.current

                systemUiController = rememberSystemUiController()
            }
        }

        rule.activityRule.scenario.onActivity {
            systemUiController.systemBarsBehavior =
                WindowInsetsControllerCompat.BEHAVIOR_SHOW_BARS_BY_TOUCH
        }

        assertThat(WindowCompat.getInsetsController(window, contentView).systemBarsBehavior)
            .isEqualTo(WindowInsetsControllerCompat.BEHAVIOR_SHOW_BARS_BY_TOUCH)
    }

    @Test
    @SdkSuppress(minSdkVersion = 30) // TODO: https://issuetracker.google.com/issues/189366125
    fun systemBarsBehavior_showBarsBySwipe() {
        lateinit var systemUiController: SystemUiController

        rule.setContent {
            Dialog(onDismissRequest = {}) {
                window = (LocalView.current.parent as DialogWindowProvider).window
                contentView = LocalView.current

                systemUiController = rememberSystemUiController()
            }
        }

        rule.activityRule.scenario.onActivity {
            systemUiController.systemBarsBehavior =
                WindowInsetsControllerCompat.BEHAVIOR_SHOW_BARS_BY_SWIPE
        }

        assertThat(WindowCompat.getInsetsController(window, contentView).systemBarsBehavior)
            .isEqualTo(WindowInsetsControllerCompat.BEHAVIOR_SHOW_BARS_BY_SWIPE)
    }

    @Test
    @SdkSuppress(minSdkVersion = 30) // TODO: https://issuetracker.google.com/issues/189366125
    fun systemBarsBehavior_showTransientBarsBySwipe() {
        lateinit var systemUiController: SystemUiController

        rule.setContent {
            Dialog(onDismissRequest = {}) {
                window = (LocalView.current.parent as DialogWindowProvider).window
                contentView = LocalView.current

                systemUiController = rememberSystemUiController()
            }
        }

        rule.activityRule.scenario.onActivity {
            systemUiController.systemBarsBehavior =
                WindowInsetsControllerCompat.BEHAVIOR_SHOW_TRANSIENT_BARS_BY_SWIPE
        }

        assertThat(WindowCompat.getInsetsController(window, contentView).systemBarsBehavior)
            .isEqualTo(WindowInsetsControllerCompat.BEHAVIOR_SHOW_TRANSIENT_BARS_BY_SWIPE)
    }

    @Test
    @FlakyTest(detail = "https://github.com/google/accompanist/issues/491")
    @SdkSuppress(minSdkVersion = 23) // rootWindowInsets which work
    @Category(IgnoreOnRobolectric::class)
    fun statusBarsVisibility() {
        lateinit var systemUiController: SystemUiController

        rule.setContent {
            Dialog(onDismissRequest = {}) {
                contentView = LocalView.current

                systemUiController = rememberSystemUiController()
            }
        }

        // First show the bars
        rule.activityRule.scenario.onActivity {
            systemUiController.isStatusBarVisible = true
        }
        waitUntil { isRootWindowTypeVisible(WindowInsetsCompat.Type.statusBars()) }

        // Now hide the bars
        rule.activityRule.scenario.onActivity {
            systemUiController.isStatusBarVisible = false
        }
        waitUntil { !isRootWindowTypeVisible(WindowInsetsCompat.Type.statusBars()) }
    }

    @Test
    @FlakyTest(detail = "https://github.com/google/accompanist/issues/491")
    @SdkSuppress(minSdkVersion = 23) // rootWindowInsets which work
    @Category(IgnoreOnRobolectric::class)
    fun navigationBarsVisibility() {
        lateinit var systemUiController: SystemUiController

        rule.setContent {
            Dialog(onDismissRequest = {}) {
                contentView = LocalView.current

                systemUiController = rememberSystemUiController()
            }
        }

        // First show the bars
        rule.activityRule.scenario.onActivity {
            systemUiController.isNavigationBarVisible = true
        }
        waitUntil { isRootWindowTypeVisible(WindowInsetsCompat.Type.navigationBars()) }

        // Now hide the bars
        rule.activityRule.scenario.onActivity {
            systemUiController.isNavigationBarVisible = false
        }
        waitUntil { !isRootWindowTypeVisible(WindowInsetsCompat.Type.navigationBars()) }
    }

    @Test
    @Category(IgnoreOnRobolectric::class)
    @FlakyTest(detail = "https://github.com/google/accompanist/issues/491")
    @SdkSuppress(minSdkVersion = 23) // rootWindowInsets which work
    fun systemBarsVisibility() {
        lateinit var systemUiController: SystemUiController

        rule.setContent {
            Dialog(onDismissRequest = {}) {
                contentView = LocalView.current

                systemUiController = rememberSystemUiController()
            }
        }

        // First show the bars
        rule.activityRule.scenario.onActivity {
            systemUiController.isSystemBarsVisible = true
        }
        waitUntil { isRootWindowTypeVisible(WindowInsetsCompat.Type.navigationBars()) }
        waitUntil { isRootWindowTypeVisible(WindowInsetsCompat.Type.statusBars()) }

        // Now hide the bars
        rule.activityRule.scenario.onActivity {
            systemUiController.isSystemBarsVisible = false
        }
        waitUntil { !isRootWindowTypeVisible(WindowInsetsCompat.Type.navigationBars()) }
        waitUntil { !isRootWindowTypeVisible(WindowInsetsCompat.Type.statusBars()) }
    }

    private fun isRootWindowTypeVisible(type: Int): Boolean {
        return rule.activityRule.scenario.withActivity {
            ViewCompat.getRootWindowInsets(contentView)!!.isVisible(type)
        }
    }
}