mirror of
https://github.com/status-im/PyGithub.git
synced 2026-08-31 19:01:15 +00:00
Steve Kowalik
37122cf2bb
Now that Python 2 support is gone, we can switch to no argument super() rather than calling our superclass by name.
80 lines
3.8 KiB
Python
80 lines
3.8 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
############################ Copyrights and license ############################
|
|
# #
|
|
# Copyright 2018 Justin Kufro <jkufro@andrew.cmu.edu> #
|
|
# Copyright 2018 Ivan Minno <iminno@andrew.cmu.edu> #
|
|
# Copyright 2018 Zilei Gu <zileig@andrew.cmu.edu> #
|
|
# Copyright 2018 Yves Zumbach <yzumbach@andrew.cmu.edu> #
|
|
# Copyright 2018 Leying Chen <leyingc@andrew.cmu.edu> #
|
|
# #
|
|
# 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 requests
|
|
import urllib3
|
|
from httpretty import httpretty
|
|
|
|
import github
|
|
|
|
from . import Framework
|
|
|
|
REPO_NAME = "PyGithub/PyGithub"
|
|
|
|
|
|
class Retry(Framework.TestCase):
|
|
def setUp(self):
|
|
# status codes returned on random github server errors
|
|
status_forcelist = (500, 502, 504)
|
|
retry = urllib3.Retry(
|
|
total=3, read=3, connect=3, status_forcelist=status_forcelist
|
|
)
|
|
|
|
Framework.enableRetry(retry)
|
|
super().setUp()
|
|
|
|
def testShouldNotRetryWhenStatusNotOnList(self):
|
|
with self.assertRaises(github.GithubException):
|
|
self.g.get_repo(REPO_NAME)
|
|
self.assertEqual(len(httpretty.latest_requests), 1)
|
|
|
|
def testReturnsRepoAfter3Retries(self):
|
|
repository = self.g.get_repo(REPO_NAME)
|
|
self.assertEqual(len(httpretty.latest_requests), 4)
|
|
for request in httpretty.latest_requests:
|
|
self.assertEqual(request.path, "/repos/" + REPO_NAME)
|
|
|
|
self.assertIsInstance(repository, github.Repository.Repository)
|
|
self.assertEqual(repository.full_name, REPO_NAME)
|
|
|
|
def testReturnsRepoAfter1Retry(self):
|
|
repository = self.g.get_repo(REPO_NAME)
|
|
self.assertEqual(len(httpretty.latest_requests), 2)
|
|
for request in httpretty.latest_requests:
|
|
self.assertEqual(request.path, "/repos/" + REPO_NAME)
|
|
|
|
self.assertIsInstance(repository, github.Repository.Repository)
|
|
self.assertEqual(repository.full_name, REPO_NAME)
|
|
|
|
def testRaisesRetryErrorAfterMaxRetries(self):
|
|
with self.assertRaises(requests.exceptions.RetryError):
|
|
self.g.get_repo("PyGithub/PyGithub")
|
|
self.assertEqual(len(httpretty.latest_requests), 4)
|
|
for request in httpretty.latest_requests:
|
|
self.assertEqual(request.path, "/repos/PyGithub/PyGithub")
|