mirror of
https://github.com/httprunner/httprunner.git
synced 2026-05-15 04:19:28 +08:00
34 lines
1.3 KiB
Go
34 lines
1.3 KiB
Go
package ai
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/httprunner/httprunner/v5/uixt/types"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestParseAction(t *testing.T) {
|
|
actionStr := "click(point='<point>200 300</point>')"
|
|
result, err := ParseAction(actionStr)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
assert.Equal(t, result.Function, "click")
|
|
assert.Equal(t, result.Args["point"], "<point>200 300</point>")
|
|
}
|
|
|
|
func TestParseActionToStructureOutput(t *testing.T) {
|
|
text := "Thought: test\nAction: click(point='<point>200 300</point>')"
|
|
parser := &UITARSContentParser{}
|
|
result, err := parser.Parse(text, types.Size{Height: 224, Width: 224})
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, result.Actions[0].ActionType, "click")
|
|
assert.Contains(t, result.Actions[0].ActionInputs, "start_box")
|
|
|
|
text = "Thought: 我看到页面上有几个帖子，第二个帖子的标题是\"字节四年，头发白了\"。要完成任务，我需要点击这个帖子下方的作者头像，这样就能进入作者的个人主页了。\nAction: click(start_point='<point>550 450 550 450</point>')"
|
|
result, err = parser.Parse(text, types.Size{Height: 2341, Width: 1024})
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, result.Actions[0].ActionType, "click")
|
|
assert.Contains(t, result.Actions[0].ActionInputs, "start_box")
|
|
}
|