python/tests/functional/push/test_list_push_provisions.py at a0c666aecb0b798388e10ae99cfee8f5baa6d0d6 · pubnub/python · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
import unittest
import pytest

from pubnub.endpoints.push.list_push_provisions import ListPushProvisions
from pubnub.enums import PNPushType
from pubnub.exceptions import PubNubException

try:
    from mock import MagicMock
except ImportError:
    from unittest.mock import MagicMock

from pubnub.pubnub import PubNub

import pubnub.enums

from tests.helper import pnconf, sdk_name


class TestListPushProvisions(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_ListChannelsInCGTest"
        self.list_push = ListPushProvisions(self.pubnub)

    def test_list_channel_group_apns(self):
        self.list_push.push_type(PNPushType.APNS).device_id('coolDevice')

        self.assertEqual(self.list_push.build_path(),
                         ListPushProvisions.LIST_PATH % (
                             pnconf.subscribe_key, "coolDevice"))

        self.assertEqual(self.list_push.build_params_callback()({}), {
            'pnsdk': sdk_name,
            'uuid': self.pubnub.uuid,
            'type': 'apns'
        })

    def test_list_channel_group_gcm(self):
        self.list_push.push_type(PNPushType.GCM).device_id('coolDevice')

        self.assertEqual(self.list_push.build_path(),
                         ListPushProvisions.LIST_PATH % (
                             pnconf.subscribe_key, "coolDevice"))

        self.assertEqual(self.list_push.build_params_callback()({}), {
            'pnsdk': sdk_name,
            'uuid': self.pubnub.uuid,
            'type': 'gcm'
        })

    def test_list_channel_group_apns2(self):
        self.list_push.push_type(PNPushType.APNS2).device_id('coolDevice')\
            .environment(pubnub.enums.PNPushEnvironment.PRODUCTION).topic("testTopic")

        self.assertEqual(self.list_push.build_path(),
                         ListPushProvisions.LIST_PATH_APNS2 % (
                             pnconf.subscribe_key, "coolDevice"))

        self.assertEqual(self.list_push.build_params_callback()({}), {
            'pnsdk': sdk_name,
            'uuid': self.pubnub.uuid,
            'environment': pubnub.enums.PNPushEnvironment.PRODUCTION,
            'topic': 'testTopic'
        })

    def test_apns2_no_topic(self):
        push = self.list_push.push_type(PNPushType.APNS2).device_id('coolDevice')

        with pytest.raises(PubNubException):
            push.validate_params()

    def test_apns2_default_environment(self):
        self.list_push.push_type(PNPushType.APNS2).device_id('coolDevice').topic("testTopic")

        self.assertEqual(self.list_push.build_path(),
                         ListPushProvisions.LIST_PATH_APNS2 % (
                             pnconf.subscribe_key, "coolDevice"))

        self.assertEqual(self.list_push.build_params_callback()({}), {
            'pnsdk': sdk_name,
            'uuid': self.pubnub.uuid,
            'environment': pubnub.enums.PNPushEnvironment.DEVELOPMENT,
            'topic': 'testTopic'
        })