Support cleaning device status bar before taking screenshot

Corona-Warn-App/src/androidTest/java/de/rki/coronawarnapp/ui/onboarding/OnboardingDeltaInteroperabilityFragmentTest.kt

@@ -19,6 +19,7 @@ import org.junit.Test
 import org.junit.runner.RunWith
 import testhelpers.BaseUITest
 import testhelpers.Screenshot
+import testhelpers.SystemUIDemoModeRule
 import testhelpers.TestDispatcherProvider
 import testhelpers.launchFragmentInContainer2
 import tools.fastlane.screengrab.Screengrab
@@ -33,6 +34,9 @@ class OnboardingDeltaInteroperabilityFragmentTest : BaseUITest() {
     @JvmField
     val localeTestRule = LocaleTestRule()
 
+    @get:Rule
+    val systemUIDemoModeRule = SystemUIDemoModeRule()
+
     @Before
     fun setup() {
         MockKAnnotations.init(this, relaxed = true)

Corona-Warn-App/src/androidTest/java/de/rki/coronawarnapp/ui/onboarding/OnboardingFragmentTest.kt

@@ -18,6 +18,7 @@ import org.junit.Test
 import org.junit.runner.RunWith
 import testhelpers.BaseUITest
 import testhelpers.Screenshot
+import testhelpers.SystemUIDemoModeRule
 import testhelpers.launchFragmentInContainer2
 import tools.fastlane.screengrab.Screengrab
 import tools.fastlane.screengrab.locale.LocaleTestRule
@@ -29,6 +30,9 @@ class OnboardingFragmentTest : BaseUITest() {
     @JvmField
     val localeTestRule = LocaleTestRule()
 
+    @get:Rule
+    val systemUIDemoModeRule = SystemUIDemoModeRule()
+
     @Before
     fun setUp() {
         setupMockViewModel(object : OnboardingFragmentViewModel.Factory {

Corona-Warn-App/src/androidTest/java/de/rki/coronawarnapp/ui/onboarding/OnboardingNotificationsFragmentTest.kt

@@ -16,6 +16,7 @@ import org.junit.Test
 import org.junit.runner.RunWith
 import testhelpers.BaseUITest
 import testhelpers.Screenshot
+import testhelpers.SystemUIDemoModeRule
 import testhelpers.launchFragmentInContainer2
 import tools.fastlane.screengrab.Screengrab
 import tools.fastlane.screengrab.locale.LocaleTestRule
@@ -27,6 +28,9 @@ class OnboardingNotificationsFragmentTest : BaseUITest() {
     @JvmField
     val localeTestRule = LocaleTestRule()
 
+    @get:Rule
+    val systemUIDemoModeRule = SystemUIDemoModeRule()
+
     @Before
     fun setup() {
         setupMockViewModel(object : OnboardingNotificationsViewModel.Factory {

Corona-Warn-App/src/androidTest/java/de/rki/coronawarnapp/ui/onboarding/OnboardingPrivacyFragmentTest.kt

@@ -15,6 +15,7 @@ import org.junit.Test
 import org.junit.runner.RunWith
 import testhelpers.BaseUITest
 import testhelpers.Screenshot
+import testhelpers.SystemUIDemoModeRule
 import testhelpers.launchFragmentInContainer2
 import tools.fastlane.screengrab.Screengrab
 import tools.fastlane.screengrab.locale.LocaleTestRule
@@ -26,6 +27,9 @@ class OnboardingPrivacyFragmentTest : BaseUITest() {
     @JvmField
     val localeTestRule = LocaleTestRule()
 
+    @get:Rule
+    val systemUIDemoModeRule = SystemUIDemoModeRule()
+
     @Before
     fun setup() {
         setupMockViewModel(object : OnboardingPrivacyViewModel.Factory {

Corona-Warn-App/src/androidTest/java/de/rki/coronawarnapp/ui/onboarding/OnboardingTestFragmentTest.kt

@@ -16,6 +16,7 @@ import org.junit.Test
 import org.junit.runner.RunWith
 import testhelpers.BaseUITest
 import testhelpers.Screenshot
+import testhelpers.SystemUIDemoModeRule
 import testhelpers.launchFragmentInContainer2
 import tools.fastlane.screengrab.Screengrab
 import tools.fastlane.screengrab.locale.LocaleTestRule
@@ -27,6 +28,9 @@ class OnboardingTestFragmentTest : BaseUITest() {
     @JvmField
     val localeTestRule = LocaleTestRule()
 
+    @get:Rule
+    val systemUIDemoModeRule = SystemUIDemoModeRule()
+
     @Before
     fun setup() {
         setupMockViewModel(object : OnboardingTestViewModel.Factory {

Corona-Warn-App/src/androidTest/java/de/rki/coronawarnapp/ui/onboarding/OnboardingTracingFragmentTest.kt

@@ -24,6 +24,7 @@ import org.junit.Test
 import org.junit.runner.RunWith
 import testhelpers.BaseUITest
 import testhelpers.Screenshot
+import testhelpers.SystemUIDemoModeRule
 import testhelpers.TestDispatcherProvider
 import testhelpers.launchFragmentInContainer2
 import tools.fastlane.screengrab.Screengrab
@@ -39,6 +40,9 @@ class OnboardingTracingFragmentTest : BaseUITest() {
     @JvmField
     val localeTestRule = LocaleTestRule()
 
+    @get:Rule
+    val systemUIDemoModeRule = SystemUIDemoModeRule()
+
     @Before
     fun setup() {
         MockKAnnotations.init(this, relaxed = true)

Corona-Warn-App/src/androidTest/java/testhelpers/SystemUIDemoModeHelper.kt

+package testhelpers
+
+import android.os.ParcelFileDescriptor
+import androidx.test.platform.app.InstrumentationRegistry
+import java.io.BufferedReader
+import java.io.InputStreamReader
+
+/**
+ * System UI Demo Mode helper that uses instrumentation commands to clean device status bar.
+ * @see <a href="https://android.googlesource.com/platform/frameworks/base/+/master/packages/SystemUI/docs/demo_mode.md" >Demo mode</a>
+ * @see <a href="https://developer.android.com/studio/debug/dev-options#general" >Dev options</a>
+ */
+class SystemUIDemoModeHelper {
+
+    /**
+     * Allows System UI demo mode and configure how the status bar should look like
+     */
+    fun enter() {
+        executeShellCommand("settings put global sysui_demo_allowed 1")
+        sendCommand("exit")
+        sendCommand("enter")
+        sendCommand("notifications", "visible" to "false")
+        sendCommand("network", "wifi" to "show", "level" to "4", "fully" to "true")
+        sendCommand("battery", "level" to "100", "plugged" to "false")
+        sendCommand("clock", "hhmm" to "1000")
+    }
+
+    /**
+     * Exist system UI demo mode
+     */
+    fun exit() {
+        sendCommand("exit")
+    }
+
+    private fun sendCommand(command: String, vararg extras: Pair<String, Any>) {
+        val exec = StringBuilder("am broadcast -a com.android.systemui.demo -e command $command")
+        for ((key, value) in extras) {
+            exec.append(" -e $key $value")
+        }
+        executeShellCommand(exec.toString())
+    }
+
+    private fun executeShellCommand(command: String) {
+        waitForCompletion(InstrumentationRegistry.getInstrumentation().uiAutomation.executeShellCommand(command))
+    }
+
+    private fun waitForCompletion(descriptor: ParcelFileDescriptor) {
+        BufferedReader(
+            InputStreamReader(
+                ParcelFileDescriptor.AutoCloseInputStream(descriptor)
+            )
+        ).use {
+            it.readText()
+        }
+    }
+}

Corona-Warn-App/src/androidTest/java/testhelpers/SystemUIDemoModeRule.kt

+package testhelpers
+
+import org.junit.rules.TestRule
+import org.junit.runner.Description
+import org.junit.runners.model.Statement
+
+/**
+ * Enters UI demo mode (clean up device status bar)
+ * before running screenshot tests and exists after it afterwards
+ */
+class SystemUIDemoModeRule : TestRule {
+
+    private val helper = SystemUIDemoModeHelper()
+
+    override fun apply(base: Statement, description: Description): Statement =
+        object : Statement() {
+            override fun evaluate() {
+                try {
+                    helper.enter()
+                    base.evaluate()
+                    helper.exit()
+                } catch (e: Exception) {
+                    e.printStackTrace()
+                }
+            }
+        }
+}