From 12ae82c435211a1ece3de159f92179480ca61e8f Mon Sep 17 00:00:00 2001
From: Jay <wisehodl@pm.me>
Date: Wed, 15 Apr 2026 14:12:04 -0400
Subject: [PATCH] Wrote pool closed error test.

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

diff --git a/pool_test.go b/pool_test.go
index cf47524..c9f3c6e 100644
--- a/pool_test.go
+++ b/pool_test.go
@@ -168,7 +168,26 @@ func TestPoolRemove(t *testing.T) {
 	})
 
 	t.Run("closed pool returns error", func(t *testing.T) {
+		mockSocket := NewMockSocket()
+		mockDialer := &MockDialer{
+			DialFunc: func(string, http.Header) (Socket, *http.Response, error) {
+				return mockSocket, nil, nil
+			},
+		}
 
+		pool, err := NewPool(nil, nil)
+		assert.NoError(t, err)
+		pool.dialer = mockDialer
+
+		pool.Add("wss://peer1")
+		pool.Add("wss://peer2")
+
+		// close pool
+		pool.Close()
+
+		// attempt to remove connection
+		err = pool.Remove("wss://peer1")
+		assert.ErrorContains(t, err, "pool is closed")
 	})
 
 }