def setUp(self): super(GoogleCloudStorageTestCase, self).setUp() self.storage = GoogleCloudStorage()
class GoogleCloudStorageTestCase(AppEngineStorageTestCase): def setUp(self): super(GoogleCloudStorageTestCase, self).setUp() self.storage = GoogleCloudStorage() def test_get_blobstorage_storage(self): self.assertEqual( get_storage_class('core.storage.GoogleCloudStorage'), GoogleCloudStorage) def test_save(self): file_name = 'test_name.txt' 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.txt' 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.txt' 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_delete(self): file_name = 'test_name.txt' file_content = b'test_content' self.storage.save(file_name, ContentFile(file_content)) self.assertTrue(self.storage.exists(file_name)) self.storage.delete(file_name) self.assertFalse(self.storage.exists(file_name)) def test_delete_bad_file(self): file_name = 'test_name.txt' file_content = b'test_content' self.storage.save(file_name, ContentFile(file_content)) self.assertTrue(self.storage.exists(file_name)) self.assertRaises(OSError, self.storage.delete, 'foo') def test_size(self): file_name = 'test_name.txt' file_content = b'test_content' self.storage.save(file_name, ContentFile(file_content)) self.assertEqual(12, self.storage.size(file_name)) def test_url(self): file_name = 'test_name.txt' file_content = b'test_content' self.storage.save(file_name, ContentFile(file_content)) self.assertEqual( u'https://storage.googleapis.com/badmovieknights/test_name.txt', self.storage.url(file_name)) def test_url_development(self): settings.DEBUG = True file_name = 'test_name.txt' file_content = b'test_content' self.storage.save(file_name, ContentFile(file_content)) self.assertEqual( u'http://localhost:8000/blobstore/blob/encoded_gs_file:' 'YmFkbW92aWVrbmlnaHRzL3Rlc3RfbmFtZS50eHQ=?display=inline', self.storage.url(file_name)) settings.DEBUG = False def test_list_dir(self): file_name = 'test_name.txt' file_content = b'test_content' self.storage.save(file_name, ContentFile(file_content)) self.assertEqual((['badmovieknights'], []), self.storage.listdir()) def test_get_file_info(self): file_name = 'test_name.txt' file_content = b'test_content' self.storage.save(file_name, ContentFile(file_content)) self.assertTrue(self.storage.size(file_name) is not None)