From 0191ba0b84a13022fa80bd9bc9315880c38443e6 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jose=20Alc=C3=A9rreca?= <jalc@google.com>
Date: Thu, 6 Jun 2024 13:04:30 +0000
Subject: [PATCH] Some more sync points, api 35 emulator crashing

Change-Id: I90422a13aaba689b035095307446d7d343a59aed
---
 .../samples/apps/nowinandroid/ui/EdgeToEdgeTest.kt       | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/app/src/androidTest/kotlin/com/google/samples/apps/nowinandroid/ui/EdgeToEdgeTest.kt b/app/src/androidTest/kotlin/com/google/samples/apps/nowinandroid/ui/EdgeToEdgeTest.kt
index 48511600c..b8b958ba5 100644
--- a/app/src/androidTest/kotlin/com/google/samples/apps/nowinandroid/ui/EdgeToEdgeTest.kt
+++ b/app/src/androidTest/kotlin/com/google/samples/apps/nowinandroid/ui/EdgeToEdgeTest.kt
@@ -153,12 +153,13 @@ class EdgeToEdgeTest {
     fun edgeToEdge_Foldable_api35_tallCutout() {
         forceTallCutout()
         onDevice().setClosedMode()
+        enableDemoMode() // Mode change resets demo mode!
         screenshotSystemBar("edgeToEdge_Foldable_closed_system_tallCutout_api35")
 
         onDevice().setFlatMode()
-        enableDemoMode() // Flat mode resets demo mode!
+        enableDemoMode() // Mode change resets demo mode!
         screenshotSystemBar("edgeToEdge_Foldable_flat_system_tallCutout_api35")
-        forceThreeButtonNavigation()
+
         onDevice().setClosedMode()
         resetCutout()
     }
@@ -178,6 +179,7 @@ class EdgeToEdgeTest {
     }
 
     private fun runFoldableTests(apiName: String) {
+        resetCutout()
         onDevice().setClosedMode()
         screenshotSystemBar("edgeToEdge_Foldable_closed_system_$apiName")
         forceThreeButtonNavigation()
@@ -200,7 +202,7 @@ class EdgeToEdgeTest {
 
         // Crop the top, adding extra pixels to check continuity
         val bitmap = takeScreenshot().let {
-            val newHeight = 130
+            val newHeight = 200
             Bitmap.createBitmap(it, 0, 0, it.width, newHeight)
         }
         assertSnapshot(bitmap, screenshotFileName)
@@ -234,6 +236,7 @@ class EdgeToEdgeTest {
                     "com.android.internal.systemui.navbar.threebutton",
             )
         }
+        waitForWindowUpdate()
     }
 
     private fun forceGestureNavigation() {