def setUp(self):
     self.pubnub = MagicMock(spec=PubNub,
                             config=pnconf,
                             sdk_name=sdk_name,
                             uuid=None)
     self.pubnub.uuid = "UUID_GetStateTest"
     self.pubnub._telemetry_manager = TelemetryManager()
     self.get_state = GetState(self.pubnub)
示例#2
0
 def setUp(self):
     self.pubnub = MagicMock(
         spec=PubNub,
         config=pnconf,
         sdk_name=sdk_name,
         uuid=None
     )
     self.pubnub.uuid = "UUID_GetStateTest"
     self.get_state = GetState(self.pubnub)
示例#3
0
class TestGetState(unittest.TestCase):
    def setUp(self):
        self.pubnub = MagicMock(
            spec=PubNub,
            config=pnconf,
            sdk_name=sdk_name,
            uuid=None
        )
        self.pubnub.uuid = "UUID_GetStateTest"
        self.pubnub._telemetry_manager = TelemetryManager()
        self.get_state = GetState(self.pubnub)

    def test_get_state_single_channel(self):
        self.get_state.channels('ch')

        self.assertEquals(self.get_state.build_path(), GetState.GET_STATE_PATH % (pnconf.subscribe_key,
                                                                                  "ch",
                                                                                  self.pubnub.uuid))

        self.assertEqual(self.get_state.build_params_callback()({}), {
            'pnsdk': sdk_name,
            'uuid': self.pubnub.uuid,
        })

        self.assertEqual(self.get_state._channels, ['ch'])

    def test_get_state_single_group(self):
        self.get_state.channel_groups('gr')

        self.assertEquals(self.get_state.build_path(), GetState.GET_STATE_PATH % (pnconf.subscribe_key,
                                                                                  ",",
                                                                                  self.pubnub.uuid))

        self.assertEqual(self.get_state.build_params_callback()({}), {
            'pnsdk': sdk_name,
            'uuid': self.pubnub.uuid,
            'channel-group': 'gr'
        })

        assert len(self.get_state._channels) == 0
        self.assertEqual(self.get_state._groups, ['gr'])
class TestGetState(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_GetStateTest"
        self.pubnub._telemetry_manager = TelemetryManager()
        self.get_state = GetState(self.pubnub)

    def test_get_state_single_channel(self):
        self.get_state.channels('ch')

        self.assertEqual(
            self.get_state.build_path(), GetState.GET_STATE_PATH %
            (pnconf.subscribe_key, "ch", self.pubnub.uuid))

        self.assertEqual(self.get_state.build_params_callback()({}), {
            'pnsdk': sdk_name,
            'uuid': self.pubnub.uuid,
        })

        self.assertEqual(self.get_state._channels, ['ch'])

    def test_get_state_single_group(self):
        self.get_state.channel_groups('gr')

        self.assertEqual(
            self.get_state.build_path(), GetState.GET_STATE_PATH %
            (pnconf.subscribe_key, ",", self.pubnub.uuid))

        self.assertEqual(self.get_state.build_params_callback()({}), {
            'pnsdk': sdk_name,
            'uuid': self.pubnub.uuid,
            'channel-group': 'gr'
        })

        assert len(self.get_state._channels) == 0
        self.assertEqual(self.get_state._groups, ['gr'])