def __init__(self, py_version=None, checker=None, store=None): if py_version is None: py_version = '3' if checker is None: checker = compatibility_checker.CompatibilityChecker() self.py_version = py_version self._checker = checker self._store = store self._dependency_info_getter = utils.DependencyInfo( py_version, self._checker, self._store)
def __init__(self, py_version=None, checker=None, store=None, max_workers=50): if py_version is None: py_version = '3' if checker is None: checker = compatibility_checker.CompatibilityChecker() self.max_workers = max_workers self.py_version = py_version self._checker = checker self._store = store self._dependency_info_getter = utils.DependencyInfo( py_version, self._checker, self._store) # Share a common pool for PyPI requests to avoid creating too many # threads. self._pypi_thread_pool = concurrent.futures.ThreadPoolExecutor( max_workers=self.max_workers)
def test__get_from_endpoint(self): dep_info_getter = utils.DependencyInfo(checker=self.mock_checker, store=self.fake_store) dep_info = dep_info_getter._get_from_endpoint('opencensus') self.assertEqual(dep_info, DEP_INFO)
def test__get_from_bigquery_not_exists(self): dep_info_getter = utils.DependencyInfo(checker=self.mock_checker, store=self.fake_store) dep_info = dep_info_getter._get_from_bigquery('pkg_not_in_config') self.assertIsNone(dep_info)
def test__get_from_bigquery_exists(self): dep_info_getter = utils.DependencyInfo(checker=self.mock_checker, store=self.fake_store) dep_info = dep_info_getter._get_from_bigquery('opencensus') self.assertIsNotNone(dep_info)
def test_constructor_explicit(self): dep_info_getter = utils.DependencyInfo(py_version='2', checker=self.mock_checker, store=self.fake_store) self.assertEqual(dep_info_getter.py_version, '2')
def test_constructor_default(self): dep_info_getter = utils.DependencyInfo(checker=self.mock_checker, store=self.fake_store) self.assertEqual(dep_info_getter.py_version, '3')
def test__get_from_endpoint_raise_exception(self): dep_info_getter = utils.DependencyInfo(checker=self.mock_checker, store=self.fake_store) with self.assertRaises(utils.PackageNotSupportedError): dep_info_getter._get_from_endpoint('pkg_not_in_config')
def test_get_dependency_info_compatibility_store(self): dep_info_getter = utils.DependencyInfo(checker=self.mock_checker, store=self.fake_store) dep_info = dep_info_getter.get_dependency_info('opencensus') self.assertIsNotNone(dep_info)
def test_get_dependency_info_endpoint(self): dep_info_getter = utils.DependencyInfo( checker=self.mock_checker, store=self.fake_store) dep_info = dep_info_getter.get_dependency_info('package1') self.assertEqual(dep_info, self.DEP_INFO)