# -*- 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):
        with self.assertRaises(requests.exceptions.RetryError):
            self.g.get_repo('PyGithub/PyGithub')
        self.assertEquals(len(httpretty.latest_requests), 4)
        for request in httpretty.latest_requests:
            self.assertEquals(request.path, '/repos/PyGithub/PyGithub')