GithubBackup/httprunner
1
0
Fork 0
mirror of https://github.com/httprunner/httprunner.git synced 2026-05-12 02:21:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: get screenshot info with retry

Browse Source
This commit is contained in:
lilong.129
2024-09-14 10:17:16 +08:00
parent 6bb67e8a68
commit bfbd4ba6e6
1 changed files with 10 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
hrp/pkg/uixt/screenshot.go
View File

@@ -50,15 +50,19 @@ func (dExt *DriverExt) GetScreenResult(options ...ActionOption) (screenResult *S
var imageResult *ImageResult
var imagePath string
var windowSize Size
var lastErr error
// get screenshot info with retry
for i := 0; i < actionOptions.MaxRetryTimes; i++ {
bufSource, imagePath, err = dExt.GetScreenShot(fileName)
if err != nil {
lastErr = err
continue
}
windowSize, err = dExt.Driver.WindowSize()
if err != nil {
err = errors.Wrap(code.MobileUIDriverError, err.Error())
lastErr = errors.Wrap(code.MobileUIDriverError, err.Error())
continue
}
@@ -71,13 +75,15 @@ func (dExt *DriverExt) GetScreenResult(options ...ActionOption) (screenResult *S
imageResult, err = dExt.ImageService.GetImage(bufSource, options...)
if err != nil {
log.Error().Err(err).Msg("GetImage from ImageService failed")
lastErr = err
continue
}
// success
// success, break the loop
lastErr = nil
break
}
if err != nil {
return nil, err
if lastErr != nil {
return nil, lastErr
}
// cache screen result