Add BrowserPool TryGet test

go-rod · Apr 30, 2024 · 95cc251 · 95cc251
1 parent 93a60a7
commit 95cc251
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/browser_test.go b/browser_test.go
@@ -449,6 +449,23 @@ func TestBrowserPool(_ *testing.T) {
 	})
 }
 
+func TestBrowserPool_TryGet(t *testing.T) {
+	pool := rod.NewBrowserPool(3)
+	create := func() (*rod.Browser, error) {
+		b := rod.New()
+		err := b.Connect()
+		return b, err
+	}
+	b, err := pool.TryGet(create)
+	if err != nil {
+		t.Fatal(err)
+	}
+	pool.Put(b)
+	pool.Cleanup(func(p *rod.Browser) {
+		p.MustClose()
+	})
+}
+
 func TestOldBrowser(t *testing.T) {
 	t.Skip()