add methods for group

sheregeda · May 10, 2016 · 4b50089 · 4b50089
1 parent 7625f5e
commit 4b50089
Show file tree

Hide file tree

Showing 2 changed files with 49 additions and 0 deletions.
diff --git a/smsaero/api.py b/smsaero/api.py
@@ -92,3 +92,12 @@ def senders(self):
 
     def sign(self, sign):
         return self._request('/sign/', {'sign': sign})
+
+    def checkgroup(self):
+        return self._request('/checkgroup/', {})
+
+    def addgroup(self, group):
+        return self._request('/addgroup/', {'group': group})
+
+    def delgroup(self, group):
+        return self._request('/delgroup/', {'group': group})
diff --git a/tests/test_api.py b/tests/test_api.py
@@ -187,6 +187,46 @@ def test_senders(self):
         response = self.api.senders()
         self.assertEqual(response, [u'NEWS', u'awesome'])
 
+    @httpretty.activate
+    def test_checkgroup(self):
+        httpretty.register_uri(
+            httpretty.POST,
+            urljoin(SmsAero.URL_GATE, '/checkgroup/'),
+            body='{"reason": ["Личные контакты"], "result": "accepted "}',
+            status=200,
+            content_type='text/json',
+        )
+
+        response = self.api.checkgroup()
+        self.assertEqual(response['reason'], [u'Личные контакты'])
+
+    @httpretty.activate
+    def test_addgroup(self):
+        httpretty.register_uri(
+            httpretty.POST,
+            urljoin(SmsAero.URL_GATE, '/addgroup/'),
+            body='{"reason": "Group created", "result": "accepted"}',
+            status=200,
+            content_type='text/json',
+        )
+
+        response = self.api.addgroup('test')
+        self.assertEqual(response['result'], u'accepted')
+        self.assertEqual(response['reason'], u'Group created')
+
+    @httpretty.activate
+    def test_delgroup(self):
+        httpretty.register_uri(
+            httpretty.POST,
+            urljoin(SmsAero.URL_GATE, '/delgroup/'),
+            body='{"reason": "Group delete", "result": "accepted"}',
+            status=200,
+            content_type='text/json',
+        )
+
+        response = self.api.delgroup('test')
+        self.assertEqual(response['result'], u'accepted')
+        self.assertEqual(response['reason'], u'Group delete')
 
 if __name__ == '__main__':
     unittest.main()