32d4bae280
First, collapse Browser.Start with NewBrowser. There's no reason to split them up. Second, unexport Browser.userDataDir, since it's only needed for a test. It's also a bad precedent, as only the ExecAllocator will control the user data directory. Third, export Context.Browser, since we were already exporting Context.Allocator. Finally, remove the Executor interface, a duplicate of cdp.Executor.
44 lines
849 B
Go
44 lines
849 B
Go
package chromedp
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"testing"
|
|
)
|
|
|
|
func TestExecAllocator(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
poolCtx, cancel := NewAllocator(context.Background(), WithExecAllocator(allocOpts...))
|
|
defer cancel()
|
|
|
|
// TODO: test that multiple child contexts are run in different
|
|
// processes and browsers.
|
|
|
|
taskCtx, cancel := NewContext(poolCtx)
|
|
defer cancel()
|
|
|
|
want := "insert"
|
|
var got string
|
|
if err := Run(taskCtx, Tasks{
|
|
Navigate(testdataDir + "/form.html"),
|
|
Text("#foo", &got, ByID),
|
|
}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if got != want {
|
|
t.Fatalf("wanted %q, got %q", want, got)
|
|
}
|
|
|
|
tempDir := FromContext(taskCtx).Browser.userDataDir
|
|
pool := FromContext(taskCtx).Allocator
|
|
|
|
cancel()
|
|
pool.Wait()
|
|
|
|
if _, err := os.Lstat(tempDir); os.IsNotExist(err) {
|
|
return
|
|
}
|
|
t.Fatalf("temporary user data dir %q not deleted", tempDir)
|
|
}
|