Changeset View
Changeset View
Standalone View
Standalone View
swh/lister/gitlab/tests/test_gitlab_lister.py
| # Copyright (C) 2017-2018 the Software Heritage developers | # Copyright (C) 2017-2018 the Software Heritage developers | ||||
| # License: GNU General Public License version 3, or any later version | # License: GNU General Public License version 3, or any later version | ||||
| # See top-level LICENSE file for more information | # See top-level LICENSE file for more information | ||||
| import re | |||||
| import unittest | import unittest | ||||
| from datetime import datetime, timedelta | from datetime import datetime, timedelta | ||||
| from swh.lister.gitlab.lister import GitLabLister | from swh.lister.gitlab.lister import GitLabLister | ||||
| from swh.lister.core.tests.test_lister import HttpListerTesterBase | from swh.lister.core.tests.test_lister import HttpListerTesterBase | ||||
| class GitLabListerTester(HttpListerTesterBase, unittest.TestCase): | class GitLabListerTester(HttpListerTesterBase, unittest.TestCase): | ||||
| Lister = GitLabLister | Lister = GitLabLister | ||||
| test_re = GitLabLister.API_URL_INDEX_RE | test_re = re.compile(r'^.*/projects.*page=(\d+).*') | ||||
| lister_subdir = 'gitlab' | lister_subdir = 'gitlab' | ||||
| good_api_response_file = 'api_response.json' | good_api_response_file = 'api_response.json' | ||||
| bad_api_response_file = 'api_empty_response.json' | bad_api_response_file = 'api_empty_response.json' | ||||
| first_index = 1 | first_index = 1 | ||||
| last_index = 2 | |||||
| entries_per_page = 10 | entries_per_page = 10 | ||||
| def response_headers(self, request): | def response_headers(self, request): | ||||
| headers = {'RateLimit-Remaining': '1'} | headers = {'RateLimit-Remaining': '1'} | ||||
| if self.request_index(request) == str(self.first_index): | if self.request_index(request) == str(self.first_index): | ||||
| headers.update({ | headers.update({ | ||||
| 'Link': '<https://gitlab.com/v4/projects?page=2>;' | 'Link': '<https://gitlab.com/v4/projects?page=2>;' | ||||
| ' rel="next",' | ' rel="next",' | ||||
| Show All 18 Lines | |||||