private_test.go

package launcher

import (
	"fmt"
	"io/ioutil"
	"net/url"
	"os"
	"os/exec"
	"path/filepath"
	"sync"
	"testing"
	"time"

	"github.com/go-rod/rod/lib/cdp"
	"github.com/go-rod/rod/lib/defaults"
	"github.com/go-rod/rod/lib/launcher/flags"
	"github.com/go-rod/rod/lib/utils"
	"github.com/ysmood/got"
)

func HostTest(host string) Host {
	return func(revision int) string {
		return fmt.Sprintf(
			"%s/chromium-browser-snapshots/%s/%d/%s",
			host,
			hostConf.urlPrefix,
			revision,
			hostConf.zipName,
		)
	}
}

type T struct {
	got.G
}

func TestPrivate(t *testing.T) {
	NewBrowser().MustGet() // preload browser to local

	got.Each(t, T{})
}

func (t T) ToHTTP() {
	u, _ := url.Parse("wss://a.com")
	t.Eq("https", toHTTP(*u).Scheme)

	u, _ = url.Parse("ws://a.com")
	t.Eq("http", toHTTP(*u).Scheme)
}

func (t T) ToWS() {
	u, _ := url.Parse("https://a.com")
	t.Eq("wss", toWS(*u).Scheme)

	u, _ = url.Parse("http://a.com")
	t.Eq("ws", toWS(*u).Scheme)
}

func (t T) Unzip() {
	t.Err(unzip(ioutil.Discard, "", ""))
}

func (t T) LaunchOptions() {
	defaults.Show = true
	defaults.Devtools = true
	inContainer = true

	// restore
	defer func() {
		defaults.ResetWithEnv("")
		inContainer = utils.InContainer
	}()

	l := New()

	t.False(l.Has(flags.Headless))

	t.True(l.Has(flags.NoSandbox))

	t.True(l.Has("auto-open-devtools-for-tabs"))
}

func (t T) GetURLErr() {
	l := New()

	l.ctxCancel()
	_, err := l.getURL()
	t.Err(err)

	l = New()
	l.parser.lock.Lock()
	l.parser.Buffer = "err"
	l.parser.lock.Unlock()
	close(l.exit)
	_, err = l.getURL()
	t.Eq("[launcher] Failed to get the debug url: err", err.Error())
}

func (t T) RemoteLaunch() {
	ctx := t.Timeout(5 * time.Second)

	s := got.New(t).Serve()
	rl := NewManager()
	s.Mux.Handle("/", rl)

	l := MustNewManaged(s.URL()).KeepUserDataDir().Delete(flags.KeepUserDataDir)
	client := l.Client()
	b := client.MustConnect(ctx)
	t.E(b.Call(ctx, "", "Browser.getVersion", nil))
	utils.Sleep(1)
	_, _ = b.Call(ctx, "", "Browser.crash", nil)
	dir := l.Get(flags.UserDataDir)

	for ctx.Err() == nil {
		utils.Sleep(0.1)
		_, err := os.Stat(dir)
		if err != nil {
			break
		}
	}
	t.Err(os.Stat(dir))

	err := MustNewManaged(s.URL()).Bin("go").Client().Connect(ctx).(*cdp.ErrBadHandshake)
	t.Eq(err.Body, "not allowed rod-bin path: go")
}

func (t T) LaunchErrs() {
	l := New().Bin("echo")
	_, err := l.Launch()
	t.Err(err)

	s := t.Serve()
	s.Route("/", "", nil)
	l = New().Bin("")
	l.browser.Logger = ioutil.Discard
	l.browser.Dir = filepath.Join("tmp", "browser-from-mirror", t.Srand(16))
	l.browser.Hosts = []Host{HostTest(s.URL())}
	_, err = l.Launch()
	t.Err(err)
}

func (t T) Progresser() {
	p := progresser{size: 100, logger: ioutil.Discard}

	t.E(p.Write(make([]byte, 100)))
	t.E(p.Write(make([]byte, 100)))
	t.E(p.Write(make([]byte, 100)))
}

func (t T) URLParserErr() {
	u := &URLParser{
		Buffer: "error",
		lock:   &sync.Mutex{},
	}

	t.Eq(u.Err().Error(), "[launcher] Failed to get the debug url: error")

	u.Buffer = "/tmp/rod/chromium-818858/chrome-linux/chrome: error while loading shared libraries: libgobject-2.0.so.0: cannot open shared object file: No such file or directory"
	t.Eq(u.Err().Error(), "[launcher] Failed to launch the browser, the doc might help https://go-rod.github.io/#/compatibility?id=os: /tmp/rod/chromium-818858/chrome-linux/chrome: error while loading shared libraries: libgobject-2.0.so.0: cannot open shared object file: No such file or directory")
}

func (t T) BrowserDownloadErr() {
	b := NewBrowser()
	b.Logger = ioutil.Discard
	malURL := "https://npm.taobao.org/mirrors/chromium-browser-snapshots//869685/"
	t.Has(b.download(t.Context(), malURL).Error(), "failed to download the browser: 200")
}

func (t T) TestOpen() {
	openExec = func(name string, arg ...string) *exec.Cmd {
		cmd := exec.Command("not-exists")
		cmd.Process = &os.Process{}
		return cmd
	}
	defer func() { openExec = exec.Command }()

	Open("about:blank")
}