mirror of
https://github.com/status-im/PyGithub.git
synced 2026-09-01 19:31:10 +00:00
Steve Kowalik
befce37e20
The Retry test class was importing Repository directly, which had the side effect of masking the tests.Repository import in AllTests with github.Repository. Switch to importing github only, which is safe. As a drive-by, switch to using self.assertRaises for one of the tests.
85 lines
3.9 KiB
Python
85 lines
3.9 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 urllib3
|
|
from httpretty import httpretty
|
|
|
|
import Framework
|
|
|
|
import requests
|
|
|
|
import github
|
|
|
|
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)
|
|
Framework.TestCase.setUp(self)
|
|
|
|
def testShouldNotRetryWhenStatusNotOnList(self):
|
|
with self.assertRaises(github.GithubException):
|
|
self.g.get_repo(REPO_NAME)
|
|
self.assertEquals(len(httpretty.latest_requests), 1)
|
|
|
|
def testReturnsRepoAfter3Retries(self):
|
|
repository = self.g.get_repo(REPO_NAME)
|
|
self.assertEquals(len(httpretty.latest_requests), 4)
|
|
for request in httpretty.latest_requests:
|
|
self.assertEquals(request.path, '/repos/' + REPO_NAME)
|
|
|
|
self.assertIsInstance(repository, github.Repository.Repository)
|
|
self.assertEquals(repository.full_name, REPO_NAME)
|
|
|
|
def testReturnsRepoAfter1Retry(self):
|
|
repository = self.g.get_repo(REPO_NAME)
|
|
self.assertEquals(len(httpretty.latest_requests), 2)
|
|
for request in httpretty.latest_requests:
|
|
self.assertEquals(request.path, '/repos/' + REPO_NAME)
|
|
|
|
self.assertIsInstance(repository, github.Repository.Repository)
|
|
self.assertEquals(repository.full_name, REPO_NAME)
|
|
|
|
def testRaisesRetryErrorAfterMaxRetries(self):
|
|
try:
|
|
response = self.g.get_repo('PyGithub/PyGithub')
|
|
self.fail("RetryError should have been raised")
|
|
except requests.exceptions.RetryError:
|
|
self.assertEquals(len(httpretty.latest_requests), 4)
|
|
for request in httpretty.latest_requests:
|
|
self.assertEquals(request.path, '/repos/PyGithub/PyGithub')
|