From ebf28a190e08e23b13a02bb3d205f2050114acd6 Mon Sep 17 00:00:00 2001
From: buyuxiang <347586493@qq.com>
Date: Wed, 31 May 2023 11:09:41 +0800
Subject: [PATCH] avoid incorrect sleep because of screenshot after step (or
 each loop of step)

---
 hrp/step_mobile_ui.go | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/hrp/step_mobile_ui.go b/hrp/step_mobile_ui.go
index 42119c21..9e489d48 100644
--- a/hrp/step_mobile_ui.go
+++ b/hrp/step_mobile_ui.go
@@ -643,10 +643,11 @@ func runStepMobileUI(s *SessionRunner, step *TStep) (stepResult *StepResult, err
 
 		// wait for screenshot actions done
 		uiDriver.Wg.Wait()
-		// save screenshot after each step
-		err := uiDriver.SaveScreenShot(builtin.GenNameWithTimestampMS("step_%d_") + step.Name)
-		if err != nil {
-			log.Error().Err(err).Str("step", step.Name).Msg("take screenshot failed on step finished")
+		// save screenshot after each step, except the step ends with sleep
+		if len(actions) > 0 && actions[len(actions)-1].Method != uixt.CtlSleep {
+			if err := uiDriver.SaveScreenShot(builtin.GenNameWithTimestampMS("step_%d_") + step.Name); err != nil {
+				log.Error().Err(err).Str("step", step.Name).Msg("take screenshot failed on step finished")
+			}
 		}
 
 		// save attachments