def setUp(self): self.storage = BlobStorage() self.testbed = TestbedWithFiles() self.testbed.activate() self.testbed.init_blobstore_stub() self.testbed.init_datastore_v3_stub()
def test_get_blobstorage_storage(self): self.assertEqual( get_storage_class('rocket_engine.storage.BlobStorage'), BlobStorage ) self.storage = BlobStorage()
def test_get_blobstorage_storage(self): self.assertEqual( get_storage_class('rocket_engine.storage.BlobStorage'), BlobStorage) self.storage = BlobStorage()
class TestBlobStorage(TestCase): def setUp(self): self.storage = BlobStorage() self.testbed = TestbedWithFiles() self.testbed.activate() self.testbed.init_blobstore_stub() self.testbed.init_datastore_v3_stub() def tearDown(self): self.testbed.deactivate() def test_get_blobstorage_storage(self): self.assertEqual( get_storage_class('rocket_engine.storage.BlobStorage'), BlobStorage) self.storage = BlobStorage() def test_save(self): file_name = 'test_name' file_content = b'test_content' f_name = self.storage.save(file_name, ContentFile(file_content)) self.assertEqual(f_name, file_name) def test_open(self): file_name = 'test_name' file_content = b'test_content' self.storage.save(file_name, ContentFile(file_content)) f = self.storage.open(file_name) self.assertEqual(f.read(), file_content) def test_exists(self): file_name = 'test_name' file_content = b'test_content' self.assertFalse(self.storage.exists(file_name)) self.storage.save(file_name, ContentFile(file_content)) self.assertTrue(self.storage.exists(file_name)) def test_available_name(self): file_name = 'test_name' file_content = b'test_content' self.storage.save(file_name, ContentFile(file_content)) f_name_1 = self.storage.save(file_name, ContentFile(file_content)) f_name_2 = self.storage.save(file_name, ContentFile(file_content)) self.assertEqual(f_name_1, "%s_1" % file_name) self.assertEqual(f_name_2, "%s_2" % file_name) def test_unicode_file_names(self): file_name = u'¿Cómo?' file_content = b'test_content' f_name = self.storage.save(file_name, ContentFile(file_content)) self.assertEqual(f_name, file_name)
class TestBlobStorage(TestCase): def setUp(self): self.storage = BlobStorage() self.testbed = TestbedWithFiles() self.testbed.activate() self.testbed.init_blobstore_stub() self.testbed.init_datastore_v3_stub() def tearDown(self): self.testbed.deactivate() def test_get_blobstorage_storage(self): self.assertEqual( get_storage_class('rocket_engine.storage.BlobStorage'), BlobStorage ) self.storage = BlobStorage() def test_save(self): file_name = 'test_name' file_content = b'test_content' f_name = self.storage.save(file_name, ContentFile(file_content)) self.assertEqual(f_name, file_name) def test_open(self): file_name = 'test_name' file_content = b'test_content' self.storage.save(file_name, ContentFile(file_content)) f = self.storage.open(file_name) self.assertEqual(f.read(), file_content) def test_exists(self): file_name = 'test_name' file_content = b'test_content' self.assertFalse(self.storage.exists(file_name)) self.storage.save(file_name, ContentFile(file_content)) self.assertTrue(self.storage.exists(file_name)) def test_available_name(self): file_name = 'test_name' file_content = b'test_content' self.storage.save(file_name, ContentFile(file_content)) f_name_1 = self.storage.save(file_name, ContentFile(file_content)) f_name_2 = self.storage.save(file_name, ContentFile(file_content)) self.assertEqual(f_name_1, "%s_1" % file_name) self.assertEqual(f_name_2, "%s_2" % file_name) def test_unicode_file_names(self): file_name = u'¿Cómo?' file_content = b'test_content' f_name = self.storage.save(file_name, ContentFile(file_content)) self.assertEqual(f_name, file_name)