def setUp(self): self.pubnub = MagicMock(spec=PubNub, config=pnconf, sdk_name=sdk_name, uuid=None) self.pubnub.uuid = "UUID_AddChannelsTest" self.add_channels = AddChannelsToPush(self.pubnub)
def setUp(self): self.pubnub = MagicMock(spec=PubNub, config=pnconf, sdk_name=sdk_name, uuid=None, _get_token=lambda: None) self.pubnub.uuid = "UUID_AddChannelsTest" self.pubnub._telemetry_manager = TelemetryManager() self.add_channels = AddChannelsToPush(self.pubnub)
def setUp(self): self.pubnub = MagicMock( spec=PubNub, config=pnconf, sdk_name=sdk_name, uuid=None ) self.pubnub.uuid = "UUID_AddChannelsTest" self.add_channels = AddChannelsToPush(self.pubnub)
class TestAddChannelsFromPush(unittest.TestCase): def setUp(self): self.pubnub = MagicMock(spec=PubNub, config=pnconf, sdk_name=sdk_name, uuid=None) self.pubnub.uuid = "UUID_AddChannelsTest" self.pubnub._telemetry_manager = TelemetryManager() self.add_channels = AddChannelsToPush(self.pubnub) def test_push_add_single_channel(self): self.add_channels.channels(['ch']).push_type( pubnub.enums.PNPushType.APNS).device_id("coolDevice") params = (pnconf.subscribe_key, "coolDevice") self.assertEqual(self.add_channels.build_path(), AddChannelsToPush.ADD_PATH % params) self.assertEqual( self.add_channels.build_params_callback()({}), { 'pnsdk': sdk_name, 'uuid': self.pubnub.uuid, 'type': 'apns', 'add': 'ch' }) self.assertEqual(self.add_channels._channels, ['ch']) def test_push_add_multiple_channels(self): self.add_channels.channels(['ch1', 'ch2']).push_type( pubnub.enums.PNPushType.MPNS).device_id("coolDevice") params = (pnconf.subscribe_key, "coolDevice") self.assertEqual(self.add_channels.build_path(), AddChannelsToPush.ADD_PATH % params) self.assertEqual( self.add_channels.build_params_callback()({}), { 'pnsdk': sdk_name, 'uuid': self.pubnub.uuid, 'type': 'mpns', 'add': 'ch1,ch2' }) self.assertEqual(self.add_channels._channels, ['ch1', 'ch2']) def test_push_add_google(self): self.add_channels.channels(['ch1', 'ch2', 'ch3']).push_type( pubnub.enums.PNPushType.GCM).device_id("coolDevice") params = (pnconf.subscribe_key, "coolDevice") self.assertEqual(self.add_channels.build_path(), AddChannelsToPush.ADD_PATH % params) self.assertEqual( self.add_channels.build_params_callback()({}), { 'pnsdk': sdk_name, 'uuid': self.pubnub.uuid, 'type': 'gcm', 'add': 'ch1,ch2,ch3' }) self.assertEqual(self.add_channels._channels, ['ch1', 'ch2', 'ch3']) def test_push_add_single_channel_apns2(self): self.add_channels.channels(['ch']).push_type(pubnub.enums.PNPushType.APNS2).device_id("coolDevice")\ .environment(pubnub.enums.PNPushEnvironment.PRODUCTION).topic("testTopic") params = (pnconf.subscribe_key, "coolDevice") self.assertEqual(self.add_channels.build_path(), AddChannelsToPush.ADD_PATH_APNS2 % params) self.assertEqual( self.add_channels.build_params_callback()({}), { 'pnsdk': sdk_name, 'uuid': self.pubnub.uuid, 'add': 'ch', 'environment': pubnub.enums.PNPushEnvironment.PRODUCTION, 'topic': 'testTopic' }) self.assertEqual(self.add_channels._channels, ['ch'])
class TestAddChannelsFromPush(unittest.TestCase): def setUp(self): self.pubnub = MagicMock( spec=PubNub, config=pnconf, sdk_name=sdk_name, uuid=None ) self.pubnub.uuid = "UUID_AddChannelsTest" self.pubnub._telemetry_manager = TelemetryManager() self.add_channels = AddChannelsToPush(self.pubnub) def test_push_add_single_channel(self): self.add_channels.channels(['ch']).push_type(pubnub.enums.PNPushType.APNS).device_id("coolDevice") params = (pnconf.subscribe_key, "coolDevice") self.assertEquals(self.add_channels.build_path(), AddChannelsToPush.ADD_PATH % params) self.assertEqual(self.add_channels.build_params_callback()({}), { 'pnsdk': sdk_name, 'uuid': self.pubnub.uuid, 'type': 'apns', 'add': 'ch' }) self.assertEqual(self.add_channels._channels, ['ch']) def test_push_add_multiple_channels(self): self.add_channels.channels(['ch1', 'ch2']).push_type(pubnub.enums.PNPushType.MPNS).device_id("coolDevice") params = (pnconf.subscribe_key, "coolDevice") self.assertEquals(self.add_channels.build_path(), AddChannelsToPush.ADD_PATH % params) self.assertEqual(self.add_channels.build_params_callback()({}), { 'pnsdk': sdk_name, 'uuid': self.pubnub.uuid, 'type': 'mpns', 'add': 'ch1,ch2' }) self.assertEqual(self.add_channels._channels, ['ch1', 'ch2']) def test_push_add_google(self): self.add_channels.channels(['ch1', 'ch2', 'ch3']).push_type(pubnub.enums.PNPushType.GCM).device_id("coolDevice") params = (pnconf.subscribe_key, "coolDevice") self.assertEquals(self.add_channels.build_path(), AddChannelsToPush.ADD_PATH % params) self.assertEqual(self.add_channels.build_params_callback()({}), { 'pnsdk': sdk_name, 'uuid': self.pubnub.uuid, 'type': 'gcm', 'add': 'ch1,ch2,ch3' }) self.assertEqual(self.add_channels._channels, ['ch1', 'ch2', 'ch3'])