128 lines
2.7 KiB
Go
128 lines
2.7 KiB
Go
package chromedp
|
|
|
|
import (
|
|
"context"
|
|
"log"
|
|
"os"
|
|
"os/exec"
|
|
"path"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/chromedp/chromedp/runner"
|
|
)
|
|
|
|
var (
|
|
pool *Pool
|
|
testdataDir string
|
|
|
|
defaultContext, defaultCancel = context.WithCancel(context.Background())
|
|
|
|
cliOpts = []runner.CommandLineOption{
|
|
runner.NoDefaultBrowserCheck,
|
|
runner.NoFirstRun,
|
|
}
|
|
)
|
|
|
|
func testAllocate(t *testing.T, path string) *Res {
|
|
c, err := pool.Allocate(defaultContext, cliOpts...)
|
|
if err != nil {
|
|
t.Fatalf("could not allocate from pool: %v", err)
|
|
}
|
|
|
|
err = WithLogf(t.Logf)(c.c)
|
|
if err != nil {
|
|
t.Fatalf("could not set logf: %v", err)
|
|
}
|
|
|
|
err = WithDebugf(t.Logf)(c.c)
|
|
if err != nil {
|
|
t.Fatalf("could not set debugf: %v", err)
|
|
}
|
|
|
|
err = WithErrorf(t.Errorf)(c.c)
|
|
if err != nil {
|
|
t.Fatalf("could not set errorf: %v", err)
|
|
}
|
|
|
|
h := c.c.GetHandlerByIndex(0)
|
|
th, ok := h.(*TargetHandler)
|
|
if !ok {
|
|
t.Fatalf("handler is invalid type")
|
|
}
|
|
|
|
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)
|
|
}
|
|
}
|
|
|
|
return c
|
|
}
|
|
|
|
func TestMain(m *testing.M) {
|
|
var err error
|
|
|
|
wd, err := os.Getwd()
|
|
if err != nil {
|
|
log.Fatalf("could not get working directory: %v", err)
|
|
os.Exit(1)
|
|
}
|
|
testdataDir = "file://" + path.Join(wd, "testdata")
|
|
|
|
// its worth noting that newer versions of chrome (64+) run much faster
|
|
// than older ones -- same for headless_shell ...
|
|
execPath := runner.DefaultChromePath
|
|
if testRunner := os.Getenv("CHROMEDP_TEST_RUNNER"); testRunner != "" {
|
|
execPath = testRunner
|
|
} else {
|
|
// use headless_shell, if on path
|
|
var hsPath string
|
|
hsPath, err = exec.LookPath("headless_shell")
|
|
if err == nil {
|
|
execPath = hsPath
|
|
}
|
|
}
|
|
cliOpts = append(cliOpts, runner.ExecPath(execPath))
|
|
|
|
// 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()
|
|
|
|
defaultCancel()
|
|
|
|
err = pool.Shutdown()
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
os.Exit(code)
|
|
}
|