class TestRemoveChannelToChannelGroup(unittest.TestCase): def setUp(self): self.pubnub = MagicMock(spec=PubNub, config=pnconf, sdk_name=sdk_name, uuid=None, _get_token=lambda: None) self.pubnub.uuid = "UUID_RemoveChannelToCGTest" self.pubnub._telemetry_manager = TelemetryManager() self.remove = RemoveChannelFromChannelGroup(self.pubnub) def test_remove_single_channel(self): self.remove.channels('ch').channel_group('gr') self.assertEqual( self.remove.build_path(), RemoveChannelFromChannelGroup.REMOVE_PATH % (pnconf.subscribe_key, "gr")) self.assertEqual(self.remove.build_params_callback()({}), { 'pnsdk': sdk_name, 'uuid': self.pubnub.uuid, 'remove': "ch" }) self.assertEqual(self.remove._channels, ['ch']) def test_remove_multiple_channels(self): self.remove.channels(['ch1', 'ch2']).channel_group('gr') self.assertEqual( self.remove.build_path(), RemoveChannelFromChannelGroup.REMOVE_PATH % (pnconf.subscribe_key, "gr")) self.assertEqual(self.remove.build_params_callback()({}), { 'pnsdk': sdk_name, 'uuid': self.pubnub.uuid, 'remove': "ch1,ch2" }) self.assertEqual(self.remove._channels, ['ch1', 'ch2'])
class TestRemoveChannelToChannelGroup(unittest.TestCase): def setUp(self): self.pubnub = MagicMock( spec=PubNub, config=pnconf, sdk_name=sdk_name, uuid=None ) self.pubnub.uuid = "UUID_RemoveChannelToCGTest" self.remove = RemoveChannelFromChannelGroup(self.pubnub) def test_remove_single_channel(self): self.remove.channels('ch').channel_group('gr') self.assertEquals(self.remove.build_path(), RemoveChannelFromChannelGroup.REMOVE_PATH % ( pnconf.subscribe_key, "gr")) self.assertEqual(self.remove.build_params_callback()({}), { 'pnsdk': sdk_name, 'uuid': self.pubnub.uuid, 'remove': "ch" }) self.assertEqual(self.remove._channels, ['ch']) def test_remove_multiple_channels(self): self.remove.channels(['ch1', 'ch2']).channel_group('gr') self.assertEquals(self.remove.build_path(), RemoveChannelFromChannelGroup.REMOVE_PATH % ( pnconf.subscribe_key, "gr")) self.assertEqual(self.remove.build_params_callback()({}), { 'pnsdk': sdk_name, 'uuid': self.pubnub.uuid, 'remove': "ch1,ch2" }) self.assertEqual(self.remove._channels, ['ch1', 'ch2'])