Changeset View
Changeset View
Standalone View
Standalone View
swh/lister/phabricator/tests/test_lister.py
- This file was added.
| # Copyright (C) 2019 the Software Heritage developers | |||||
| # License: GNU General Public License version 3, or any later version | |||||
| # See top-level LICENSE file for more information | |||||
| import re | |||||
| import json | |||||
| import unittest | |||||
| from swh.lister.core.tests.test_lister import HttpListerTester | |||||
| from swh.lister.phabricator.lister import PhabricatorLister | |||||
| from swh.lister.phabricator.lister import get_repo_url | |||||
| class PhabricatorListerTester(HttpListerTester, unittest.TestCase): | |||||
| Lister = PhabricatorLister | |||||
| test_re = re.compile(r'\&after=([^?&]+)') | |||||
| lister_subdir = 'phabricator' | |||||
| good_api_response_file = 'api_response.json' | |||||
| good_api_response_undefined_protocol = 'api_response_undefined_'\ | |||||
| 'protocol.json' | |||||
| bad_api_response_file = 'api_empty_response.json' | |||||
| first_index = 1 | |||||
| last_index = 12 | |||||
| entries_per_page = 10 | |||||
| def get_fl(self, override_config=None): | |||||
| """(Override) Retrieve an instance of fake lister (fl). | |||||
| """ | |||||
| if override_config or self.fl is None: | |||||
| self.fl = self.Lister(forge_url='https://fakeurl', api_token='a-1', | |||||
| override_config=override_config) | |||||
| self.fl.INITIAL_BACKOFF = 1 | |||||
| self.fl.reset_backoff() | |||||
| return self.fl | |||||
| def test_get_repo_url(self): | |||||
| f = open('swh/lister/%s/tests/%s' % (self.lister_subdir, | |||||
| self.good_api_response_file)) | |||||
| api_response = json.load(f) | |||||
| repos = api_response['result']['data'] | |||||
| for repo in repos: | |||||
| self.assertEqual( | |||||
| 'https://forge.softwareheritage.org/source/%s.git' % | |||||
| (repo['fields']['shortName']), | |||||
| get_repo_url(repo['attachments']['uris']['uris'])) | |||||
| f = open('swh/lister/%s/tests/%s' % | |||||
| (self.lister_subdir, | |||||
| self.good_api_response_undefined_protocol)) | |||||
| repo = json.load(f) | |||||
| self.assertEqual( | |||||
| 'https://svn.blender.org/svnroot/bf-blender/', | |||||
| get_repo_url(repo['attachments']['uris']['uris'])) | |||||