def _get_timeouting_connect_mock(self): stomp = Stomp(CONFIG) stomp._transportFactory = mock.Mock() transport = stomp._transportFactory.return_value = mock.Mock() transport.host = 'mock' transport.port = 0 transport.canRead.return_value = False return stomp
def _get_connect_mock(self, receive=None, config=None): stomp = Stomp(config or CONFIG) stomp._transportFactory = Mock() transport = stomp._transportFactory.return_value = Mock() transport.host = 'mock' transport.port = 0 if receive: transport.receive.return_value = receive return stomp