def _new_heartbeat_frame(): """Return a new heartbeat frame. :rtype pika.frame.Heartbeat """ return frame.Heartbeat()
def _new_heartbeat_frame(self): #pragma: no cover """Return a new heartbeat frame. :returns: pika.frame.Heartbeat """ return frame.Heartbeat()
def heartbeat_marshal_test(self): heartbeat = frame.Heartbeat() self.assertEqual(heartbeat.marshal(), self.HEARTBEAT)
__author__ = 'gmr' import sys sys.path.insert(0, '..') import mock import unittest from pika import frame from pika import heartbeat _BYTES_RECEIVED = 1024 _BYTES_SENT = 2048 _HEARTBEAT = frame.Heartbeat() _INTERVAL = 60 def _connection_close(code, message): pass class HeartbeatCheckerTest(unittest.TestCase): 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()
def _send_heartbeat_frame(self): now = int(round(time.time())) if now > (self.last_activity_time + self.heartbeat): LOGGER.debug("Sending heartbeat frame") self._connection.send_frame(frame.Heartbeat())