def test_resolve_not_found(self):
os_path_exists_mock = MagicMock(side_effect=[True, False])
cache_path_mock = MagicMock(return_value='/bundle-cached-path')
urlretrieve_mock = MagicMock(side_effect=URLError('no_such.bundle'))
get_url_mock = MagicMock(return_value=('bundle-name', '/bundle-url-resolved'))
get_logger_mock, log_mock = create_mock_logger()
with patch('os.path.exists', os_path_exists_mock), \
patch('conductr_cli.resolvers.uri_resolver.cache_path', cache_path_mock), \
patch('conductr_cli.resolvers.uri_resolver.get_url', get_url_mock), \
patch('conductr_cli.resolvers.uri_resolver.urlretrieve', urlretrieve_mock), \
patch('logging.getLogger', get_logger_mock):
is_resolved, bundle_name, bundle_file = uri_resolver.resolve_bundle('/cache-dir', '/bundle-url')
self.assertFalse(is_resolved)
self.assertIsNone(bundle_name)
self.assertIsNone(bundle_file)
self.assertEqual([
call('/cache-dir'),
call('/bundle-cached-path.tmp')
], os_path_exists_mock.call_args_list)
cache_path_mock.assert_called_with('/cache-dir', '/bundle-url')
get_url_mock.assert_called_with('/bundle-url')
urlretrieve_mock.assert_called_with('/bundle-url-resolved', '/bundle-cached-path.tmp')
get_logger_mock.assert_called_with('conductr_cli.resolvers.uri_resolver')
log_mock.info.assert_called_with('Retrieving /bundle-url-resolved')
def test_resolve_not_found(self):
os_path_exists_mock = MagicMock(side_effect=[True, False])
cache_path_mock = MagicMock(return_value='/bundle-cached-path')
urlretrieve_mock = MagicMock(side_effect=URLError('no_such.bundle'))
get_url_mock = MagicMock(return_value=('bundle-name',
'/bundle-url-resolved'))
get_logger_mock, log_mock = create_mock_logger()
with patch('os.path.exists', os_path_exists_mock), \
patch('conductr_cli.resolvers.uri_resolver.cache_path', cache_path_mock), \
patch('conductr_cli.resolvers.uri_resolver.get_url', get_url_mock), \
patch('conductr_cli.resolvers.uri_resolver.urlretrieve', urlretrieve_mock), \
patch('logging.getLogger', get_logger_mock):
is_resolved, bundle_name, bundle_file = uri_resolver.resolve_bundle(
'/cache-dir', '/bundle-url')
self.assertFalse(is_resolved)
self.assertIsNone(bundle_name)
self.assertIsNone(bundle_file)
self.assertEqual([call('/cache-dir'),
call('/bundle-cached-path.tmp')],
os_path_exists_mock.call_args_list)
cache_path_mock.assert_called_with('/cache-dir', '/bundle-url')
get_url_mock.assert_called_with('/bundle-url')
urlretrieve_mock.assert_called_with('/bundle-url-resolved',
'/bundle-cached-path.tmp')
get_logger_mock.assert_called_with(
'conductr_cli.resolvers.uri_resolver')
log_mock.info.assert_called_with('Retrieving /bundle-url-resolved')
def test_show_progress_bar(self):
os_path_exists_mock = MagicMock(side_effect=[True, True])
os_remove_mock = MagicMock()
os_chmod_mock = MagicMock()
file_move_mock = MagicMock()
cache_path_mock = MagicMock(return_value='/bundle-cached-path')
get_url_mock = MagicMock(
return_value=('bundle-name',
'http://site.com/bundle-url-resolved'))
urlretrieve_mock = MagicMock()
report_hook_mock = MagicMock()
show_progress_mock = MagicMock(return_value=report_hook_mock)
get_logger_mock, log_mock = create_mock_logger()
with patch('os.path.exists', os_path_exists_mock), \
patch('os.remove', os_remove_mock), \
patch('os.chmod', os_chmod_mock), \
patch('shutil.move', file_move_mock), \
patch('conductr_cli.resolvers.uri_resolver.cache_path', cache_path_mock), \
patch('conductr_cli.resolvers.uri_resolver.get_url', get_url_mock), \
patch('conductr_cli.resolvers.uri_resolver.urlretrieve', urlretrieve_mock), \
patch('conductr_cli.resolvers.uri_resolver.show_progress', show_progress_mock), \
patch('logging.getLogger', get_logger_mock):
is_resolved, bundle_name, bundle_file = uri_resolver.resolve_bundle(
'/cache-dir', 'http://site.com/bundle-url')
self.assertTrue(is_resolved)
self.assertEqual('bundle-name', bundle_name)
self.assertEqual('/bundle-cached-path', bundle_file)
self.assertEqual([call('/cache-dir'),
call('/bundle-cached-path.tmp')],
os_path_exists_mock.call_args_list)
cache_path_mock.assert_called_with('/cache-dir',
'http://site.com/bundle-url')
get_url_mock.assert_called_with('http://site.com/bundle-url')
os_remove_mock.assert_called_with('/bundle-cached-path.tmp')
urlretrieve_mock.assert_called_with(
'http://site.com/bundle-url-resolved',
'/bundle-cached-path.tmp',
reporthook=report_hook_mock)
file_move_mock.assert_called_with('/bundle-cached-path.tmp',
'/bundle-cached-path')
get_logger_mock.assert_called_with(
'conductr_cli.resolvers.uri_resolver')
log_mock.info.assert_called_with(
'Retrieving http://site.com/bundle-url-resolved')
def test_custom_resolver_chain(self):
custom_settings = ConfigFactory.parse_string(
strip_margin("""|resolvers = [
| conductr_cli.resolvers.uri_resolver
|]
|""")
)
get_logger_mock, log_mock = create_mock_logger()
with patch('logging.getLogger', get_logger_mock):
result = resolver.resolver_chain(custom_settings)
expected_result = [uri_resolver]
self.assertEqual(expected_result, result)
get_logger_mock.assert_called_with('conductr_cli.resolver')
log_mock.info.assert_called_with('Using custom bundle resolver chain [\'conductr_cli.resolvers.uri_resolver\']')
def test_custom_resolver_chain(self):
custom_settings = ConfigFactory.parse_string(
strip_margin("""|resolvers = [
| conductr_cli.resolvers.uri_resolver
|]
|"""))
get_logger_mock, log_mock = create_mock_logger()
with patch('logging.getLogger', get_logger_mock):
result = resolver.resolver_chain(custom_settings, False)
expected_result = [uri_resolver]
self.assertEqual(expected_result, result)
get_logger_mock.assert_called_with('conductr_cli.resolver')
log_mock.info.assert_called_with(
'Using custom bundle resolver chain [\'conductr_cli.resolvers.uri_resolver\']'
)
def test_uri_found(self):
exists_mock = MagicMock(return_value=True)
get_logger_mock, log_mock = create_mock_logger()
with patch('os.path.exists', exists_mock), \
patch('logging.getLogger', get_logger_mock):
is_resolved, bundle_name, bundle_file = uri_resolver.load_bundle_configuration_from_cache(
'/cache-dir',
'http://site.com/path/bundle-file.zip')
self.assertTrue(is_resolved)
self.assertEqual('bundle-file.zip', bundle_name)
self.assertEqual('/cache-dir/bundle-file.zip', bundle_file)
exists_mock.assert_called_with('/cache-dir/bundle-file.zip')
get_logger_mock.assert_called_with('conductr_cli.resolvers.uri_resolver')
log_mock.info.assert_called_with('Retrieving from cache /cache-dir/bundle-file.zip')
def test_uri_found(self):
exists_mock = MagicMock(return_value=True)
get_logger_mock, log_mock = create_mock_logger()
with patch('os.path.exists', exists_mock), \
patch('logging.getLogger', get_logger_mock):
is_resolved, bundle_name, bundle_file = uri_resolver.load_bundle_configuration_from_cache(
'/cache-dir', 'http://site.com/path/bundle-file.zip')
self.assertTrue(is_resolved)
self.assertEqual('bundle-file.zip', bundle_name)
self.assertEqual('/cache-dir/bundle-file.zip', bundle_file)
exists_mock.assert_called_with('/cache-dir/bundle-file.zip')
get_logger_mock.assert_called_with(
'conductr_cli.resolvers.uri_resolver')
log_mock.info.assert_called_with(
'Retrieving from cache /cache-dir/bundle-file.zip')
def test_resolve_success(self):
os_path_exists_mock = MagicMock(side_effect=[True, True])
os_remove_mock = MagicMock()
os_chmod_mock = MagicMock()
file_move_mock = MagicMock()
cache_path_mock = MagicMock(return_value='/images/conductr-1.0.0.tgz')
get_url_mock = MagicMock(return_value=('conductr-1.0.0.tgz',
'conductr-binary-uri'))
urlretrieve_mock = MagicMock()
get_logger_mock, log_mock = create_mock_logger()
with patch('os.path.exists', os_path_exists_mock), \
patch('os.remove', os_remove_mock), \
patch('os.chmod', os_chmod_mock), \
patch('shutil.move', file_move_mock), \
patch('conductr_cli.resolvers.uri_resolver.cache_path', cache_path_mock), \
patch('conductr_cli.resolvers.uri_resolver.get_url', get_url_mock), \
patch('conductr_cli.resolvers.uri_resolver.urlretrieve', urlretrieve_mock), \
patch('logging.getLogger', get_logger_mock):
is_resolved, file_name, cached_file = uri_resolver.resolve_file(
'/images', 'conductr-binary-uri')
self.assertTrue(is_resolved)
self.assertEqual('conductr-1.0.0.tgz', file_name)
self.assertEqual('/images/conductr-1.0.0.tgz', cached_file)
self.assertEqual(
[call('/images'),
call('/images/conductr-1.0.0.tgz.tmp')],
os_path_exists_mock.call_args_list)
cache_path_mock.assert_called_with('/images', 'conductr-binary-uri')
get_url_mock.assert_called_with('conductr-binary-uri')
os_remove_mock.assert_called_with('/images/conductr-1.0.0.tgz.tmp')
urlretrieve_mock.assert_called_with('conductr-binary-uri',
'/images/conductr-1.0.0.tgz.tmp')
file_move_mock.assert_called_with('/images/conductr-1.0.0.tgz.tmp',
'/images/conductr-1.0.0.tgz')
get_logger_mock.assert_called_with(
'conductr_cli.resolvers.uri_resolver')
log_mock.info.assert_called_with('Retrieving conductr-binary-uri')
def test_show_progress_bar(self):
os_path_exists_mock = MagicMock(side_effect=[True, True])
os_remove_mock = MagicMock()
os_chmod_mock = MagicMock()
file_move_mock = MagicMock()
cache_path_mock = MagicMock(return_value='/bundle-cached-path')
get_url_mock = MagicMock(return_value=('bundle-name', 'http://site.com/bundle-url-resolved'))
urlretrieve_mock = MagicMock()
report_hook_mock = MagicMock()
show_progress_mock = MagicMock(return_value=report_hook_mock)
get_logger_mock, log_mock = create_mock_logger()
with patch('os.path.exists', os_path_exists_mock), \
patch('os.remove', os_remove_mock), \
patch('os.chmod', os_chmod_mock), \
patch('shutil.move', file_move_mock), \
patch('conductr_cli.resolvers.uri_resolver.cache_path', cache_path_mock), \
patch('conductr_cli.resolvers.uri_resolver.get_url', get_url_mock), \
patch('conductr_cli.resolvers.uri_resolver.urlretrieve', urlretrieve_mock), \
patch('conductr_cli.resolvers.uri_resolver.show_progress', show_progress_mock), \
patch('logging.getLogger', get_logger_mock):
is_resolved, bundle_name, bundle_file = uri_resolver.resolve_bundle('/cache-dir',
'http://site.com/bundle-url')
self.assertTrue(is_resolved)
self.assertEqual('bundle-name', bundle_name)
self.assertEqual('/bundle-cached-path', bundle_file)
self.assertEqual([
call('/cache-dir'),
call('/bundle-cached-path.tmp')
], os_path_exists_mock.call_args_list)
cache_path_mock.assert_called_with('/cache-dir', 'http://site.com/bundle-url')
get_url_mock.assert_called_with('http://site.com/bundle-url')
os_remove_mock.assert_called_with('/bundle-cached-path.tmp')
urlretrieve_mock.assert_called_with('http://site.com/bundle-url-resolved',
'/bundle-cached-path.tmp', reporthook=report_hook_mock)
file_move_mock.assert_called_with('/bundle-cached-path.tmp', '/bundle-cached-path')
get_logger_mock.assert_called_with('conductr_cli.resolvers.uri_resolver')
log_mock.info.assert_called_with('Retrieving http://site.com/bundle-url-resolved')
def test_resolve_success_create_cache_dir(self):
os_path_exists_mock = MagicMock(side_effect=[False, False])
file_move_mock = MagicMock()
os_chmod_mock = MagicMock()
os_mkdirs_mock = MagicMock(return_value=())
cache_path_mock = MagicMock(return_value='/bundle-cached-path')
get_url_mock = MagicMock(return_value=('bundle-name',
'/bundle-url-resolved'))
urlretrieve_mock = MagicMock()
get_logger_mock, log_mock = create_mock_logger()
with patch('os.path.exists', os_path_exists_mock), \
patch('os.makedirs', os_mkdirs_mock), \
patch('os.chmod', os_chmod_mock), \
patch('shutil.move', file_move_mock), \
patch('conductr_cli.resolvers.uri_resolver.cache_path', cache_path_mock), \
patch('conductr_cli.resolvers.uri_resolver.get_url', get_url_mock), \
patch('conductr_cli.resolvers.uri_resolver.urlretrieve', urlretrieve_mock), \
patch('logging.getLogger', get_logger_mock):
is_resolved, bundle_name, bundle_file = uri_resolver.resolve_bundle(
'/cache-dir', '/bundle-url')
self.assertTrue(is_resolved)
self.assertEqual('bundle-name', bundle_name)
self.assertEqual('/bundle-cached-path', bundle_file)
self.assertEqual([call('/cache-dir'),
call('/bundle-cached-path.tmp')],
os_path_exists_mock.call_args_list)
os_mkdirs_mock.assert_called_with('/cache-dir', mode=448)
cache_path_mock.assert_called_with('/cache-dir', '/bundle-url')
get_url_mock.assert_called_with('/bundle-url')
urlretrieve_mock.assert_called_with('/bundle-url-resolved',
'/bundle-cached-path.tmp')
file_move_mock.assert_called_with('/bundle-cached-path.tmp',
'/bundle-cached-path')
get_logger_mock.assert_called_with(
'conductr_cli.resolvers.uri_resolver')
log_mock.info.assert_called_with('Retrieving /bundle-url-resolved')
def test_resolve_success_create_cache_dir(self):
os_path_exists_mock = MagicMock(side_effect=[False, False])
file_move_mock = MagicMock()
os_chmod_mock = MagicMock()
os_mkdirs_mock = MagicMock(return_value=())
cache_path_mock = MagicMock(return_value='/bundle-cached-path')
get_url_mock = MagicMock(return_value=('bundle-name', '/bundle-url-resolved'))
urlretrieve_mock = MagicMock()
get_logger_mock, log_mock = create_mock_logger()
with patch('os.path.exists', os_path_exists_mock), \
patch('os.makedirs', os_mkdirs_mock), \
patch('os.chmod', os_chmod_mock), \
patch('shutil.move', file_move_mock), \
patch('conductr_cli.resolvers.uri_resolver.cache_path', cache_path_mock), \
patch('conductr_cli.resolvers.uri_resolver.get_url', get_url_mock), \
patch('conductr_cli.resolvers.uri_resolver.urlretrieve', urlretrieve_mock), \
patch('logging.getLogger', get_logger_mock):
is_resolved, bundle_name, bundle_file = uri_resolver.resolve_bundle_configuration('/cache-dir',
'/bundle-url')
self.assertTrue(is_resolved)
self.assertEqual('bundle-name', bundle_name)
self.assertEqual('/bundle-cached-path', bundle_file)
self.assertEqual([
call('/cache-dir'),
call('/bundle-cached-path.tmp')
], os_path_exists_mock.call_args_list)
os_mkdirs_mock.assert_called_with('/cache-dir', mode=448)
cache_path_mock.assert_called_with('/cache-dir', '/bundle-url')
get_url_mock.assert_called_with('/bundle-url')
urlretrieve_mock.assert_called_with('/bundle-url-resolved', '/bundle-cached-path.tmp')
file_move_mock.assert_called_with('/bundle-cached-path.tmp', '/bundle-cached-path')
get_logger_mock.assert_called_with('conductr_cli.resolvers.uri_resolver')
log_mock.info.assert_called_with('Retrieving /bundle-url-resolved')
