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