def test_copydir_with_children(self):
path = tempfile.mkdtemp()
_, filename = tempfile.mkstemp(dir=path)
bucket = Bucket('test')
with patch('gstorage.bucket.Blob') as mock_blob:
bucket.copydir(path)
assert mock_blob.call_count == 1
def test_bucket_badrequest(self):
Bucket.exists = MagicMock(return_value=False)
with patch('gstorage.bucket.Bucket.create',
MagicMock(side_effect=BadRequest('400'))):
with self.assertRaises(BadRequest):
bucket = Bucket.get_or_create('test', client='test')
assert bucket is None
def test_copydir_noaccess(self):
bucket = Bucket('test')
with self.assertRaises(OSError):
bucket.copydir('/root')
def test_get_default(self, mock_get_config, mock_credentials):
environ['GCLOUD_PROJECT'] = 'test'
Bucket.get_default()
assert mock_get_config.called_once_with('GCLOUD_DEFAULT_BUCKET_NAME')
def test_bucket_doesnot_exist(self, mock_create):
Bucket.exists = MagicMock(return_value=False)
Bucket.get_or_create('test', client='test')
assert mock_create.called
def test_no_call_to_client(self, mock_client):
Bucket.get_or_create('test', client=mock_client)
assert not mock_client.called
def test_call_to_client(self, mock_client, mock_credentials):
Bucket.get_or_create('test')
assert mock_client.called