def DispatchNotifications(self, timeout): current_time = self._mock_timer.time() if not self._responses: self._mock_timer.SetTime(current_time + timeout + 1) raise websocket.WebSocketTimeoutException() response, time = self._responses[0] if time - current_time > timeout: self._mock_timer.SetTime(current_time + timeout + 1) raise websocket.WebSocketTimeoutException() self._responses.pop(0) self._mock_timer.SetTime(time + 1) self._handler(response)
def DispatchNotifications(self, timeout): current_time = self._mock_timer.time() if not self._notifications: self._mock_timer.SetTime(current_time + timeout + 1) raise websocket.WebSocketTimeoutException() response, time, kind = self._notifications[0] if time - current_time > timeout: self._mock_timer.SetTime(current_time + timeout + 1) raise websocket.WebSocketTimeoutException() self._notifications.pop(0) self._mock_timer.SetTime(time + 1) if kind == self._NOTIFICATION_EVENT: self._handler(response) elif kind == self._NOTIFICATION_CALLBACK: callback = self._pending_callbacks.get(response['method']).pop(0) callback(response) else: raise Exception('Unexpected response type')
def recv(self): if not self._responses: raise Exception('No more recorded responses.') response, time = self._responses.pop(0) current_time = self._mock_timer.time() if self._timeout is not None and time - current_time > self._timeout: self._mock_timer.SetTime(current_time + self._timeout + 1) raise websocket.WebSocketTimeoutException() self._mock_timer.SetTime(time) return response
def recv(self): # pylint: disable=invalid-name if not self._responses: raise Exception('No more recorded responses.') response, time = self._responses.pop(0) current_time = self._fake_timer.time() if self._timeout is not None and time - current_time > self._timeout: self._fake_timer.SetTime(current_time + self._timeout + 1) raise websocket.WebSocketTimeoutException() self._fake_timer.SetTime(time) if isinstance(response, Exception): raise response return response