refactor: move LoadImage

This commit is contained in:
lilong.129
2025-04-30 16:21:01 +08:00
parent fcddcfb630
commit 6569121d5d
7 changed files with 156 additions and 160 deletions
+32
View File
@@ -3,9 +3,11 @@
package uixt
import (
"os"
"testing"
"time"
"github.com/httprunner/httprunner/v5/internal/builtin"
"github.com/httprunner/httprunner/v5/uixt/option"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
@@ -251,3 +253,33 @@ func TestDriverExt_Action_Offset(t *testing.T) {
option.WithTapRandomRect(true))
assert.Nil(t, err)
}
func TestSavePositionImg(t *testing.T) {
imageBase64, _, err := builtin.LoadImage("ai/testdata/popup_risk_warning.png")
require.NoError(t, err)
params := struct {
InputImgBase64 string
Rect struct {
X float64
Y float64
}
OutputPath string
}{
InputImgBase64: imageBase64,
Rect: struct {
X float64
Y float64
}{
X: 500,
Y: 500,
},
OutputPath: "ai/testdata/output.png",
}
err = SavePositionImg(params)
require.NoError(t, err)
// cleanup
defer os.Remove(params.OutputPath)
}