From c67de716ca85c04c2b7a4986b621b9a73e58bb98 Mon Sep 17 00:00:00 2001 From: Jay Date: Wed, 15 Apr 2026 14:11:19 -0400 Subject: [PATCH] Wrote unknown url error test for Remove. --- pool_test.go | 20 ++++++++++++++++---- 1 file changed, 16 insertions(+), 4 deletions(-) diff --git a/pool_test.go b/pool_test.go index 1315092..cf47524 100644 --- a/pool_test.go +++ b/pool_test.go @@ -147,12 +147,24 @@ func TestPoolRemove(t *testing.T) { assert.False(t, ok, "connection is still in pool") }) - t.Run("normalizes url before lookup", func(t *testing.T) { - - }) - t.Run("unknown url 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") + + // remove unknown connection + err = pool.Remove("wss://unknown") + assert.ErrorContains(t, err, "connection not found") }) t.Run("closed pool returns error", func(t *testing.T) {