assertTrue(transport.connect(1000));
        }

        @Test
        @DisplayName("connect should throw TransportException on connection failure")
        void shouldThrowOnConnectionFailure() {
            transport.setState(0); // Not connected
            transport.setConnectFails(true);
            assertThrows(TransportException.class, () -> transport.connect(1000));
        }