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()
