def testContainsWildcardNoMatches(self, mock_CreateWildcardIterator,
mock_gsutil_api):
mock_CreateWildcardIterator.return_value = iter([])
with self.assertRaises(CommandException):
ExpandUrlToSingleBlr('gs://test/hello*/', mock_gsutil_api,
'project_id', False,
CreateOrGetGsutilLogger('copy_test'))
def testWithSlashNoMatch(self, mock_gsutil_api):
mock_gsutil_api.ListObjects.return_value = iter([])
(exp_url, have_existing_dst_container) = ExpandUrlToSingleBlr(
'gs://test/folder/', mock_gsutil_api, 'project_id', False,
CreateOrGetGsutilLogger('copy_test'))
self.assertTrue(have_existing_dst_container)
self.assertEqual(exp_url, StorageUrlFromString('gs://test/folder/'))
def testContainsWildcardMultipleMatches(self, mock_CreateWildcardIterator,
mock_gsutil_api):
mock_CreateWildcardIterator.return_value = iter([
BucketListingObject(StorageUrlFromString('gs://test/helloworld')),
BucketListingObject(StorageUrlFromString('gs://test/helloworld2'))
])
with self.assertRaises(CommandException):
ExpandUrlToSingleBlr('gs://test/hello*/', mock_gsutil_api, 'project_id',
False, CreateOrGetGsutilLogger('copy_test'))
def testNoSlashPrefixSubstringMatch(self, mock_gsutil_api):
mock_gsutil_api.ListObjects.return_value = iter([
CloudApi.CsObjectOrPrefix('folderone/',
CloudApi.CsObjectOrPrefixType.PREFIX)
])
(exp_url, have_existing_dst_container) = ExpandUrlToSingleBlr(
'gs://test/folder', mock_gsutil_api, 'project_id', False,
CreateOrGetGsutilLogger('copy_test'))
self.assertFalse(have_existing_dst_container)
self.assertEqual(exp_url, StorageUrlFromString('gs://test/folder'))
def testLocalFileDirectory(self, mock_StorageUrlFromString, mock_gsutil_api):
mock_storage_url = mock.Mock()
mock_storage_url.isFileUrl.return_value = True
mock_storage_url.IsDirectory.return_value = True
mock_StorageUrlFromString.return_value = mock_storage_url
(exp_url, have_existing_dst_container) = ExpandUrlToSingleBlr(
'/home/test', mock_gsutil_api, 'project_id', False,
CreateOrGetGsutilLogger('copy_test'))
self.assertTrue(have_existing_dst_container)
self.assertEqual(exp_url, mock_storage_url)
def testContainsWildcardMatchesObject(self, mock_CreateWildcardIterator,
mock_gsutil_api):
storage_url = StorageUrlFromString('gs://test/helloworld')
mock_CreateWildcardIterator.return_value = iter(
[BucketListingObject(storage_url)])
(exp_url, have_existing_dst_container) = ExpandUrlToSingleBlr(
'gs://test/hello*/', mock_gsutil_api, 'project_id', False,
CreateOrGetGsutilLogger('copy_test'))
self.assertFalse(have_existing_dst_container)
self.assertEqual(exp_url, storage_url)
def testNoSlashFolderPlaceholder(self, mock_gsutil_api):
mock_gsutil_api.ListObjects.return_value = iter([
CloudApi.CsObjectOrPrefix(
apitools_messages.Object(name='folder_$folder$'),
CloudApi.CsObjectOrPrefixType.OBJECT)
])
(exp_url, have_existing_dst_container) = ExpandUrlToSingleBlr(
'gs://test/folder', mock_gsutil_api, 'project_id', False,
CreateOrGetGsutilLogger('copy_test'))
self.assertTrue(have_existing_dst_container)
self.assertEqual(exp_url, StorageUrlFromString('gs://test/folder'))