Пример #1
0
class CGTestCase(unittest.TestCase):
    def setUp(self):
        self.pool = HTTPConnectionPool(reactor, persistent=False)
        self.pubnub = PubNubTwisted(pnconf, reactor=reactor, pool=self.pool)

    def tearDown(self):
        return self.pool.closeCachedConnections()

    def assert_valid_cg_envelope(self, envelope, type):
        self.assertIsInstance(envelope.result, type)

    @inlineCallbacks
    @pn_vcr.use_cassette(
        'tests/integrational/fixtures/twisted/groups/add_single_channel.yaml',
        filter_query_parameters=['uuid'])
    def test_adding_channel(self):
        channel = 'cgttc'
        group = 'cgttg'

        envelope = yield self.pubnub.add_channel_to_channel_group() \
            .channels(channel).channel_group(group).deferred()

        self.assert_valid_cg_envelope(envelope, PNChannelGroupsAddChannelResult)

        returnValue(envelope)

    @inlineCallbacks
    @pn_vcr.use_cassette(
        'tests/integrational/fixtures/twisted/groups/remove_single_channel.yaml',
        filter_query_parameters=['uuid'])
    def test_removing_channel(self):
        channel = 'cgttc'
        group = 'cgttg'

        envelope = yield self.pubnub.remove_channel_from_channel_group() \
            .channels(channel).channel_group(group).deferred()

        self.assert_valid_cg_envelope(envelope, PNChannelGroupsRemoveChannelResult)

        returnValue(envelope)

    @inlineCallbacks
    @pn_vcr.use_cassette(
        'tests/integrational/fixtures/twisted/groups/add_channels.yaml',
        filter_query_parameters=['uuid'])
    def test_adding_channels(self):
        channel = ['cgttc0', 'cgttc1']
        group = 'cgttg'

        envelope = yield self.pubnub.add_channel_to_channel_group() \
            .channels(channel).channel_group(group).deferred()

        self.assert_valid_cg_envelope(envelope, PNChannelGroupsAddChannelResult)

        returnValue(envelope)

    @inlineCallbacks
    @pn_vcr.use_cassette(
        'tests/integrational/fixtures/twisted/groups/remove_channels.yaml',
        filter_query_parameters=['uuid'])
    def test_removing_channels(self):
        channel = ['cgttc0', 'cgttc1']
        group = 'cgttg'

        envelope = yield self.pubnub.remove_channel_from_channel_group() \
            .channels(channel).channel_group(group).deferred()

        self.assert_valid_cg_envelope(envelope, PNChannelGroupsRemoveChannelResult)

        returnValue(envelope)

    @inlineCallbacks
    @pn_vcr.use_cassette(
        'tests/integrational/fixtures/twisted/groups/list_channels.yaml',
        filter_query_parameters=['uuid'])
    def test_list_channels(self):
        group = 'cgttg'

        envelope = yield self.pubnub.list_channels_in_channel_group() \
            .channel_group(group).deferred()

        self.assert_valid_cg_envelope(envelope, PNChannelGroupsListResult)

        returnValue(envelope)
Пример #2
0
class CGTestCase(unittest.TestCase):
    def setUp(self):
        self.pool = HTTPConnectionPool(reactor, persistent=False)
        self.pubnub = PubNubTwisted(pnconf, reactor=reactor, pool=self.pool)

    def tearDown(self):
        return self.pool.closeCachedConnections()

    def assert_valid_cg_envelope(self, envelope, type):
        self.assertIsInstance(envelope.result, type)

    @inlineCallbacks
    @pn_vcr.use_cassette(
        "tests/integrational/fixtures/twisted/groups/add_single_channel.yaml", filter_query_parameters=["uuid"]
    )
    def test_adding_channel(self):
        channel = "cgttc"
        group = "cgttg"

        envelope = yield self.pubnub.add_channel_to_channel_group().channels(channel).channel_group(group).deferred()

        self.assert_valid_cg_envelope(envelope, PNChannelGroupsAddChannelResult)

        returnValue(envelope)

    @inlineCallbacks
    @pn_vcr.use_cassette(
        "tests/integrational/fixtures/twisted/groups/remove_single_channel.yaml", filter_query_parameters=["uuid"]
    )
    def test_removing_channel(self):
        channel = "cgttc"
        group = "cgttg"

        envelope = (
            yield self.pubnub.remove_channel_from_channel_group().channels(channel).channel_group(group).deferred()
        )

        self.assert_valid_cg_envelope(envelope, PNChannelGroupsRemoveChannelResult)

        returnValue(envelope)

    @inlineCallbacks
    @pn_vcr.use_cassette(
        "tests/integrational/fixtures/twisted/groups/add_channels.yaml", filter_query_parameters=["uuid"]
    )
    def test_adding_channels(self):
        channel = ["cgttc0", "cgttc1"]
        group = "cgttg"

        envelope = yield self.pubnub.add_channel_to_channel_group().channels(channel).channel_group(group).deferred()

        self.assert_valid_cg_envelope(envelope, PNChannelGroupsAddChannelResult)

        returnValue(envelope)

    @inlineCallbacks
    @pn_vcr.use_cassette(
        "tests/integrational/fixtures/twisted/groups/remove_channels.yaml", filter_query_parameters=["uuid"]
    )
    def test_removing_channels(self):
        channel = ["cgttc0", "cgttc1"]
        group = "cgttg"

        envelope = (
            yield self.pubnub.remove_channel_from_channel_group().channels(channel).channel_group(group).deferred()
        )

        self.assert_valid_cg_envelope(envelope, PNChannelGroupsRemoveChannelResult)

        returnValue(envelope)

    @inlineCallbacks
    @pn_vcr.use_cassette(
        "tests/integrational/fixtures/twisted/groups/list_channels.yaml", filter_query_parameters=["uuid"]
    )
    def test_list_channels(self):
        group = "cgttg"

        envelope = yield self.pubnub.list_channels_in_channel_group().channel_group(group).deferred()

        self.assert_valid_cg_envelope(envelope, PNChannelGroupsListResult)

        returnValue(envelope)