Add BrowserType.Connect test

Also adds a new method in common.Browser API so we can get the
underlying browser process WS URL. This is required in order to reuse
the current TestBrowser implementation to launch a new browser and
connecto to it through BrowserType.Connect.

common/browser.go

@@ -532,3 +532,8 @@ func (b *Browser) Version() string {
 	}
 	return product[i+1:]
 }
+
+// WsURL returns the Websocket URL that the browser is listening on for CDP clients.
+func (b *Browser) WsURL() string {
+	return b.browserProc.WsURL()
+}

tests/browser_type_test.go

@@ -0,0 +1,20 @@
+package tests
+
+import (
+	"testing"
+
+	"github.com/grafana/xk6-browser/chromium"
+	"github.com/grafana/xk6-browser/k6ext/k6test"
+)
+
+func TestBrowserTypeConnect(t *testing.T) {
+	// Start a test browser so we can get its WS URL
+	// and use it to connect through BrowserType.Connect.
+	tb := newTestBrowser(t)
+	vu := k6test.NewVU(t)
+	bt := chromium.NewBrowserType(vu)
+	vu.MoveToVUContext()
+
+	b := bt.Connect(tb.WsURL(), nil)
+	b.NewPage(nil)
+}

tests/test_browser.go

@@ -11,6 +11,7 @@ import (
 
 	"github.com/grafana/xk6-browser/api"
 	"github.com/grafana/xk6-browser/chromium"
+	"github.com/grafana/xk6-browser/common"
 	"github.com/grafana/xk6-browser/k6ext"
 	"github.com/grafana/xk6-browser/k6ext/k6test"
 
@@ -32,7 +33,8 @@ type testBrowser struct {
 	vu       *k6test.VU
 	logCache *logCache
 
-	pid int // the browser process ID
+	pid   int // the browser process ID
+	wsURL string
 
 	api.Browser
 }
@@ -110,7 +112,12 @@ func newTestBrowser(tb testing.TB, opts ...any) *testBrowser {
 		state.TLSConfig = testServer.TLSClientConfig
 		state.Transport = testServer.HTTPTransport
 	}
+
 	b, pid := bt.Launch(rt.ToValue(launchOpts))
+	cb, ok := b.(*common.Browser)
+	if !ok {
+		panic(fmt.Errorf("testBrowser: unexpected browser %T", b))
+	}
 
 	tb.Cleanup(func() {
 		select {
@@ -130,6 +137,7 @@ func newTestBrowser(tb testing.TB, opts ...any) *testBrowser {
 		logCache: lc,
 		Browser:  b,
 		pid:      pid,
+		wsURL:    cb.WsURL(),
 	}
 	if enableFileServer {
 		tbr = tbr.withFileServer()
@@ -187,6 +195,13 @@ func (b *testBrowser) staticURL(path string) string {
 	return b.URL("/" + testBrowserStaticDir + "/" + path)
 }
 
+// WsURL returns the Websocket URL that the test browser is listening on for CDP clients.
+func (b *testBrowser) WsURL() string {
+	b.t.Helper()
+
+	return b.wsURL
+}
+
 // attachFrame attaches the frame to the page and returns it.
 func (b *testBrowser) attachFrame(page api.Page, frameID string, url string) api.Frame {
 	b.t.Helper()