From 8f8e2d8cfa548ee1f2735ea2d0fcbfbb7b5454e4 Mon Sep 17 00:00:00 2001
From: Jack Christensen <jack@jackchristensen.com>
Date: Sat, 27 Nov 2021 10:24:10 -0600
Subject: [PATCH] Add TryAcquire to stress test

---
 pool_test.go | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/pool_test.go b/pool_test.go
index 44f2a13..b0314a9 100644
--- a/pool_test.go
+++ b/pool_test.go
@@ -761,6 +761,19 @@ func TestStress(t *testing.T) {
 			time.Sleep(time.Duration(rand.Int63n(2000)) * time.Nanosecond)
 			releaseOrDestroyOrHijack(res)
 		},
+		// TryAcquire
+		func() {
+			res, err := pool.TryAcquire(context.Background())
+			if err != nil {
+				if err != puddle.ErrClosedPool && err != puddle.ErrNotAvailable {
+					assert.Failf(t, "stress TryAcquire", "pool.TryAcquire returned unexpected err: %v", err)
+				}
+				return
+			}
+
+			time.Sleep(time.Duration(rand.Int63n(100)) * time.Millisecond)
+			releaseOrDestroyOrHijack(res)
+		},
 		// AcquireAllIdle (though under heavy load this will almost certainly always get an empty slice)
 		func() {
 			resources := pool.AcquireAllIdle()