From dd7d0ee708dc7efb69fe4ac5c8494863b01b18e6 Mon Sep 17 00:00:00 2001
From: "xucong.053" <xucong.053@outlook.com>
Date: Wed, 19 Apr 2023 19:37:26 +0800
Subject: [PATCH] fix: unittest

---
 hrp/pkg/uixt/algorithm.go | 28 ++++++++++++++--------------
 1 file changed, 14 insertions(+), 14 deletions(-)

diff --git a/hrp/pkg/uixt/algorithm.go b/hrp/pkg/uixt/algorithm.go
index 33c39cd4..20e4453f 100644
--- a/hrp/pkg/uixt/algorithm.go
+++ b/hrp/pkg/uixt/algorithm.go
@@ -143,7 +143,7 @@ func (dExt *DriverExt) FindAllImageRect(search string, options ...DataOption) (r
 	if bufSearch, err = im.read(); err != nil {
 		return nil, err
 	}
-	if bufSource, err = dExt.takeScreenShot(); err != nil {
+	if bufSource, err = dExt.TakeScreenShot(); err != nil {
 		return nil, err
 	}
 
@@ -162,7 +162,7 @@ func (dExt *DriverExt) FindImageRectInUIKit(imagePath string, options ...DataOpt
 	if bufSearch, err = im.read(); err != nil {
 		return 0, 0, 0, 0, err
 	}
-	if bufSource, err = dExt.takeScreenShot(); err != nil {
+	if bufSource, err = dExt.TakeScreenShot(); err != nil {
 		return 0, 0, 0, 0, err
 	}
 
@@ -197,8 +197,8 @@ type CVService interface {
 
 func (dExt *DriverExt) FindImageByCV(cvImage string, options ...DataOption) (rect image.Rectangle, err error) {
 	var bufSource *bytes.Buffer
-	if bufSource, err = dExt.takeScreenShot(); err != nil {
-		err = fmt.Errorf("takeScreenShot error: %v", err)
+	if bufSource, err = dExt.TakeScreenShot(); err != nil {
+		err = fmt.Errorf("TakeScreenShot error: %v", err)
 		return
 	}
 
@@ -245,8 +245,8 @@ func (dExt *DriverExt) ClosePopupHandler() {
 
 func (dExt *DriverExt) FindPopupCloseButton(options ...DataOption) (rect image.Rectangle, err error) {
 	var bufSource *bytes.Buffer
-	if bufSource, err = dExt.takeScreenShot(); err != nil {
-		err = fmt.Errorf("takeScreenShot error: %v", err)
+	if bufSource, err = dExt.TakeScreenShot(); err != nil {
+		err = fmt.Errorf("TakeScreenShot error: %v", err)
 		return
 	}
 
@@ -273,8 +273,8 @@ type OCRService interface {
 
 func (dExt *DriverExt) GetTextsByOCR(options ...DataOption) (texts OCRTexts, err error) {
 	var bufSource *bytes.Buffer
-	if bufSource, err = dExt.takeScreenShot(); err != nil {
-		err = fmt.Errorf("takeScreenShot error: %v", err)
+	if bufSource, err = dExt.TakeScreenShot(); err != nil {
+		err = fmt.Errorf("TakeScreenShot error: %v", err)
 		return
 	}
 
@@ -289,8 +289,8 @@ func (dExt *DriverExt) GetTextsByOCR(options ...DataOption) (texts OCRTexts, err
 
 func (dExt *DriverExt) FindTextByOCR(ocrText string, options ...DataOption) (x, y, width, height float64, err error) {
 	var bufSource *bytes.Buffer
-	if bufSource, err = dExt.takeScreenShot(); err != nil {
-		err = fmt.Errorf("takeScreenShot error: %v", err)
+	if bufSource, err = dExt.TakeScreenShot(); err != nil {
+		err = fmt.Errorf("TakeScreenShot error: %v", err)
 		return
 	}
 
@@ -308,8 +308,8 @@ func (dExt *DriverExt) FindTextByOCR(ocrText string, options ...DataOption) (x,
 
 func (dExt *DriverExt) FindTextsByOCR(ocrTexts []string, options ...DataOption) (points [][]float64, err error) {
 	var bufSource *bytes.Buffer
-	if bufSource, err = dExt.takeScreenShot(); err != nil {
-		err = fmt.Errorf("takeScreenShot error: %v", err)
+	if bufSource, err = dExt.TakeScreenShot(); err != nil {
+		err = fmt.Errorf("TakeScreenShot error: %v", err)
 		return
 	}
 
@@ -335,8 +335,8 @@ type SDService interface {
 
 func (dExt *DriverExt) ScenarioDetect(scenarioType string, options ...DataOption) (res bool, err error) {
 	var bufSource *bytes.Buffer
-	if bufSource, err = dExt.takeScreenShot(); err != nil {
-		err = fmt.Errorf("takeScreenShot error: %v", err)
+	if bufSource, err = dExt.TakeScreenShot(); err != nil {
+		err = fmt.Errorf("TakeScreenShot error: %v", err)
 		return
 	}