diff options
Diffstat (limited to 'pygithub3/tests/utils/base.py')
| -rw-r--r-- | pygithub3/tests/utils/base.py | 12 |
1 files changed, 10 insertions, 2 deletions
diff --git a/pygithub3/tests/utils/base.py b/pygithub3/tests/utils/base.py index 571d8ac..d9049c9 100644 --- a/pygithub3/tests/utils/base.py +++ b/pygithub3/tests/utils/base.py @@ -11,12 +11,20 @@ def mock_json(content): return content -def mock_response(status_code=200, content={}): +def mock_response(status_code='get', content={}): + CODES = dict(get=200, patch=200, post=201, delete=204) response = Mock(name='response') - response.status_code = status_code + response.status_code = CODES[str(status_code).lower()] or status_code response.content = content return response + +def mock_response_result(status_code='get'): + response = mock_response(status_code, content=[{}, {}]) + response.headers = {'link': ''} + return response + + class DummyResource(Resource): pass |