From f77b454cf28ec77b4f809c9e560cd4f33fa0e23b Mon Sep 17 00:00:00 2001
From: "lilong.129" <lilong.129@bytedance.com>
Date: Fri, 14 Apr 2023 13:29:22 +0800
Subject: [PATCH] change: take screenshot after each step

---
 hrp/step_mobile_ui.go | 20 ++++++++++----------
 1 file changed, 10 insertions(+), 10 deletions(-)

diff --git a/hrp/step_mobile_ui.go b/hrp/step_mobile_ui.go
index a193f0b8..56ee9e4f 100644
--- a/hrp/step_mobile_ui.go
+++ b/hrp/step_mobile_ui.go
@@ -563,6 +563,16 @@ func runStepMobileUI(s *SessionRunner, step *TStep) (stepResult *StepResult, err
 			attachments["error"] = err.Error()
 		}
 
+		// take screenshot after each step
+		screenshotPath, err := uiDriver.ScreenShot(
+			fmt.Sprintf("step_%d", time.Now().Unix()))
+		if err != nil {
+			log.Error().Err(err).Str("step", step.Name).Msg("take screenshot failed")
+		} else {
+			log.Info().Str("path", screenshotPath).Msg("take screenshot on step finished")
+			screenshots = append(screenshots, screenshotPath)
+		}
+
 		// save attachments
 		screenshots = append(screenshots, uiDriver.ScreenShots...)
 		attachments["screenshots"] = screenshots
@@ -599,16 +609,6 @@ func runStepMobileUI(s *SessionRunner, step *TStep) (stepResult *StepResult, err
 		}
 	}
 
-	// take snapshot
-	screenshotPath, err := uiDriver.ScreenShot(
-		fmt.Sprintf("validate_%d", time.Now().Unix()))
-	if err != nil {
-		log.Warn().Err(err).Str("step", step.Name).Msg("take screenshot failed")
-	} else {
-		log.Info().Str("path", screenshotPath).Msg("take screenshot before validation")
-		screenshots = append(screenshots, screenshotPath)
-	}
-
 	// validate
 	validateResults, err := validateUI(uiDriver, step.Validators)
 	if err != nil {