def setUp(self): self.mock_conn = mock.Mock(spec_set=ConstructableConnection()) self.mock_conn.bytes_received = 100 self.mock_conn.bytes_sent = 100 self.mock_conn._heartbeat_checker = mock.Mock( spec=heartbeat.HeartbeatChecker) self.obj = heartbeat.HeartbeatChecker(self.mock_conn, self.INTERVAL)
def _create_heartbeat_checker(self): """Create a heartbeat checker instance if there is a heartbeat interval set. :rtype: pika.heartbeat.Heartbeat """ LOGGER.debug('Creating a HeartbeatChecker') if self.params.heartbeat: return heartbeat.HeartbeatChecker(self, self.params.heartbeat)
def _create_heartbeat_checker(self): """Create a heartbeat checker instance if there is a heartbeat interval set. :rtype: pika.heartbeat.Heartbeat """ if self.params.heartbeat is not None and self.params.heartbeat > 0: LOGGER.debug('Creating a HeartbeatChecker: %r', self.params.heartbeat) return heartbeat.HeartbeatChecker(self, self.params.heartbeat)
def setUp(self): self._connection = mock.Mock() self._connection.add_timeout = mock.Mock() self._connection.bytes_received = _BYTES_RECEIVED self._connection.bytes_sent = _BYTES_SENT self._connection.close = mock.mocksignature(_connection_close) self._connection.force_reconnect = mock.Mock() self._connection.send_frame = mock.Mock() self._heartbeat = \ heartbeat.HeartbeatChecker(self._connection, _INTERVAL) self._heartbeat._new_heartbeat_frame = \ mock.Mock(return_value=_HEARTBEAT)
def test_send_and_check_missed_heartbeats(self, close_connection): obj = heartbeat.HeartbeatChecker(self.mock_conn, self.INTERVAL) obj._idle_heartbeat_intervals = self.INTERVAL obj.send_and_check() close_connection.assert_called_once_with()
def test_send_and_check_not_closed(self, close_connection): obj = heartbeat.HeartbeatChecker(self.mock_conn, self.INTERVAL) obj.send_and_check() close_connection.assert_not_called()
def test_constructor_called_setup_timer(self, timer): obj = heartbeat.HeartbeatChecker(self.mock_conn, self.INTERVAL) timer.assert_called_once_with()
def setUp(self): self.mock_conn = mock.Mock(spec=connection.Connection) self.mock_conn.bytes_received = 100 self.mock_conn.bytes_sent = 100 self.mock_conn.heartbeat = mock.Mock(spec=heartbeat.HeartbeatChecker) self.obj = heartbeat.HeartbeatChecker(self.mock_conn, self.INTERVAL)
def test_send_and_check_start_timer(self, start_timer): obj = heartbeat.HeartbeatChecker(self.mock_conn, self.INTERVAL) obj.send_and_check() start_timer.assert_called_once_with()
def test_send_and_check_send_heartbeat_frame(self, send_heartbeat_frame): obj = heartbeat.HeartbeatChecker(self.mock_conn, self.INTERVAL) obj.send_and_check() send_heartbeat_frame.assert_called_once_with()
def test_send_and_check_update_counters(self, update_counters): obj = heartbeat.HeartbeatChecker(self.mock_conn, self.INTERVAL) obj.send_and_check() update_counters.assert_called_once_with()
def test_check_heartbeat_start_timer(self, start_check_timer): heartbeat.HeartbeatChecker(self.mock_conn, self.INTERVAL) start_check_timer.assert_called_once_with()
def test_check_heartbeat_update_counters(self, update_counters): heartbeat.HeartbeatChecker(self.mock_conn, self.INTERVAL) update_counters.assert_called_once_with()
def test_check_heartbeat_not_closed(self, close_connection): obj = heartbeat.HeartbeatChecker(self.mock_conn, self.INTERVAL) self.mock_conn.bytes_received = 128 obj._check_heartbeat() close_connection.assert_not_called()