Merge pull request #2 from publicapi/return-status-code

api.entry: Return the status code when scheduled

Author: Vraj Mohan

aweber_api/entry.py

@@ -135,7 +135,7 @@ def schedule_broadcast(self, bc_id, scheduled_for):
         self._method_for('list')
         body = {'scheduled_for': scheduled_for}
         url = '{0}/broadcasts/{1}/schedule'.format(self.url, bc_id)
-        self.adapter.request('POST', url, body, response='headers')
+        return self.adapter.request('POST', url, body, response='status')
 
     def _get_total_size(self, uri, **kwargs):
         """Get actual total size number from total_size_link."""

tests/test_aweber_entry.py

@@ -123,10 +123,13 @@ class TestListScheduleBroadcast(ListTestCase):
     def setUp(self):
         super(TestListScheduleBroadcast, self).setUp()
         self.aweber.adapter.requests = []
-        subscribers = self.list_.schedule_broadcast(
+        self.status = self.list_.schedule_broadcast(
             bc_id=2, scheduled_for='2014-09-06 18:55:00')
         self.request = self.aweber.adapter.requests[0]
 
+    def test_should_return_status(self):
+        self.assertEqual(int(self.status), 201)
+
     def test_should_make_post_request(self):
         self.assertEqual(self.request['method'], 'POST')