lib/launcher/load_test.go

package launcher_test

import (
	"context"
	"math/rand"
	"sync"
	"testing"

	"github.com/assetnote/rod"
	"github.com/assetnote/rod/lib/cdp"
	"github.com/assetnote/rod/lib/launcher"
	"github.com/assetnote/rod/lib/utils"
	"github.com/ysmood/got"
)

func BenchmarkManager(b *testing.B) {
	const concurrent = 30 // how many browsers will run at the same time
	const num = 300       // how many browsers we will launch

	limiter := make(chan int, concurrent)

	s := got.New(b).Serve()

	// docker run --rm -p 7317:7317 ghcr.io/go-rod/rod
	s.HostURL.Host = "host.docker.internal"

	s.Route("/", ".html", `<html><body>
		ok
	</body><script>
		function wait() {
			return new Promise(r => setTimeout(r, 1000 * Math.random()))
		}
	</script></html>`)

	wg := &sync.WaitGroup{}
	wg.Add(num)
	for i := 0; i < num; i++ {
		limiter <- 0

		go func() {
			utils.Sleep(rand.Float64())

			ctx, cancel := context.WithCancel(context.Background())
			defer func() {
				go func() {
					utils.Sleep(2)
					cancel()
				}()
			}()

			l := launcher.MustNewManaged("")
			u, h := l.ClientHeader()
			browser := rod.New().Client(cdp.MustStartWithURL(ctx, u, h)).MustConnect()
			page := browser.MustPage()
			wait := page.MustWaitNavigation()
			page.MustNavigate(s.URL())
			wait()
			page.MustEval(`wait()`)

			if rand.Int()%10 == 0 {
				// 10% we will drop the websocket connection without call the api to gracefully close the browser
				cancel()
			} else {
				browser.MustClose()
			}

			wg.Done()
			<-limiter
		}()
	}
	wg.Wait()
}