class TestCaseWithBucket(TestBase):
def setUp(self):
self.bucket_name = 'my-bucket'
self.simulator = RawSimulator()
self.account_info = StubAccountInfo()
self.api = B2Api(self.account_info, raw_api=self.simulator)
(self.account_id, self.master_key) = self.simulator.create_account()
self.api.authorize_account('production', self.account_id,
self.master_key)
self.api_url = self.account_info.get_api_url()
self.account_auth_token = self.account_info.get_account_auth_token()
self.bucket = self.api.create_bucket('my-bucket', 'allPublic')
self.bucket_id = self.bucket.id_
def assertBucketContents(self, expected, *args, **kwargs):
"""
*args and **kwargs are passed to self.bucket.ls()
"""
actual = [(info.file_name, info.size, info.action, folder)
for (info, folder) in self.bucket.ls(*args, **kwargs)]
self.assertEqual(expected, actual)
class TestApi(TestBase):
def setUp(self):
self.account_info = InMemoryAccountInfo()
self.cache = DummyCache()
self.raw_api = RawSimulator()
self.api = B2Api(self.account_info, self.cache, self.raw_api)
(self.account_id, self.master_key) = self.raw_api.create_account()
def test_list_buckets(self):
self._authorize_account()
self.api.create_bucket('bucket1', 'allPrivate')
self.api.create_bucket('bucket2', 'allPrivate')
self.assertEqual(
['bucket1', 'bucket2'],
[b.name for b in self.api.list_buckets()],
)
def test_list_buckets_with_name(self):
self._authorize_account()
self.api.create_bucket('bucket1', 'allPrivate')
self.api.create_bucket('bucket2', 'allPrivate')
self.assertEqual(
['bucket1'],
[b.name for b in self.api.list_buckets(bucket_name='bucket1')],
)
def test_list_buckets_with_restriction(self):
self._authorize_account()
bucket1 = self.api.create_bucket('bucket1', 'allPrivate')
self.api.create_bucket('bucket2', 'allPrivate')
key = self.api.create_key(['listBuckets'], 'key1', bucket_id=bucket1.id_)
self.api.authorize_account('production', key['applicationKeyId'], key['applicationKey'])
self.assertEqual(
['bucket1'],
[b.name for b in self.api.list_buckets(bucket_name=bucket1.name)],
)
def test_get_bucket_by_name_with_bucket_restriction(self):
self._authorize_account()
bucket1 = self.api.create_bucket('bucket1', 'allPrivate')
key = self.api.create_key(['listBuckets'], 'key1', bucket_id=bucket1.id_)
self.api.authorize_account('production', key['applicationKeyId'], key['applicationKey'])
self.assertEqual(
bucket1.id_,
self.api.get_bucket_by_name('bucket1').id_,
)
def test_list_buckets_with_restriction_and_wrong_name(self):
self._authorize_account()
bucket1 = self.api.create_bucket('bucket1', 'allPrivate')
bucket2 = self.api.create_bucket('bucket2', 'allPrivate')
key = self.api.create_key(['listBuckets'], 'key1', bucket_id=bucket1.id_)
self.api.authorize_account('production', key['applicationKeyId'], key['applicationKey'])
with self.assertRaises(RestrictedBucket):
self.api.list_buckets(bucket_name=bucket2.name)
def test_list_buckets_with_restriction_and_no_name(self):
self._authorize_account()
bucket1 = self.api.create_bucket('bucket1', 'allPrivate')
self.api.create_bucket('bucket2', 'allPrivate')
key = self.api.create_key(['listBuckets'], 'key1', bucket_id=bucket1.id_)
self.api.authorize_account('production', key['applicationKeyId'], key['applicationKey'])
with self.assertRaises(RestrictedBucket):
self.api.list_buckets()
def _authorize_account(self):
self.api.authorize_account('production', self.account_id, self.master_key)
