def setUp(self):
self.pubnub = MagicMock(spec=PubNub,
config=pnconf_copy(),
sdk_name=sdk_name)
self.pubnub.uuid = "UUID_HeartbeatUnitTest"
self.hb = Heartbeat(self.pubnub)
self.pubnub.config.set_presence_timeout(20)
def setUp(self):
self.pubnub = MagicMock(spec=PubNub,
config=pnconf_copy(),
sdk_name=sdk_name,
_get_token=lambda: None)
self.pubnub.uuid = "UUID_HeartbeatUnitTest"
self.hb = Heartbeat(self.pubnub)
self.pubnub._telemetry_manager = TelemetryManager()
self.pubnub.config.set_presence_timeout(20)
def setUp(self):
self.pubnub = MagicMock(
spec=PubNub,
config=pnconf_copy(),
sdk_name=sdk_name
)
self.pubnub.uuid = "UUID_HeartbeatUnitTest"
self.hb = Heartbeat(self.pubnub)
self.pubnub.config.set_presence_timeout(20)
class TestHeartbeat(unittest.TestCase):
def setUp(self):
self.pubnub = MagicMock(spec=PubNub,
config=pnconf_copy(),
sdk_name=sdk_name)
self.pubnub.uuid = "UUID_HeartbeatUnitTest"
self.hb = Heartbeat(self.pubnub)
self.pubnub._telemetry_manager = TelemetryManager()
self.pubnub.config.set_presence_timeout(20)
def test_sub_single_channel(self):
self.hb.channels('ch')
self.assertEqual(
self.hb.build_path(),
Heartbeat.HEARTBEAT_PATH % (pnconf.subscribe_key, 'ch'))
self.assertEqual(self.hb.build_params_callback()({}), {
'pnsdk': sdk_name,
'uuid': self.pubnub.uuid,
'heartbeat': '20'
})
self.assertEqual(self.hb._channels, ['ch'])
def test_hb_multiple_channels_using_list(self):
self.hb.channels(['ch1', 'ch2', 'ch3'])
self.assertEqual(
self.hb.build_path(),
Heartbeat.HEARTBEAT_PATH % (pnconf.subscribe_key, "ch1,ch2,ch3"))
self.assertEqual(self.hb.build_params_callback()({}), {
'pnsdk': sdk_name,
'uuid': self.pubnub.uuid,
'heartbeat': '20'
})
self.assertEqual(self.hb._channels, ['ch1', 'ch2', 'ch3'])
def test_hb_single_group(self):
self.hb.channel_groups("gr")
self.assertEqual(
self.hb.build_path(),
Heartbeat.HEARTBEAT_PATH % (pnconf.subscribe_key, ","))
self.assertEqual(
self.hb.build_params_callback()({}), {
'channel-group': 'gr',
'pnsdk': sdk_name,
'uuid': self.pubnub.uuid,
'heartbeat': '20'
})
self.assertEqual(self.hb._groups, ['gr'])
def test_hb_multiple_groups_using_list(self):
self.hb.channel_groups(['gr1', 'gr2', 'gr3'])
self.assertEqual(
self.hb.build_path(),
Heartbeat.HEARTBEAT_PATH % (pnconf.subscribe_key, ","))
self.assertEqual(
self.hb.build_params_callback()({}), {
'channel-group': 'gr1,gr2,gr3',
'pnsdk': sdk_name,
'uuid': self.pubnub.uuid,
'heartbeat': '20'
})
self.assertEqual(self.hb._groups, ['gr1', 'gr2', 'gr3'])
def test_hb_with_state(self):
state = {"name": "Alex", "count": 7}
self.hb.channels('ch1,ch2').state(state)
self.assertEqual(
self.hb.build_path(),
Heartbeat.HEARTBEAT_PATH % (pnconf.subscribe_key, "ch1,ch2"))
params = self.hb.build_params_callback()({})
params['state'] = json.loads(urllib.parse.unquote(params['state']))
self.assertEqual(
params, {
'pnsdk': sdk_name,
'uuid': self.pubnub.uuid,
'heartbeat': '20',
'state': state
})
self.assertEqual(self.hb._groups, [])
self.assertEqual(self.hb._channels, ['ch1', 'ch2'])
class TestHeartbeat(unittest.TestCase):
def setUp(self):
self.pubnub = MagicMock(
spec=PubNub,
config=pnconf_copy(),
sdk_name=sdk_name
)
self.pubnub.uuid = "UUID_HeartbeatUnitTest"
self.hb = Heartbeat(self.pubnub)
self.pubnub._telemetry_manager = TelemetryManager()
self.pubnub.config.set_presence_timeout(20)
def test_sub_single_channel(self):
self.hb.channels('ch')
self.assertEquals(self.hb.build_path(), Heartbeat.HEARTBEAT_PATH
% (pnconf.subscribe_key, 'ch'))
self.assertEqual(self.hb.build_params_callback()({}), {
'pnsdk': sdk_name,
'uuid': self.pubnub.uuid,
'heartbeat': '20'
})
self.assertEqual(self.hb._channels, ['ch'])
def test_hb_multiple_channels_using_list(self):
self.hb.channels(['ch1', 'ch2', 'ch3'])
self.assertEquals(self.hb.build_path(), Heartbeat.HEARTBEAT_PATH
% (pnconf.subscribe_key, "ch1,ch2,ch3"))
self.assertEqual(self.hb.build_params_callback()({}), {
'pnsdk': sdk_name,
'uuid': self.pubnub.uuid,
'heartbeat': '20'
})
self.assertEqual(self.hb._channels, ['ch1', 'ch2', 'ch3'])
def test_hb_single_group(self):
self.hb.channel_groups("gr")
self.assertEquals(self.hb.build_path(), Heartbeat.HEARTBEAT_PATH
% (pnconf.subscribe_key, ","))
self.assertEqual(self.hb.build_params_callback()({}), {
'channel-group': 'gr',
'pnsdk': sdk_name,
'uuid': self.pubnub.uuid,
'heartbeat': '20'
})
self.assertEqual(self.hb._groups, ['gr'])
def test_hb_multiple_groups_using_list(self):
self.hb.channel_groups(['gr1', 'gr2', 'gr3'])
self.assertEquals(self.hb.build_path(), Heartbeat.HEARTBEAT_PATH
% (pnconf.subscribe_key, ","))
self.assertEqual(self.hb.build_params_callback()({}), {
'channel-group': 'gr1,gr2,gr3',
'pnsdk': sdk_name,
'uuid': self.pubnub.uuid,
'heartbeat': '20'
})
self.assertEqual(self.hb._groups, ['gr1', 'gr2', 'gr3'])
def test_hb_with_state(self):
import six
state = {"name": "Alex", "count": 7}
self.hb.channels('ch1,ch2').state(state)
self.assertEquals(self.hb.build_path(), Heartbeat.HEARTBEAT_PATH
% (pnconf.subscribe_key, "ch1,ch2"))
params = self.hb.build_params_callback()({})
params['state'] = json.loads(six.moves.urllib.parse.unquote(params['state']))
self.assertEqual(params, {
'pnsdk': sdk_name,
'uuid': self.pubnub.uuid,
'heartbeat': '20',
'state': state
})
self.assertEqual(self.hb._groups, [])
self.assertEqual(self.hb._channels, ['ch1', 'ch2'])
