Repos collaborators added

copitux · copitux · commit 464722af90cd · 2012-02-12T23:37:30.000+01:00
diff --git a/pygithub3/requests/repos/collaborators.py b/pygithub3/requests/repos/collaborators.py
@@ -0,0 +1,25 @@
+#!/usr/bin/env python
+# -*- encoding: utf-8 -*-
+
+from . import Request
+from pygithub3.resources.users import User
+
+class List(Request):
+
+    uri = 'repos/{user}/{repo}/collaborators'
+    resource = User
+
+
+class Is_collaborator(Request):
+
+    uri = 'repos/{user}/{repo}/collaborators/{collaborator}'
+
+
+class Add(Request):
+
+    uri = 'repos/{user}/{repo}/collaborators/{collaborator}'
+
+
+class Delete(Request):
+
+    uri = 'repos/{user}/{repo}/collaborators/{collaborator}'
diff --git a/pygithub3/services/repos.py b/pygithub3/services/repos.py
@@ -4,9 +4,40 @@
 from .base import Base
 
 
+class Collaborator(Base):
+
+    def list(self, user=None, repo=None):
+        request = self.make_request('repos.collaborators.list',
+            user=user or self.get_user(),
+            repo=repo or self.get_repo())
+        return self._get_result(request)
+
+    def add(self, collaborator, user=None, repo=None):
+        request = self.make_request('repos.collaborators.add',
+            collaborator=collaborator,
+            user=user or self.get_user(),
+            repo=repo or self.get_repo())
+        return self._put(request)
+
+    def is_collaborator(self, collaborator, user=None, repo=None):
+        request = self.make_request('repos.collaborators.is_collaborator',
+            collaborator=collaborator,
+            user=user or self.get_user(),
+            repo=repo or self.get_repo())
+        return self._bool(request)
+
+    def delete(self, collaborator, user=None, repo=None):
+        request = self.make_request('repos.collaborators.delete',
+            collaborator=collaborator,
+            user=user or self.get_user(),
+            repo=repo or self.get_repo())
+        self._delete(request)
+
+
 class Repo(Base):
 
     def __init__(self, **config):
+        self.collaborators = Collaborator(**config)
         super(Repo, self).__init__(**config)
 
     def list(self, user=None, type='all'):
diff --git a/pygithub3/tests/services/test_repos.py b/pygithub3/tests/services/test_repos.py
@@ -6,7 +6,7 @@
 import requests
 from mock import patch, Mock
 
-from pygithub3.services.repos import Repo
+from pygithub3.services.repos import Repo, Collaborator
 from pygithub3.resources.base import json
 from pygithub3.tests.utils.base import mock_response, mock_response_result
 from pygithub3.tests.utils.services import _, mock_json
@@ -99,7 +99,7 @@ def test_UPDATE_with_repo_in_service(self, request_method):
         self.assertEqual(request_method.call_args[0],
                          ('patch', _('repos/octocat/octocat_repo')))
 
-    """ From here I stop to do '*in_args' and '*filter' tests, I consider 
+    """ From here I stop to do '*in_args' and '*filter' tests, I consider
     that I tested it enough... """
 
     def test_LIST_contributors(self, request_method):
@@ -138,3 +138,35 @@ def test_LIST_branches(self, request_method):
         self.rs.list_branches().all()
         self.assertEqual(request_method.call_args[0],
                          ('get', _('repos/octocat/octocat_repo/branches')))
+
+
+@patch.object(requests.sessions.Session, 'request')
+class TestCollaboratorsService(TestCase):
+
+    def setUp(self):
+        self.cs = Collaborator()
+        self.cs.set_user('octocat')
+        self.cs.set_repo('oc_repo')
+
+    def test_LIST(self, request_method):
+        request_method.return_value = mock_response_result()
+        self.cs.list().all()
+        self.assertEqual(request_method.call_args[0],
+            ('get', _('repos/octocat/oc_repo/collaborators')))
+
+    def test_IS_colaborator(self, request_method):
+        request_method.return_value = mock_response()
+        self.cs.is_collaborator('user')
+        self.assertEqual(request_method.call_args[0],
+            ('head', _('repos/octocat/oc_repo/collaborators/user')))
+
+    def test_ADD(self, request_method):
+        self.cs.add('user')
+        self.assertEqual(request_method.call_args[0],
+            ('put', _('repos/octocat/oc_repo/collaborators/user')))
+
+    def test_DELETE(self, request_method):
+        request_method.return_value = mock_response('delete')
+        self.cs.delete('user')
+        self.assertEqual(request_method.call_args[0],
+            ('delete', _('repos/octocat/oc_repo/collaborators/user')))
