mirror of
https://github.com/httprunner/httprunner.git
synced 2026-06-03 14:59:38 +08:00
98 lines
3.1 KiB
Go
98 lines
3.1 KiB
Go
package uixt
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/httprunner/httprunner/v4/hrp/internal/builtin"
|
|
"github.com/rs/zerolog/log"
|
|
)
|
|
|
|
func assertRelative(p float64) bool {
|
|
return p >= 0 && p <= 1
|
|
}
|
|
|
|
// SwipeRelative swipe from relative position [fromX, fromY] to relative position [toX, toY]
|
|
func (dExt *DriverExt) SwipeRelative(fromX, fromY, toX, toY float64, options ...DataOption) error {
|
|
width := dExt.windowSize.Width
|
|
height := dExt.windowSize.Height
|
|
|
|
if !assertRelative(fromX) || !assertRelative(fromY) ||
|
|
!assertRelative(toX) || !assertRelative(toY) {
|
|
return fmt.Errorf("fromX(%f), fromY(%f), toX(%f), toY(%f) must be less than 1",
|
|
fromX, fromY, toX, toY)
|
|
}
|
|
|
|
fromX = float64(width) * fromX
|
|
fromY = float64(height) * fromY
|
|
toX = float64(width) * toX
|
|
toY = float64(height) * toY
|
|
|
|
return dExt.Driver.SwipeFloat(fromX, fromY, toX, toY, options...)
|
|
}
|
|
|
|
func (dExt *DriverExt) SwipeTo(direction string, options ...DataOption) (err error) {
|
|
switch direction {
|
|
case "up":
|
|
return dExt.SwipeUp(options...)
|
|
case "down":
|
|
return dExt.SwipeDown(options...)
|
|
case "left":
|
|
return dExt.SwipeLeft(options...)
|
|
case "right":
|
|
return dExt.SwipeRight(options...)
|
|
}
|
|
return fmt.Errorf("unexpected direction: %s", direction)
|
|
}
|
|
|
|
func (dExt *DriverExt) SwipeUp(options ...DataOption) (err error) {
|
|
return dExt.SwipeRelative(0.5, 0.5, 0.5, 0.1, options...)
|
|
}
|
|
|
|
func (dExt *DriverExt) SwipeDown(options ...DataOption) (err error) {
|
|
return dExt.SwipeRelative(0.5, 0.5, 0.5, 0.9, options...)
|
|
}
|
|
|
|
func (dExt *DriverExt) SwipeLeft(options ...DataOption) (err error) {
|
|
return dExt.SwipeRelative(0.5, 0.5, 0.1, 0.5, options...)
|
|
}
|
|
|
|
func (dExt *DriverExt) SwipeRight(options ...DataOption) (err error) {
|
|
return dExt.SwipeRelative(0.5, 0.5, 0.9, 0.5, options...)
|
|
}
|
|
|
|
// FindCondition indicates the condition to find a UI element
|
|
type FindCondition func(driver *DriverExt) error
|
|
|
|
// FoundAction indicates the action to do after a UI element is found
|
|
type FoundAction func(driver *DriverExt) error
|
|
|
|
func (dExt *DriverExt) SwipeUntil(direction interface{}, condition FindCondition, action FoundAction, maxTimes int, waitTime float64) error {
|
|
for i := 0; i < maxTimes; i++ {
|
|
if err := condition(dExt); err == nil {
|
|
// do action after found
|
|
return action(dExt)
|
|
}
|
|
if d, ok := direction.(string); ok {
|
|
if err := dExt.SwipeTo(d); err != nil {
|
|
log.Error().Err(err).Msgf("swipe %s failed", d)
|
|
}
|
|
} else if d, ok := direction.([]float64); ok {
|
|
if err := dExt.SwipeRelative(d[0], d[1], d[2], d[3]); err != nil {
|
|
log.Error().Err(err).Msgf("swipe %s failed", d)
|
|
}
|
|
} else if d, ok := direction.([]interface{}); ok {
|
|
sx, _ := builtin.Interface2Float64(d[0])
|
|
sy, _ := builtin.Interface2Float64(d[1])
|
|
ex, _ := builtin.Interface2Float64(d[2])
|
|
ey, _ := builtin.Interface2Float64(d[3])
|
|
if err := dExt.SwipeRelative(sx, sy, ex, ey); err != nil {
|
|
log.Error().Err(err).Msgf("swipe (%v, %v) to (%v, %v) failed", sx, sy, ex, ey)
|
|
}
|
|
}
|
|
// wait for swipe action to completed and content to load completely
|
|
time.Sleep(time.Duration(1000*waitTime) * time.Millisecond)
|
|
}
|
|
return fmt.Errorf("swipe %s %d times, match condition failed", direction, maxTimes)
|
|
}
|