Allow adding repos to teams with specific permissions

Evan Borgstrom · Evan Borgstrom · commit c20489d65b74 · 2016-07-09T10:43:51.000-07:00
diff --git a/AUTHORS.rst b/AUTHORS.rst
@@ -130,3 +130,5 @@ Contributors
 - Dejan Svetec (@dsvetec)
 
 - Billy Keyes (@bluekeyes)
+
+- Evan Borgstrom (@borgstrom)
diff --git a/github3/orgs.py b/github3/orgs.py
@@ -78,14 +78,16 @@ def add_member(self, username):
         return self._boolean(self._put(url), 204, 404)
 
     @requires_auth
-    def add_repository(self, repository):
+    def add_repository(self, repository, permission=''):
         """Add ``repository`` to this team.
 
         :param str repository: (required), form: 'user/repo'
+        :param str permission: (optional), ('pull', 'push', 'admin')
         :returns: bool
         """
+        data = {'permission': permission}
         url = self._build_url('repos', repository, base_url=self._api)
-        return self._boolean(self._put(url), 204, 404)
+        return self._boolean(self._put(url, data=dumps(data)), 204, 404)
 
     @requires_auth
     def delete(self):
diff --git a/tests/unit/test_orgs_team.py b/tests/unit/test_orgs_team.py
@@ -24,7 +24,13 @@ def test_add_repository(self):
         """Show that one can add a repository to an organization team."""
         self.instance.add_repository('name-of-repo')
 
-        self.session.put.assert_called_once_with(url_for('repos/name-of-repo'))
+        self.put_called_with(url_for('repos/name-of-repo'),
+                             data={'permission': ''})
+
+        self.instance.add_repository('name-of-repo', permission='push')
+
+        self.put_called_with(url_for('repos/name-of-repo'),
+                             data={'permission': 'push'})
 
     def test_delete(self):
         """Show that a user can delete an organization team."""
