Adding ASM Groups endpoint

Author: thinkingserious

README.rst

@@ -402,16 +402,15 @@ Tests
     pyenv install 2.6.9
     pyenv install 2.7.8
     pyenv install 3.2.6
-    pyenv local 3.2.6 2.7.8 2.6.9
-    pyenv rehash
-    virtualenv venv
-    source venv/bin/activate #or . ./activate.sh
-    python setup.py install
 
 **Run the tests:**
 
 .. code:: python
-
+    virtualenv venv
+    source venv/bin/activate #or . ./activate.sh
+    python setup.py install
+    pyenv local 3.2.6 2.7.8 2.6.9
+    pyenv rehash
     tox
 
 Deploying

example_v3_test.py

@@ -10,6 +10,12 @@
 
 client = sendgrid.SendGridAPIClient(os.environ.get('SENDGRID_API_KEY'))
 
+status, msg = client.asm_groups.get()
+print status
+print msg
+
+"""
+
 name = "My Amazing API Key"
 status, msg = client.apikeys.post(name)
 msg = json.loads(msg)
@@ -29,7 +35,6 @@
 print status
 print msg
 
-"""
 # Get a list of all valid API Keys from your account
 status, msg = client.apikeys.get()
 print status

sendgrid/client.py

@@ -12,6 +12,7 @@
 
 from .exceptions import SendGridClientError, SendGridServerError
 from .resources.apikeys import APIKeys
+from .resources.asm_groups import ASMGroups
 
 class SendGridAPIClient(object):
 
@@ -32,6 +33,7 @@ def __init__(self, apikey, **opts):
         self.proxies = opts.get('proxies', None)
 
         self.apikeys = APIKeys(self)
+        self.asm_groups = ASMGroups(self)
 
     @property
     def apikey(self):

sendgrid/resources/asm_groups.py

@@ -0,0 +1,40 @@
+class ASMGroups(object):
+    """Advanced Suppression Manager gives your recipients more control over the types of emails they want to receive 
+       by letting them opt out of messages from a certain type of email.
+       
+       Groups are specific types of email you would like your recipients to be able to unsubscribe from or subscribe to. 
+       For example: Daily Newsletters, Invoices, System Alerts.
+       """
+    
+    def __init__(self, client, **opts):
+        """
+        Constructs SendGrid ASM object.
+        
+        See https://sendgrid.com/docs/API_Reference/Web_API_v3/Advanced_Suppression_Manager/index.html and
+        https://sendgrid.com/docs/API_Reference/Web_API_v3/Advanced_Suppression_Manager/groups.html
+        """
+        self._name = None
+        self._base_endpoint = "/v3/asm/groups"
+        self._endpoint = "/v3/asm/groups"
+        self._client = client
+
+    @property
+    def base_endpoint(self):
+        return self._base_endpoint
+
+    @property
+    def endpoint(self):
+        endpoint = self._endpoint
+        return endpoint
+
+    @endpoint.setter
+    def endpoint(self, value):
+        self._endpoint = value
+
+    @property
+    def client(self):
+        return self._client
+        
+    # Retrieve all suppression groups associated with the user.
+    def get(self):
+        return self.client.get(self)

test/test_asm_groups.py

@@ -0,0 +1,85 @@
+import os
+try:
+    import unittest2 as unittest
+except ImportError:
+    import unittest
+import json
+import sys
+try:
+    from StringIO import StringIO
+except ImportError:  # Python 3
+    from io import StringIO
+try:
+    import urllib.request as urllib_request
+    from urllib.parse import urlencode
+    from urllib.error import HTTPError
+except ImportError:  # Python 2
+    import urllib2 as urllib_request
+    from urllib2 import HTTPError
+    from urllib import urlencode
+
+import sendgrid
+from sendgrid import SendGridClient, Mail
+from sendgrid.exceptions import SendGridClientError, SendGridServerError
+from sendgrid.sendgrid import HTTPError
+from sendgrid.client import SendGridAPIClient
+from sendgrid.version import __version__
+
+SG_KEY  = os.getenv('SG_KEY') or 'SENDGRID_APIKEY'
+
+class MockSendGridAPIClientRequest(SendGridAPIClient):
+    def __init__(self, apikey, **opts):
+        super(MockSendGridAPIClientRequest, self).__init__(apikey, **opts)
+        self._req = None
+        
+    def _build_request(self, url=None, json_header=False, method='GET', data=None):
+        req = urllib_request.Request(url)
+        req.get_method = lambda: method
+        req.add_header('User-Agent', self.useragent)
+        req.add_header('Authorization', 'Bearer ' + self.apikey)
+        if json_header:
+            req.add_header('Content-Type', 'application/json')
+        body = data
+        if method == 'POST':
+            response = 201
+        if method == 'PATCH':
+            response = 200
+        if method == 'DELETE':
+            response = 204
+        if method == 'GET':
+            response = 200
+        return response, body
+
+class TestSendGridAPIClient(unittest.TestCase):
+    def setUp(self):
+        self.client = MockSendGridAPIClientRequest
+        self.client = SendGridAPIClient(SG_KEY)
+
+    def test_apikey_init(self):
+        self.assertEqual(self.client.apikey, SG_KEY)
+
+    def test_useragent(self):
+        useragent = 'sendgrid/' + __version__ + ';python_v3'
+        self.assertEqual(self.client.useragent, useragent)
+
+    def test_host(self):
+        host = 'https://api.sendgrid.com'
+        self.assertEqual(self.client.host, host)
+
+class TestASMGroups(unittest.TestCase):
+    def setUp(self):
+        SendGridAPIClient = MockSendGridAPIClientRequest
+        self.client = SendGridAPIClient(SG_KEY)
+        
+    def test_apikeys_init(self):
+        self.asm_groups = self.client.asm_groups
+        self.assertEqual(self.asm_groups.base_endpoint, "/v3/asm/groups")
+        self.assertEqual(self.asm_groups.endpoint, "/v3/asm/groups")
+        self.assertEqual(self.asm_groups.client, self.client)
+
+    def test_asm_groups_get(self):
+        status, msg = self.client.apikeys.get()
+        self.assertEqual(status, 200)
+
+if __name__ == '__main__':
+    unittest.main()