mirror of
https://github.com/status-im/PyGithub.git
synced 2026-08-31 19:01:15 +00:00
Steve Kowalik
e113e37de1
To help us switch to f-strings and other 3.6+ changes, add pyupgrade to our pre-commit configuration to keep the codebase clean.
310 lines
14 KiB
Python
310 lines
14 KiB
Python
############################ 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 2015 Kyle Hornberg <khornberg@users.noreply.github.com> #
|
|
# Copyright 2016 Jannis Gebauer <ja.geb@me.com> #
|
|
# Copyright 2016 Peter Buckley <dx-pbuckley@users.noreply.github.com> #
|
|
# Copyright 2018 Steve Kowalik <steven@wedontsleep.org> #
|
|
# 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/>. #
|
|
# #
|
|
################################################################################
|
|
|
|
import github
|
|
|
|
from . import Framework
|
|
|
|
|
|
class Branch(Framework.TestCase):
|
|
def setUp(self):
|
|
super().setUp()
|
|
self.repo = self.g.get_user().get_repo("PyGithub")
|
|
self.branch = self.repo.get_branch("topic/RewriteWithGeneratedCode")
|
|
self.protected_branch = self.repo.get_branch("integrations")
|
|
self.organization_branch = self.g.get_repo(
|
|
"PyGithub/PyGithub", lazy=True
|
|
).get_branch("master")
|
|
|
|
def testAttributes(self):
|
|
self.assertEqual(self.branch.name, "topic/RewriteWithGeneratedCode")
|
|
self.assertEqual(
|
|
self.branch.commit.sha, "1292bf0e22c796e91cc3d6e24b544aece8c21f2a"
|
|
)
|
|
self.assertEqual(
|
|
self.branch.protection_url,
|
|
"https://api.github.com/repos/jacquev6/PyGithub/branches/topic/RewriteWithGeneratedCode/protection",
|
|
)
|
|
self.assertFalse(self.branch.protected)
|
|
self.assertEqual(
|
|
repr(self.branch), 'Branch(name="topic/RewriteWithGeneratedCode")'
|
|
)
|
|
|
|
def testEditProtection(self):
|
|
self.protected_branch.edit_protection(
|
|
strict=True,
|
|
require_code_owner_reviews=True,
|
|
required_approving_review_count=2,
|
|
)
|
|
branch_protection = self.protected_branch.get_protection()
|
|
self.assertTrue(branch_protection.required_status_checks.strict)
|
|
self.assertEqual(branch_protection.required_status_checks.contexts, [])
|
|
self.assertTrue(branch_protection.enforce_admins)
|
|
self.assertFalse(
|
|
branch_protection.required_pull_request_reviews.dismiss_stale_reviews
|
|
)
|
|
self.assertTrue(
|
|
branch_protection.required_pull_request_reviews.require_code_owner_reviews
|
|
)
|
|
self.assertEqual(
|
|
branch_protection.required_pull_request_reviews.required_approving_review_count,
|
|
2,
|
|
)
|
|
|
|
def testEditProtectionDismissalUsersWithUserOwnedBranch(self):
|
|
with self.assertRaises(github.GithubException) as raisedexp:
|
|
self.protected_branch.edit_protection(dismissal_users=["jacquev6"])
|
|
self.assertEqual(raisedexp.exception.status, 422)
|
|
self.assertEqual(
|
|
raisedexp.exception.data,
|
|
{
|
|
"documentation_url": "https://developer.github.com/v3/repos/branches/#update-branch-protection",
|
|
"message": "Validation Failed",
|
|
"errors": [
|
|
"Only organization repositories can have users and team restrictions"
|
|
],
|
|
},
|
|
)
|
|
|
|
def testEditProtectionPushRestrictionsWithUserOwnedBranch(self):
|
|
with self.assertRaises(github.GithubException) as raisedexp:
|
|
self.protected_branch.edit_protection(
|
|
user_push_restrictions=["jacquev6"], team_push_restrictions=[]
|
|
)
|
|
self.assertEqual(raisedexp.exception.status, 422)
|
|
self.assertEqual(
|
|
raisedexp.exception.data,
|
|
{
|
|
"documentation_url": "https://developer.github.com/v3/repos/branches/#update-branch-protection",
|
|
"message": "Validation Failed",
|
|
"errors": [
|
|
"Only organization repositories can have users and team restrictions"
|
|
],
|
|
},
|
|
)
|
|
|
|
def testEditProtectionPushRestrictionsAndDismissalUser(self):
|
|
self.organization_branch.edit_protection(
|
|
dismissal_users=["jacquev6"], user_push_restrictions=["jacquev6"]
|
|
)
|
|
branch_protection = self.organization_branch.get_protection()
|
|
self.assertListKeyEqual(
|
|
branch_protection.required_pull_request_reviews.dismissal_users,
|
|
lambda u: u.login,
|
|
["jacquev6"],
|
|
)
|
|
self.assertListKeyEqual(
|
|
branch_protection.required_pull_request_reviews.dismissal_teams,
|
|
lambda u: u.slug,
|
|
[],
|
|
)
|
|
self.assertListKeyEqual(
|
|
branch_protection.get_user_push_restrictions(),
|
|
lambda u: u.login,
|
|
["jacquev6"],
|
|
)
|
|
self.assertListKeyEqual(
|
|
branch_protection.get_team_push_restrictions(), lambda u: u.slug, []
|
|
)
|
|
|
|
def testRemoveProtection(self):
|
|
self.assertTrue(self.protected_branch.protected)
|
|
self.protected_branch.remove_protection()
|
|
protected_branch = self.repo.get_branch("integrations")
|
|
self.assertFalse(protected_branch.protected)
|
|
with self.assertRaises(github.GithubException) as raisedexp:
|
|
protected_branch.get_protection()
|
|
self.assertEqual(raisedexp.exception.status, 404)
|
|
self.assertEqual(
|
|
raisedexp.exception.data,
|
|
{
|
|
"documentation_url": "https://developer.github.com/v3/repos/branches/#get-branch-protection",
|
|
"message": "Branch not protected",
|
|
},
|
|
)
|
|
|
|
def testEditRequiredStatusChecks(self):
|
|
self.protected_branch.edit_required_status_checks(strict=True)
|
|
required_status_checks = self.protected_branch.get_required_status_checks()
|
|
self.assertTrue(required_status_checks.strict)
|
|
self.assertEqual(required_status_checks.contexts, ["foo/bar"])
|
|
|
|
def testRemoveRequiredStatusChecks(self):
|
|
self.protected_branch.remove_required_status_checks()
|
|
with self.assertRaises(github.GithubException) as raisedexp:
|
|
self.protected_branch.get_required_status_checks()
|
|
self.assertEqual(raisedexp.exception.status, 404)
|
|
self.assertEqual(
|
|
raisedexp.exception.data,
|
|
{
|
|
"documentation_url": "https://developer.github.com/v3/repos/branches/#get-required-status-checks-of-protected-branch",
|
|
"message": "Required status checks not enabled",
|
|
},
|
|
)
|
|
|
|
def testEditRequiredPullRequestReviews(self):
|
|
self.protected_branch.edit_required_pull_request_reviews(
|
|
dismiss_stale_reviews=True, required_approving_review_count=2
|
|
)
|
|
required_pull_request_reviews = (
|
|
self.protected_branch.get_required_pull_request_reviews()
|
|
)
|
|
self.assertTrue(required_pull_request_reviews.dismiss_stale_reviews)
|
|
self.assertTrue(required_pull_request_reviews.require_code_owner_reviews)
|
|
self.assertEqual(
|
|
required_pull_request_reviews.required_approving_review_count, 2
|
|
)
|
|
|
|
def testEditRequiredPullRequestReviewsWithTooLargeApprovingReviewCount(self):
|
|
with self.assertRaises(github.GithubException) as raisedexp:
|
|
self.protected_branch.edit_required_pull_request_reviews(
|
|
required_approving_review_count=9
|
|
)
|
|
self.assertEqual(raisedexp.exception.status, 422)
|
|
self.assertEqual(
|
|
raisedexp.exception.data,
|
|
{
|
|
"documentation_url": "https://developer.github.com/v3/repos/branches/#update-pull-request-review-enforcement-of-protected-branch",
|
|
"message": "Invalid request.\n\n9 must be less than or equal to 6.",
|
|
},
|
|
)
|
|
|
|
def testEditRequiredPullRequestReviewsWithUserBranchAndDismissalUsers(self):
|
|
with self.assertRaises(github.GithubException) as raisedexp:
|
|
self.protected_branch.edit_required_pull_request_reviews(
|
|
dismissal_users=["jacquev6"]
|
|
)
|
|
self.assertEqual(raisedexp.exception.status, 422)
|
|
self.assertEqual(
|
|
raisedexp.exception.data,
|
|
{
|
|
"documentation_url": "https://developer.github.com/v3/repos/branches/#update-pull-request-review-enforcement-of-protected-branch",
|
|
"message": "Dismissal restrictions are supported only for repositories owned by an organization.",
|
|
},
|
|
)
|
|
|
|
def testRemoveRequiredPullRequestReviews(self):
|
|
self.protected_branch.remove_required_pull_request_reviews()
|
|
required_pull_request_reviews = (
|
|
self.protected_branch.get_required_pull_request_reviews()
|
|
)
|
|
self.assertFalse(required_pull_request_reviews.dismiss_stale_reviews)
|
|
self.assertFalse(required_pull_request_reviews.require_code_owner_reviews)
|
|
self.assertEqual(
|
|
required_pull_request_reviews.required_approving_review_count, 1
|
|
)
|
|
|
|
def testAdminEnforcement(self):
|
|
self.protected_branch.remove_admin_enforcement()
|
|
self.assertFalse(self.protected_branch.get_admin_enforcement())
|
|
self.protected_branch.set_admin_enforcement()
|
|
self.assertTrue(self.protected_branch.get_admin_enforcement())
|
|
|
|
def testAddUserPushRestrictions(self):
|
|
self.organization_branch.add_user_push_restrictions("sfdye")
|
|
self.assertListKeyEqual(
|
|
self.organization_branch.get_user_push_restrictions(),
|
|
lambda u: u.login,
|
|
["jacquev6", "sfdye"],
|
|
)
|
|
|
|
def testReplaceUserPushRestrictions(self):
|
|
self.assertListKeyEqual(
|
|
self.organization_branch.get_user_push_restrictions(),
|
|
lambda u: u.login,
|
|
["jacquev6"],
|
|
)
|
|
self.organization_branch.replace_user_push_restrictions("sfdye")
|
|
self.assertListKeyEqual(
|
|
self.organization_branch.get_user_push_restrictions(),
|
|
lambda u: u.login,
|
|
["sfdye"],
|
|
)
|
|
|
|
def testRemoveUserPushRestrictions(self):
|
|
self.organization_branch.remove_user_push_restrictions("jacquev6")
|
|
self.assertListKeyEqual(
|
|
self.organization_branch.get_user_push_restrictions(),
|
|
lambda u: u.login,
|
|
["sfdye"],
|
|
)
|
|
|
|
def testAddTeamPushRestrictions(self):
|
|
self.organization_branch.add_team_push_restrictions("pygithub-owners")
|
|
self.assertListKeyEqual(
|
|
self.organization_branch.get_team_push_restrictions(),
|
|
lambda t: t.slug,
|
|
["pygithub-owners"],
|
|
)
|
|
|
|
def testReplaceTeamPushRestrictions(self):
|
|
self.assertListKeyEqual(
|
|
self.organization_branch.get_team_push_restrictions(),
|
|
lambda t: t.slug,
|
|
["pygithub-owners"],
|
|
)
|
|
self.organization_branch.replace_team_push_restrictions("org-team")
|
|
self.assertListKeyEqual(
|
|
self.organization_branch.get_team_push_restrictions(),
|
|
lambda t: t.slug,
|
|
["org-team"],
|
|
)
|
|
|
|
def testRemoveTeamPushRestrictions(self):
|
|
self.organization_branch.remove_team_push_restrictions("org-team")
|
|
self.assertListKeyEqual(
|
|
self.organization_branch.get_team_push_restrictions(),
|
|
lambda t: t.slug,
|
|
["pygithub-owners"],
|
|
)
|
|
|
|
def testRemovePushRestrictions(self):
|
|
self.organization_branch.remove_push_restrictions()
|
|
with self.assertRaises(github.GithubException) as raisedexp:
|
|
list(self.organization_branch.get_user_push_restrictions())
|
|
self.assertEqual(raisedexp.exception.status, 404)
|
|
self.assertEqual(
|
|
raisedexp.exception.data,
|
|
{
|
|
"documentation_url": "https://developer.github.com/v3/repos/branches/#list-team-restrictions-of-protected-branch",
|
|
"message": "Push restrictions not enabled",
|
|
},
|
|
)
|
|
|
|
def testGetRequiredSignatures(self):
|
|
required_signature = self.protected_branch.get_required_signatures()
|
|
assert required_signature
|
|
|
|
def testRemoveRequiredSignatures(self):
|
|
self.protected_branch.remove_required_signatures()
|
|
|
|
def testAddRequiredSignatures(self):
|
|
self.protected_branch.add_required_signatures()
|