FilesExpand file tree

black

/

test_github_core.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

36 lines (28 loc) · 1.55 KB

black

/

test_github_core.py

Copy path

Top

File metadata and controls

• Code
• Blame

36 lines (28 loc) · 1.55 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

from .helper import IntegrationHelper

class TestGitHubCore(IntegrationHelper):

def test_ratelimit_remaining(self):

cassette_name = self.cassette_name("ratelimit_remaining")

with self.recorder.use_cassette(cassette_name):

assert self.gh.ratelimit_remaining > 0

def test_ratelimit_remaining_search(self):

"""Test if search iterators return search ratelimit"""

def _get_ratelimit(resource):

resources = self.gh.rate_limit().get("resources", {})

rate_limit = resources.get(resource, {})

return rate_limit.get("remaining", -1)

cassette_name = self.cassette_name("ratelimit_remaining_search")

# Run cassette to get correct remaining rate limit from responses.

with self.recorder.use_cassette(cassette_name):

correct_ratelimit_search = _get_ratelimit("search")

correct_ratelimit_core = _get_ratelimit("core")

# Re-run cassette to test functions under test.

with self.recorder.use_cassette(cassette_name):

result_iterator = self.gh.search_code(

"HTTPAdapter in:file language:python"

" repo:kennethreitz/requests"

)

ratelimit_remaining_search = result_iterator.ratelimit_remaining

ratelimit_remaining_core = self.gh.ratelimit_remaining

assert ratelimit_remaining_search != ratelimit_remaining_core

assert ratelimit_remaining_core == correct_ratelimit_core

assert ratelimit_remaining_search == correct_ratelimit_search