mirror of
https://github.com/status-im/PyGithub.git
synced 2026-08-31 19:01:15 +00:00
TechnicalPirate
78d283b9b5
# Problem # The `.whl` available on PyPi for this repository contains its entire `tests` submodule. The total size of the `pygithub` module is less than `1mb` without tests and `>23mb` with. Previous discussions on this topic indicated the removal of tests would be accepted if someone created a pull request to do so: https://github.com/PyGithub/PyGithub/issues/672 This is that pull request! 😄 # Solution/Implementation # - Moved "tests" to root of the repository ( outside the package ) - Added `tests.` module path to the imports in `AllTests.py` - Updated `setup.py` - Removed `package_data` from product - Removed "tests" from packages definition - Repointed `test_suite` to new location - Updated `travis.yml` - Added conditional ( if not `py2.7` ) conversion of code to `py3` _... apologies for all the failed build spam below - new to working with Travis CI / had some gotcha moments!_
87 lines
3.9 KiB
Python
87 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
|
|
|
|
from github import GithubException
|
|
from github.Repository import Repository
|
|
|
|
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):
|
|
try:
|
|
self.g.get_repo(REPO_NAME)
|
|
except GithubException:
|
|
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, 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, 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')
|