def test_override_settings(self): with override_settings(AZURE_CONTAINER='foo1'): storage = azure_storage.AzureStorage() self.assertEqual(storage.azure_container, 'foo1') with override_settings(AZURE_CONTAINER='foo2'): storage = azure_storage.AzureStorage() self.assertEqual(storage.azure_container, 'foo2')
def test_blob_custom_service_params(self): storage = azure_storage.AzureStorage() storage.is_emulated = True storage.endpoint_suffix = 'foo_suffix' storage.account_name = 'foo_name' storage.account_key = 'foo_key' storage.sas_token = 'foo_token' storage.azure_ssl = True storage.custom_domain = 'foo_domain' storage.custom_connection_string = 'foo_conn' storage.token_credential = 'foo_cred' with mock.patch( 'storages.backends.azure_storage.BlockBlobService', autospec=True) as c_mocked: self.assertIsNotNone(storage.custom_service) c_mocked.assert_called_once_with( account_name='foo_name', account_key='foo_key', sas_token='foo_token', is_emulated=True, protocol='https', custom_domain='foo_domain', connection_string='foo_conn', token_credential='foo_cred', endpoint_suffix='foo_suffix')
def setUp(self, *args): self.storage = azure_storage.AzureStorage() self.storage._service = mock.MagicMock() self.storage._custom_service = mock.MagicMock() self.storage.overwrite_files = True self.container_name = 'test' self.storage.azure_container = self.container_name
def test_container_client_default_params(self): storage = azure_storage.AzureStorage() storage.account_name = self.account_name with mock.patch('storages.backends.azure_storage.ContainerClient', autospec=True) as c_mocked: self.assertIsNotNone(storage.client) c_mocked.assert_called_once_with( 'https://test.blob.core.windows.net', None, credential=None)
def setUp(self, *args): self.storage = azure_storage.AzureStorage() self.storage._client = mock.MagicMock() self.storage.overwrite_files = True self.account_name = 'test' self.account_key = 'key' self.container_name = 'test' self.storage.azure_account = self.account_name self.storage.account_key = self.account_key self.storage.azure_container = self.container_name
def test_container_client_default_params(self): storage = azure_storage.AzureStorage() storage.account_name = self.account_name with mock.patch('storages.backends.azure_storage.BlobServiceClient', autospec=True) as bsc_mocked: client_mock = mock.MagicMock() bsc_mocked.return_value.get_container_client.return_value = client_mock self.assertEqual(storage.client, client_mock) bsc_mocked.assert_called_once_with( 'https://test.blob.core.windows.net', credential=None)
def setUp(self, *args): self.storage = azure_storage.AzureStorage() self.storage.is_emulated = True self.storage.account_name = "XXX" self.storage.account_key = "KXXX" self.storage.azure_container = "test" self.storage.service.delete_container( self.storage.azure_container, fail_not_exist=False) self.storage.service.create_container( self.storage.azure_container, public_access=False, fail_on_exist=False)
def test_container_client_params_connection_string(self): storage = azure_storage.AzureStorage() storage.account_name = self.account_name storage.connection_string = 'foo_conn' with mock.patch( 'storages.backends.azure_storage.BlobServiceClient.from_connection_string', spec=azure_storage.BlobServiceClient.from_connection_string ) as bsc_mocked: client_mock = mock.MagicMock() bsc_mocked.return_value.get_container_client.return_value = client_mock self.assertEqual(storage.client, client_mock) bsc_mocked.assert_called_once_with('foo_conn')
def test_container_client_params_sas_token(self): storage = azure_storage.AzureStorage() storage.account_name = 'foo_name' storage.azure_ssl = False storage.custom_domain = 'foo_domain' storage.sas_token = 'foo_token' with mock.patch('storages.backends.azure_storage.BlobServiceClient', autospec=True) as bsc_mocked: client_mock = mock.MagicMock() bsc_mocked.return_value.get_container_client.return_value = client_mock self.assertEqual(storage.client, client_mock) bsc_mocked.assert_called_once_with('http://foo_domain', credential='foo_token')
def test_blob_custom_service_default_params(self): storage = azure_storage.AzureStorage() with mock.patch('storages.backends.azure_storage.BlockBlobService', autospec=True) as c_mocked: self.assertIsNotNone(storage.custom_service) c_mocked.assert_called_once_with(account_name=None, account_key=None, sas_token=None, is_emulated=False, protocol='https', custom_domain=None, connection_string=None, token_credential=None, endpoint_suffix=None)
def test_container_client_params(self): storage = azure_storage.AzureStorage() storage.account_name = 'foo_name' storage.account_key = 'foo_key' storage.sas_token = 'foo_token' storage.azure_ssl = True storage.custom_domain = 'foo_domain' storage.connection_string = 'foo_conn' storage.token_credential = 'foo_cred' with mock.patch('storages.backends.azure_storage.ContainerClient', autospec=True) as c_mocked: self.assertIsNotNone(storage.client) c_mocked.assert_called_once_with('foo_conn', None, credential='foo_key')
def test_blob_custom_service_params_no_emulator(self): """Should pass custom domain when emulator is not used""" storage = azure_storage.AzureStorage() storage.is_emulated = False storage.custom_domain = 'foo_domain' with mock.patch('storages.backends.azure_storage.BlockBlobService', autospec=True) as c_mocked: self.assertIsNotNone(storage.custom_service) c_mocked.assert_called_once_with(account_name=None, account_key=None, sas_token=None, is_emulated=False, protocol='https', custom_domain='foo_domain', connection_string=None, token_credential=None, endpoint_suffix=None)
def test_override_init_argument(self): storage = azure_storage.AzureStorage(azure_container='foo1') self.assertEqual(storage.azure_container, 'foo1') storage = azure_storage.AzureStorage(azure_container='foo2') self.assertEqual(storage.azure_container, 'foo2')
def test_url_custom_endpoint(self): storage = azure_storage.AzureStorage() storage.is_emulated = True storage.custom_domain = 'foobar:123456' self.assertTrue( storage.url("my_file.txt").startswith('https://foobar:123456/'))
def setUp(self, *args): self.storage = azure_storage.AzureStorage() self.storage._connection = mock.MagicMock() self.container_name = 'test' self.storage.azure_container = self.container_name