Merge pull request #43 from vivekfantain/master

Added support for dry_run

Author: geeknam

gcm/gcm.py

@@ -90,7 +90,7 @@ def __init__(self, api_key, url=GCM_URL, proxy=None):
 
 
     def construct_payload(self, registration_ids, data=None, collapse_key=None,
-                            delay_while_idle=False, time_to_live=None, is_json=True):
+                            delay_while_idle=False, time_to_live=None, is_json=True, dry_run=False):
         """
         Construct the dictionary mapping of parameters.
         Encodes the dictionary into JSON if for json requests.
@@ -125,6 +125,9 @@ def construct_payload(self, registration_ids, data=None, collapse_key=None,
         if collapse_key:
             payload['collapse_key'] = collapse_key
 
+        if dry_run:
+            payload['dry_run'] = True
+
         if is_json:
             payload = json.dumps(payload)
 
@@ -215,7 +218,7 @@ def extract_unsent_reg_ids(self, info):
         return []
 
     def plaintext_request(self, registration_id, data=None, collapse_key=None,
-                            delay_while_idle=False, time_to_live=None, retries=5):
+                            delay_while_idle=False, time_to_live=None, retries=5, dry_run=False):
         """
         Makes a plaintext request to GCM servers
 
@@ -230,7 +233,7 @@ def plaintext_request(self, registration_id, data=None, collapse_key=None,
 
         payload = self.construct_payload(
             registration_id, data, collapse_key,
-            delay_while_idle, time_to_live, False
+            delay_while_idle, time_to_live, False, dry_run
         )
 
         attempt = 0
@@ -248,7 +251,7 @@ def plaintext_request(self, registration_id, data=None, collapse_key=None,
         raise IOError("Could not make request after %d attempts" % attempt)
 
     def json_request(self, registration_ids, data=None, collapse_key=None,
-                        delay_while_idle=False, time_to_live=None, retries=5):
+                        delay_while_idle=False, time_to_live=None, retries=5, dry_run=False):
         """
         Makes a JSON request to GCM servers
 
@@ -268,7 +271,7 @@ def json_request(self, registration_ids, data=None, collapse_key=None,
         for attempt in range(retries):
             payload = self.construct_payload(
                 registration_ids, data, collapse_key,
-                delay_while_idle, time_to_live
+                delay_while_idle, time_to_live, dry_run
             )
             response = self.make_request(payload, is_json=True)
             info = self.handle_json_response(response, registration_ids)

gcm/test.py

@@ -74,10 +74,10 @@ def setUp(self):
     def test_construct_payload(self):
         res = self.gcm.construct_payload(
             registration_ids=['1', '2'], data=self.data, collapse_key='foo',
-            delay_while_idle=True, time_to_live=3600, is_json=True
+            delay_while_idle=True, time_to_live=3600, is_json=True, dry_run = True
         )
         payload = json.loads(res)
-        for arg in ['registration_ids', 'data', 'collapse_key', 'delay_while_idle', 'time_to_live']:
+        for arg in ['registration_ids', 'data', 'collapse_key', 'delay_while_idle', 'time_to_live', 'dry_run']:
             self.assertIn(arg, payload)
 
     def test_json_payload(self):