54 lines
1.4 KiB
Go
54 lines
1.4 KiB
Go
package smoketests
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/go-rod/rod"
|
|
"github.com/go-rod/rod/lib/proto"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/ysmood/gson"
|
|
)
|
|
|
|
func TestExampleExecuteScript(t *testing.T) {
|
|
setupPageTest(t, "examples/execute_script", func(runner runnerFn) {
|
|
runner("console log", func(t *testing.T, page *rod.Page) {
|
|
btn := page.MustElement("article > div > button:nth-of-type(1)")
|
|
btn.MustClick()
|
|
|
|
results := make([]gson.JSON, 0, 1)
|
|
|
|
page.EachEvent(func(e *proto.RuntimeConsoleAPICalled) (stop bool) {
|
|
if e.Type == proto.RuntimeConsoleAPICalledTypeLog {
|
|
// t.Log(page.MustObjectsToJSON(e.Args))
|
|
results = append(results, page.MustObjectsToJSON(e.Args))
|
|
if len(results) == 1 {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
})()
|
|
|
|
assert.Equal(t, 1, len(results))
|
|
})
|
|
runner("console error", func(t *testing.T, page *rod.Page) {
|
|
btn := page.MustElement("article > div > button:nth-of-type(2)")
|
|
btn.MustClick()
|
|
|
|
results := make([]gson.JSON, 0, 1)
|
|
|
|
page.EachEvent(func(e *proto.RuntimeConsoleAPICalled) (stop bool) {
|
|
if e.Type == proto.RuntimeConsoleAPICalledTypeError {
|
|
// t.Log(page.MustObjectsToJSON(e.Args))
|
|
results = append(results, page.MustObjectsToJSON(e.Args))
|
|
if len(results) == 1 {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
})()
|
|
|
|
assert.Equal(t, 1, len(results))
|
|
})
|
|
})
|
|
}
|