assertThrows(TransportException.class, () -> transport.connect(1000));
        }

        @Test
        @DisplayName("disconnect should succeed when connected")
        void shouldDisconnectWhenConnected() throws IOException {
            transport.setState(3); // Connected
            assertFalse(transport.disconnect(false));
            assertEquals(6, transport.state); // Should be disconnected
        }