Added method to add a new team.

JNRowe · JNRowe · commit 05f4e79c8a48 · 2011-12-20T05:13:49.000Z
Thanks to Clint Savage, refs ask#69.
diff --git a/github2/organizations.py b/github2/organizations.py
@@ -1,4 +1,5 @@
-from github2.core import BaseData, GithubCommand, Attribute, DateAttribute
+from github2.core import (BaseData, GithubCommand, Attribute, DateAttribute,
+                          requires_auth)
 from github2.repositories import Repository
 from github2.teams import Team
 from github2.users import User
@@ -81,3 +82,18 @@ def teams(self, organization):
         """
         return self.get_values(organization, 'teams', filter="teams",
                                datatype=Team)
+
+    @requires_auth
+    def add_team(self, organization, name, permission='pull', projects=None):
+        """Add a team to an organization
+
+        :param str organization: organization to add team to
+        :param str team: name of team to add
+        :param str permission: permissions for team(push, pull or admin)
+        :param list projects: optional GitHub projects for this team
+        """
+        team_data = {'team[name]': name, 'team[permission]': permission}
+        if projects:
+            team_data['team[repo_names][]'] = projects
+        return self.get_value(organization, 'teams', post_data=team_data,
+                              method='POST', filter='team', datatype=Team)
diff --git a/tests/data/github.com,api,v2,json,organizations,JNRowe-test-org,teams,6869b96ae6fbaa7511d92aa73803898b b/tests/data/github.com,api,v2,json,organizations,JNRowe-test-org,teams,6869b96ae6fbaa7511d92aa73803898b
@@ -0,0 +1,14 @@
+status: 200
+x-ratelimit-remaining: 60
+content-location: https://github.com/api/v2/json/organizations/JNRowe-test-org/teams
+x-runtime: 20ms
+content-length: 61
+server: nginx/1.0.4
+connection: keep-alive
+x-ratelimit-limit: 58
+etag: "0ea8a446a00b4d7894676a34bd8d501f"
+cache-control: private, max-age=0, must-revalidate
+date: Mon, 19 Dec 2011 20:50:41 GMT
+content-type: application/json; charset=utf-8
+
+{"team":{"name":"team_pull","id":121678,"permission":"pull"}}
diff --git a/tests/data/github.com,api,v2,json,teams,121678,repositories,access_token=xxx,00192d3a6cc4fc297b5a9cb0742c3663 b/tests/data/github.com,api,v2,json,teams,121678,repositories,access_token=xxx,00192d3a6cc4fc297b5a9cb0742c3663
@@ -0,0 +1,14 @@
+status: 200
+x-ratelimit-remaining: 60
+content-location: https://github.com/api/v2/json/organizations/JNRowe-test-org/teams
+x-runtime: 20ms
+content-length: 676
+server: nginx/1.0.4
+connection: keep-alive
+x-ratelimit-limit: 58
+etag: "397a2fcea970a6391b74eabcb62ed265"
+cache-control: private, max-age=0, must-revalidate
+date: Thu, 19 Dec 2011 20:50:41 GMT
+content-type: application/json; charset=utf-8
+
+{"repositories":[{"open_issues":0,"description":"","watchers":1,"forks":1,"has_issues":true,"has_downloads":true,"fork":false,"created_at":"2011/12/18 17:36:08 -0800","homepage":"","size":0,"private":false,"name":"test1","owner":"JNRowe-test-org","has_wiki":true,"url":"https://github.com/JNRowe-test-org/test1","organization":"JNRowe-test-org"},{"open_issues":0,"description":"","watchers":1,"forks":1,"has_issues":true,"has_downloads":true,"fork":false,"created_at":"2011/12/18 17:36:24 -0800","homepage":"","size":0,"private":false,"name":"test2","owner":"JNRowe-test-org","has_wiki":true,"url":"https://github.com/JNRowe-test-org/test2","organization":"JNRowe-test-org"}]}
diff --git a/tests/test_organizations.py b/tests/test_organizations.py
@@ -29,3 +29,20 @@ def test_public_members(self):
         members = self.client.organizations.public_members('github')
         assert_equals(len(members), 35)
         assert_equals(members[2].name, 'Ben Burkert')
+
+
+class OrganizationsEdits(utils.HttpMockAuthenticatedTestCase):
+    def test_add_team(self):
+        team = self.client.organizations.add_team('JNRowe-test-org',
+                                                  'test_pull', 'pull')
+        assert_equals(team.name, 'team_pull')
+        assert_equals(team.permission, 'pull')
+
+    def test_add_team_with_repos(self):
+        projects = ['JNRowe-test-org/test1', 'JNRowe-test-org/test2']
+        team = self.client.organizations.add_team('JNRowe-test-org',
+                                                  'test_push', 'push', projects)
+
+        team_repos = self.client.teams.repositories(team.id)
+        assert_equals(['/'.join([x.organization, x.name]) for x in team_repos],
+                      projects)
