Implement thread-id option

2017-03-05 14:41:00 +03:00
parent 7fe9d6aa91
commit de18e2990c
2 changed files with 80 additions and 1 deletions
@@ -0,0 +1,76 @@
+# pylint: disable=protected-access
+
+from unittest import TestCase
+
+from apns2.payload import Payload, PayloadAlert
+
+
+class PayloadTestCase(TestCase):
+
+    def setUp(self):
+        self.payload_alert = payload_alert = PayloadAlert(
+            title='title', title_localized_key='loc_k', title_localized_args='loc_a',
+            body='body', body_localized_key='body_loc_k', body_localized_args='body_loc_a',
+            action_localized_key='ac_loc_k', action='send',
+            launch_image='img')
+
+    def test_payload(self):
+        payload = Payload(
+            alert='my_alert', badge=2, sound='chime',
+            content_available=1, mutable_content=3,
+            category='my_category', url_args='args', custom={'extra': 'something'}, thread_id=42)
+        self.assertEqual(payload.dict(), {
+            'aps': {
+                'alert': 'my_alert',
+                'badge': 2,
+                'sound': 'chime',
+                'content-available': 1,
+                'mutable-content': 1,
+                'thread-id': 42,
+                'category': 'my_category',
+                'url-args': 'args'
+            },
+            'extra': 'something'
+        })
+
+    def test_payload_with_payload_alert(self):
+        payload = Payload(
+            alert=self.payload_alert, badge=2, sound='chime',
+            content_available=1, mutable_content=1,
+            category='my_category', url_args='args', custom={'extra': 'something'}, thread_id=42)
+        self.assertEqual(payload.dict(), {
+            'aps': {
+                'alert': {
+                    'title': 'title',
+                    'title-loc-key': 'loc_k',
+                    'title-loc-args': 'loc_a',
+                    'body': 'body',
+                    'loc-key': 'body_loc_k',
+                    'loc-args': 'body_loc_a',
+                    'action-loc-key': 'ac_loc_k',
+                    'action': 'send',
+                    'launch-image': 'img'
+                },
+                'badge': 2,
+                'sound': 'chime',
+                'content-available': 1,
+                'mutable-content': 1,
+                'thread-id': 42,
+                'category': 'my_category',
+                'url-args': 'args',
+            },
+            'extra': 'something'
+        })
+
+    def test_payload_alert(self):
+        self.assertEqual(self.payload_alert.dict(), {
+            'title': 'title',
+            'title-loc-key': 'loc_k',
+            'title-loc-args': 'loc_a',
+            'body': 'body',
+            'loc-key': 'body_loc_k',
+            'loc-args': 'body_loc_a',
+            'action-loc-key': 'ac_loc_k',
+            'action': 'send',
+            'launch-image': 'img'
+        })