def setUp(self):
super(GSStorageTest, self).setUp()
self.storage = AppEngineStorage(storage_service=CLOUD_STORAGE_SERVICE, cloud_storage_bucket='test_bucket')
file_name = '/test_bucket/file.test'
with cloudstorage.open(file_name, 'w') as f:
f.write('abcdef')
self.file_key = '/gs' + file_name
self.test_file_info = self.file_info(self.file_key)
def setUp(self):
super(BlobstoreStorageTest, self).setUp()
self.storage = AppEngineStorage(storage_service=BLOBSTORE_SERVICE)
file_name = files.blobstore.create()
with files.open(file_name, 'a') as f:
f.write('abcdef')
files.finalize(file_name)
self.blob_key = files.blobstore.get_blob_key(file_name)
self.file_key = str(self.blob_key)
self.test_file_info = self.file_info(self.file_key)
class GSStorageTest(AppEngineStorageBaseTest, TestCase):
def setUp(self):
super(GSStorageTest, self).setUp()
self.storage = AppEngineStorage(storage_service=CLOUD_STORAGE_SERVICE, cloud_storage_bucket='test_bucket')
file_name = '/test_bucket/file.test'
with cloudstorage.open(file_name, 'w') as f:
f.write('abcdef')
self.file_key = '/gs' + file_name
self.test_file_info = self.file_info(self.file_key)
def file_info(self, name):
return CloudStorageInfo(name)
def test_file_url(self):
url = self.storage.url(self.file_key)
self.assertTrue(url.startswith('/_ah/img/encoded_gs_file:'))
class BlobstoreStorageTest(AppEngineStorageBaseTest, TestCase):
def setUp(self):
super(BlobstoreStorageTest, self).setUp()
self.storage = AppEngineStorage(storage_service=BLOBSTORE_SERVICE)
file_name = files.blobstore.create()
with files.open(file_name, 'a') as f:
f.write('abcdef')
files.finalize(file_name)
self.blob_key = files.blobstore.get_blob_key(file_name)
self.file_key = str(self.blob_key)
self.test_file_info = self.file_info(self.file_key)
def file_info(self, name):
key = BlobKey(name.split('/', 1)[0])
return BlobInfo(key)
def test_file_url(self):
url = self.storage.url(self.file_key)
self.assertEqual(url, '/_ah/img/%s' % self.file_key)