Add statuses service.

jone · jone · commit 167d60066a5b · 2012-11-11T12:31:20.000+01:00
diff --git a/pygithub3/services/repos/__init__.py b/pygithub3/services/repos/__init__.py
@@ -8,6 +8,7 @@
 from .keys import Keys
 from .watchers import Watchers
 from .hooks import Hooks
+from .statuses import Statuses
 
 
 class Repo(Service):
@@ -21,6 +22,7 @@ def __init__(self, **config):
         self.keys = Keys(**config)
         self.watchers = Watchers(**config)
         self.hooks = Hooks(**config)
+        self.statuses = Statuses(**config)
         super(Repo, self).__init__(**config)
 
     def list(self, user=None, type='all', sort='full_name', direction='desc'):
diff --git a/pygithub3/services/repos/statuses.py b/pygithub3/services/repos/statuses.py
@@ -0,0 +1,54 @@
+#!/usr/bin/env python
+# -*- encoding: utf-8 -*-
+
+from . import Service
+
+
+class Statuses(Service):
+    """ Consume `Repo Statuses API
+    <http://developer.github.com/v3/repos/statuses>`_ """
+
+    def list(self, sha=None, user=None, repo=None):
+        """ Get repository's collaborators
+
+        :param str sha: Commit's sha
+        :param str user: Username
+        :param str repo: Repository
+        :returns: A :doc:`result`
+
+        .. note::
+            Remember :ref:`config precedence`
+        """
+
+        request = self.make_request('repos.statuses.list',
+                                    sha=sha, user=user, repo=repo)
+        return self._get_result(request)
+
+    def create(self, data, sha, user=None, repo=None):
+        """ Create a status
+
+        :param dict data: Input. See `github statuses doc`_
+        :param str sha: Commit's sha
+        :param str user: Username
+        :param str repo: Repository
+
+        .. note::
+            Remember :ref:`config precedence`
+
+        .. warning::
+            You must be authenticated
+
+        ::
+
+            data = {
+              "state": "success",
+              "target_url": "https://example.com/build/status",
+              "description": "The build succeeded!"
+            }
+            statuses_service.create(data, '6dcb09', user='octocat',
+                repo='oct_repo')
+        """
+
+        request = self.make_request('repos.statuses.create',
+                                    sha=sha, user=user, repo=repo, body=data)
+        return self._post(request)
diff --git a/pygithub3/tests/services/test_repos.py b/pygithub3/tests/services/test_repos.py
@@ -4,7 +4,7 @@
 from mock import patch
 
 from pygithub3.services.repos import (Repo, Collaborators, Commits, Downloads,
-    Forks, Keys, Watchers, Hooks)
+    Forks, Keys, Watchers, Hooks, Statuses)
 from pygithub3.tests.utils.base import (dummy_json, mock_response,
     mock_response_result)
 from pygithub3.tests.utils.core import TestCase
@@ -416,3 +416,25 @@ def test_DELETE(self, request_method):
         self.hs.delete(1)
         self.assertEqual(request_method.call_args[0],
                 ('delete', _('repos/oct/re_oct/hooks/1')))
+
+@dummy_json
+@patch.object(requests.sessions.Session, 'request')
+class TestStatusesService(TestCase):
+
+    def setUp(self):
+        self.ss = Statuses(user='oct', repo='re_oct')
+
+    def test_LIST(self, request_method):
+        request_method.return_value = mock_response_result()
+        self.ss.list(sha='e3bc').all()
+        self.assertEqual(request_method.call_args[0],
+                         ('get', _('repos/oct/re_oct/statuses/e3bc')))
+
+    def test_CREATE(self, request_method):
+        request_method.return_value = mock_response('post')
+        self.ss.create({"state": "success",
+                        "target_url": "https://example.com/build/status",
+                        "description": "The build succeeded!"},
+                       sha='e3bc')
+        self.assertEqual(request_method.call_args[0],
+                         ('post', _('repos/oct/re_oct/statuses/e3bc')))
