chromedp/chromedp_test.go

111 lines
2.4 KiB
Go
Raw Normal View History

2017-02-09 15:01:40 +00:00
package chromedp
import (
"context"
"log"
"os"
"testing"
"time"
"github.com/knq/chromedp/runner"
2017-02-09 15:01:40 +00:00
)
var (
pool *Pool
testdataDir string
defaultContext, defaultCancel = context.WithCancel(context.Background())
cliOpts = []runner.CommandLineOption{
runner.NoDefaultBrowserCheck,
runner.NoFirstRun,
}
)
2017-02-09 15:01:40 +00:00
func testAllocate(t *testing.T, path string) *Res {
c, err := pool.Allocate(defaultContext, cliOpts...)
2017-02-09 15:01:40 +00:00
if err != nil {
t.Fatalf("could not allocate from pool: %v", err)
2017-02-09 15:01:40 +00:00
}
err = WithLogf(t.Logf)(c.c)
2017-02-09 15:01:40 +00:00
if err != nil {
t.Fatalf("could not set logf: %v", err)
2017-02-09 15:01:40 +00:00
}
err = WithDebugf(t.Logf)(c.c)
2017-02-09 15:01:40 +00:00
if err != nil {
t.Fatalf("could not set debugf: %v", err)
2017-02-09 15:01:40 +00:00
}
err = WithErrorf(t.Errorf)(c.c)
2017-02-09 15:01:40 +00:00
if err != nil {
t.Fatalf("could not set errorf: %v", err)
2017-02-09 15:01:40 +00:00
}
h := c.c.GetHandlerByIndex(0)
th, ok := h.(*TargetHandler)
if !ok {
t.Fatalf("handler is invalid type")
2017-02-09 15:01:40 +00:00
}
th.logf = t.Logf
th.debugf = t.Logf
th.errorf = func(s string, v ...interface{}) {
t.Logf("target handler error: "+s, v...)
}
if path != "" {
err = c.Run(defaultContext, Navigate(testdataDir+"/"+path))
if err != nil {
t.Fatalf("could not navigate to testdata/%s: %v", path, err)
}
2017-02-09 15:01:40 +00:00
}
return c
2017-02-09 15:01:40 +00:00
}
func TestMain(m *testing.M) {
var err error
testdataDir = "file://" + os.Getenv("GOPATH") + "/src/github.com/knq/chromedp/testdata"
// its worth noting that newer versions of chrome (64+) run much faster
// than older ones -- same for headless_shell ...
if testRunner := os.Getenv("CHROMEDP_TEST_RUNNER"); testRunner != "" {
cliOpts = append(cliOpts, runner.ExecPath(testRunner))
}
// not explicitly needed to be set, as this vastly speeds up unit tests
if noSandbox := os.Getenv("CHROMEDP_NO_SANDBOX"); noSandbox != "false" {
cliOpts = append(cliOpts, runner.NoSandbox)
}
// must be explicitly set, as disabling gpu slows unit tests
if disableGPU := os.Getenv("CHROMEDP_DISABLE_GPU"); disableGPU != "" && disableGPU != "false" {
cliOpts = append(cliOpts, runner.DisableGPU)
}
if targetTimeout := os.Getenv("CHROMEDP_TARGET_TIMEOUT"); targetTimeout != "" {
defaultNewTargetTimeout, _ = time.ParseDuration(targetTimeout)
}
if defaultNewTargetTimeout == 0 {
defaultNewTargetTimeout = 30 * time.Second
}
//pool, err = NewPool(PoolLog(log.Printf, log.Printf, log.Printf))
pool, err = NewPool()
if err != nil {
log.Fatal(err)
}
code := m.Run()
2017-02-18 06:11:46 +00:00
defaultCancel()
err = pool.Shutdown()
if err != nil {
log.Fatal(err)
}
os.Exit(code)
}