fix: unittest

This commit is contained in:
xucong.053
2023-04-19 19:37:26 +08:00
parent fe3f37d40d
commit dd7d0ee708

View File

@@ -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
}