Files
PyGithub/tests/Team.py
T
Steve Kowalik 37122cf2bb Switch to no argument super() (#1361)
Now that Python 2 support is gone, we can switch to no argument super()
rather than calling our superclass by name.
2020-01-21 21:15:22 +11:00

155 lines
7.3 KiB
Python

# -*- coding: utf-8 -*-
############################ Copyrights and license ############################
# #
# Copyright 2012 Vincent Jacques <vincent@vincent-jacques.net> #
# Copyright 2012 Zearin <zearin@gonk.net> #
# Copyright 2013 Vincent Jacques <vincent@vincent-jacques.net> #
# Copyright 2014 Vincent Jacques <vincent@vincent-jacques.net> #
# Copyright 2016 Jannis Gebauer <ja.geb@me.com> #
# Copyright 2016 Peter Buckley <dx-pbuckley@users.noreply.github.com> #
# Copyright 2016 mattjmorrison <mattjmorrison@mattjmorrison.com> #
# Copyright 2018 Isuru Fernando <isuruf@gmail.com> #
# Copyright 2018 Jacopo Notarstefano <jacopo.notarstefano@gmail.com> #
# Copyright 2018 James D'Amato <james.j.damato@gmail.com> #
# Copyright 2018 Steve Kowalik <steven@wedontsleep.org> #
# Copyright 2018 Tim Boring <tboring@hearst.com> #
# Copyright 2018 sfdye <tsfdye@gmail.com> #
# #
# This file is part of PyGithub. #
# http://pygithub.readthedocs.io/ #
# #
# PyGithub is free software: you can redistribute it and/or modify it under #
# the terms of the GNU Lesser General Public License as published by the Free #
# Software Foundation, either version 3 of the License, or (at your option) #
# any later version. #
# #
# PyGithub is distributed in the hope that it will be useful, but WITHOUT ANY #
# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS #
# FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more #
# details. #
# #
# You should have received a copy of the GNU Lesser General Public License #
# along with PyGithub. If not, see <http://www.gnu.org/licenses/>. #
# #
################################################################################
from datetime import datetime
from . import Framework
class Team(Framework.TestCase):
def setUp(self):
super().setUp()
self.org = self.g.get_organization("BeaverSoftware")
self.team = self.org.get_team(189850)
def testAttributes(self):
self.assertEqual(self.team.id, 189850)
self.assertEqual(self.team.members_count, 0)
self.assertEqual(self.team.name, "Team created by PyGithub")
self.assertEqual(self.team.permission, "pull")
self.assertEqual(self.team.repos_count, 0)
self.assertEqual(self.team.url, "https://api.github.com/teams/189850")
self.assertEqual(self.team.organization, self.org)
self.assertEqual(self.team.privacy, "closed")
self.assertEqual(self.team.parent, None)
# test __repr__() based on this attributes
self.assertEqual(
self.team.__repr__(), 'Team(name="Team created by PyGithub", id=189850)'
)
def testDiscussions(self):
discussions = list(self.team.get_discussions())
self.assertEqual(len(discussions), 1)
d = discussions[0]
self.assertEqual(d.author.login, "jacquev6")
self.assertEqual(d.body, "BODY")
self.assertEqual(d.body_html, "<p>BODY</p>")
self.assertEqual(d.body_version, "bedf0740b01d2d758cff9873c2387817")
self.assertEqual(d.comments_count, 0)
self.assertEqual(
d.comments_url, "https://api.github.com/teams/189850/discussions/1/comments"
)
self.assertEqual(d.created_at, datetime(2019, 10, 8, 21, 3, 36))
self.assertEqual(
d.html_url,
"https://github.com/orgs/BeaverSoftware/teams/Team/discussions/1",
)
self.assertEqual(d.last_edited_at, None)
self.assertEqual(d.node_id, "MDE0OlRlYW1EaXNjdXNzaW9uMzA=")
self.assertEqual(d.number, 1)
self.assertEqual(d.pinned, True)
self.assertEqual(d.private, False)
self.assertEqual(d.team_url, "https://api.github.com/teams/189850")
self.assertEqual(d.title, "TITLE")
self.assertEqual(d.updated_at, datetime(2019, 10, 8, 21, 3, 36))
self.assertEqual(d.url, "https://api.github.com/teams/189850/discussions/1")
def testMembers(self):
user = self.g.get_user("jacquev6")
self.assertListKeyEqual(self.team.get_members(), None, [])
self.assertFalse(self.team.has_in_members(user))
self.team.add_to_members(user)
self.assertListKeyEqual(
self.team.get_members(), lambda u: u.login, ["jacquev6"]
)
self.assertTrue(self.team.has_in_members(user))
self.team.remove_from_members(user)
self.assertListKeyEqual(self.team.get_members(), None, [])
self.assertFalse(self.team.has_in_members(user))
self.team.add_membership(user, "maintainer")
self.assertRaises(AssertionError, self.team.add_membership, user, "admin")
self.team.remove_membership(user)
def testRepoPermission(self):
repo = self.org.get_repo("FatherBeaver")
self.team.set_repo_permission(repo, "admin")
def testRepos(self):
repo = self.org.get_repo("FatherBeaver")
self.assertListKeyEqual(self.team.get_repos(), None, [])
self.assertFalse(self.team.has_in_repos(repo))
self.team.add_to_repos(repo)
self.assertListKeyEqual(
self.team.get_repos(), lambda r: r.name, ["FatherBeaver"]
)
self.assertTrue(self.team.has_in_repos(repo))
self.team.remove_from_repos(repo)
self.assertListKeyEqual(self.team.get_repos(), None, [])
self.assertFalse(self.team.has_in_repos(repo))
def testEditWithoutArguments(self):
self.team.edit("Name edited by PyGithub")
self.assertEqual(self.team.name, "Name edited by PyGithub")
def testEditWithAllArguments(self):
self.team.edit(
"Name edited twice by PyGithub",
"Description edited by PyGithub",
"admin",
"secret",
)
self.assertEqual(self.team.name, "Name edited twice by PyGithub")
self.assertEqual(self.team.description, "Description edited by PyGithub")
self.assertEqual(self.team.permission, "admin")
self.assertEqual(self.team.privacy, "secret")
def testGetTeams(self):
nested_teams = self.team.get_teams()
self.assertListKeyEqual(
nested_teams,
lambda t: t.name,
['DummyTeam1', 'DummyTeam2', 'DummyTeam3']
)
parent = nested_teams[0].parent
self.assertEqual(self.team.name, parent.name)
self.assertEqual(self.team.id, parent.id)
def testDelete(self):
self.team.delete()