62 lines
902 B
Go
62 lines
902 B
Go
|
package chromedp
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"log"
|
||
|
|
"os"
|
||
|
|
"strings"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
var pool *Pool
|
||
|
|
|
||
|
|
var defaultContext = context.Background()
|
||
|
|
|
||
|
|
func TestMain(m *testing.M) {
|
||
|
|
var err error
|
||
|
|
|
||
|
|
pool, err = NewPool()
|
||
|
|
if err != nil {
|
||
|
|
log.Fatal(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
code := m.Run()
|
||
|
|
|
||
|
|
err = pool.Shutdown()
|
||
|
|
if err != nil {
|
||
|
|
log.Fatal(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
os.Exit(code)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestNavigate(t *testing.T) {
|
||
|
|
var err error
|
||
|
|
|
||
|
|
c, err := pool.Allocate(defaultContext)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
defer c.Release()
|
||
|
|
|
||
|
|
err = c.Run(defaultContext, Navigate("https://www.google.com/"))
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
err = c.Run(defaultContext, WaitVisible(`#hplogo`, ByID))
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
var urlstr string
|
||
|
|
err = c.Run(defaultContext, Location(&urlstr))
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if !strings.HasPrefix(urlstr, "https://www.google.") {
|
||
|
|
t.Errorf("expected to be on google, got: %v", urlstr)
|
||
|
|
}
|
||
|
|
}
|