def send_msg(self, msg): self._conn.send(msg) socks = dict(self.poller.poll(self.timeout)) if socks.get(self._conn) == zmq.POLLIN: recv_msg = self._conn.recv() response = self.parse_response_msg(recv_msg) return response else: self._conn.setsockopt(zmq.LINGER, 0) self._conn.close() self.poller.unregister(self._conn) self.__conn = None raise exceptions.RequestTimeout()
def always_fails(): self.counter += 1 raise exceptions.RequestTimeout()
def fails_once(): self.counter += 1 if self.counter < 2: raise exceptions.RequestTimeout() else: return 'success'