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)
