doNothing().when(transportPool).removeTransport(any());

            // When
            transportPool.removeTransport(transport);
            transportPool.removeTransport(transport2);

            // Then
            verify(transportPool).removeTransport(transport);
            verify(transportPool).removeTransport(transport2);
        }
    }

    @Nested

                try {
                    trans.ensureConnected();
                } catch (final IOException e) {
                    removeTransport(trans);
                    throw e;
                }
                return trans.acquire();
            } catch (final IOException e) {
                final String hostAddress = addr.getHostAddress();

        assertTrue(pool.contains(transport), "Transport should be in pool initially");

        // When: Remove the transport
        pool.removeTransport(transport);

        // Then: Transport should no longer be in pool
        assertFalse(pool.contains(transport), "Transport should be removed from pool");
    }

    @Test