def test_download_file(self, client):
client.return_value.walk_blobs.return_value = MockBlobList([])
dirname = tempfile.mkdtemp()
fpath = dirname + "/test.txt"
def mkfile(buffer):
return open(buffer.name, "w")
client.get_container_client().download_blob().readinto.side_effect = mkfile
base_path = "path/test.txt"
store = AzureBlobStoreService()
store.set_connection(connection=client)
key_path = self.wasbs_base + base_path
# Test without basename
store.download_file(key_path, fpath, use_basename=False)
client.get_container_client().download_blob.assert_called_with(base_path)
assert os.path.exists(fpath)
# Test without basename
store.download_file(key_path, dirname, use_basename=True)
client.get_container_client().download_blob.assert_called_with(base_path)
assert os.path.exists(fpath)
def test_download_file(self, client):
client.return_value.list_blobs.return_value = MockBlobList([])
dirname = tempfile.mkdtemp()
fpath = dirname + "/test.txt"
def mkfile(container, cloud_path, fname):
return open(fname, "w")
client.return_value.get_blob_to_path.side_effect = mkfile
base_path = "path/test.txt"
store = AzureBlobStoreService()
key_path = self.wasbs_base + base_path
# Test without basename
store.download_file(key_path, fpath, use_basename=False)
client.return_value.get_blob_to_path.assert_called_with(
"container", base_path, fpath
)
# Test without basename
store.download_file(key_path, dirname, use_basename=True)
client.return_value.get_blob_to_path.assert_called_with(
"container", base_path, fpath
)
