def test_list_non_empty(self, client):
base_path = "/path/"
# Create some files to return
dir_prefix = BlobPrefix()
dir_prefix.name = base_path + "dir"
blob1_props = BlobProperties()
blob1_props.size = 42
blob1_props.name = base_path + "file1"
blob2_props = BlobProperties()
blob2_props.size = 0
blob2_props.name = base_path + "file2"
client.get_container_client().walk_blobs.return_value = MockBlobList(
[dir_prefix, blob1_props, blob2_props]
)
store = AzureBlobStoreService()
store.set_connection(connection=client)
key_path = self.wasbs_base + base_path
results = store.list(key=key_path)
assert len(results["blobs"]) == 2
assert len(results["prefixes"]) == 1
assert results["prefixes"][0] == "dir"
assert results["blobs"][0][0] == "file1"
assert results["blobs"][0][1] == 42
assert results["blobs"][1][0] == "file2"
assert results["blobs"][1][1] == 0
def test_list_empty(self, client):
client.get_container_client().walk_blobs.return_value = MockBlobList([])
store = AzureBlobStoreService()
store.set_connection(connection=client)
key_path = self.wasbs_base + "path"
assert store.list(key=key_path) == {"blobs": [], "prefixes": []}
def test_list_non_empty(self, client):
base_path = "/path/"
# Create some files to return
dir_prefix = BlobPrefix()
dir_prefix.name = base_path + "dir"
blob_props = BlobProperties()
blob_props.content_length = 42
blob = Blob(base_path + "file", props=blob_props)
client.return_value.list_blobs.return_value = MockBlobList([dir_prefix, blob])
store = AzureBlobStoreService()
key_path = self.wasbs_base + base_path
results = store.list(key=key_path)
assert len(results["blobs"]) == 1
assert len(results["prefixes"]) == 1
assert results["prefixes"][0] == "dir"
assert results["blobs"][0][0] == "file"
assert results["blobs"][0][1] == 42
def test_list_empty(self, client):
client.return_value.list_blobs.return_value = MockBlobList([])
store = AzureBlobStoreService()
key_path = self.wasbs_base + "path"
assert store.list(key=key_path) == {"blobs": [], "prefixes": []}