From a4d7f0d5a09f0ce79b7e2f977db2972ad7ada3c7 Mon Sep 17 00:00:00 2001
From: Nataniel Borges <natanieljr@google.com>
Date: Wed, 26 Jul 2023 08:07:19 +0000
Subject: [PATCH] Fix split screen tests

You can't go to overview when you are in overview

Fixes: 293212785
Test: atest WMShellFlickerTestsSplitScreenEnter WMShellFlickerTestsSplitScreenOther
Change-Id: I5799de2b649564437438be276eaea90fd7490cb2
---
 .../shell/flicker/splitscreen/benchmark/SplitScreenBase.kt   | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/libs/WindowManager/Shell/tests/flicker/src/com/android/wm/shell/flicker/splitscreen/benchmark/SplitScreenBase.kt b/libs/WindowManager/Shell/tests/flicker/src/com/android/wm/shell/flicker/splitscreen/benchmark/SplitScreenBase.kt
index fdcb7e5a09843..a0e437c25aa78 100644
--- a/libs/WindowManager/Shell/tests/flicker/src/com/android/wm/shell/flicker/splitscreen/benchmark/SplitScreenBase.kt
+++ b/libs/WindowManager/Shell/tests/flicker/src/com/android/wm/shell/flicker/splitscreen/benchmark/SplitScreenBase.kt
@@ -33,8 +33,9 @@ abstract class SplitScreenBase(flicker: LegacyFlickerTest) : BaseBenchmarkTest(f
             tapl.setEnableRotation(true)
             setRotation(flicker.scenario.startRotation)
             tapl.setExpectedRotation(flicker.scenario.startRotation.value)
-            if (tapl.workspace.switchToOverview().hasTasks()) {
-                tapl.workspace.switchToOverview().dismissAllTasks()
+            val overview = tapl.workspace.switchToOverview()
+            if (overview.hasTasks()) {
+                overview.dismissAllTasks()
             }
         }
     }