Skip to content

Commit 6bbba52

Browse files
authored
Update test to support new permissions (#562)
1 parent eb4b689 commit 6bbba52

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

azure-kusto-ingest/tests/test_e2e_ingest.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,7 @@
3333
StreamDescriptor,
3434
ManagedStreamingIngestClient,
3535
)
36+
from azure.kusto.ingest._resource_manager import _ResourceUri
3637

3738

3839
@pytest.fixture(params=["ManagedStreaming", "NormalClient"])
@@ -546,7 +547,11 @@ async def test_streaming_ingest_from_blob(self, is_managed_streaming):
546547
ingestion_mapping_reference="JsonMapping",
547548
ingestion_mapping_kind=IngestionMappingKind.JSON,
548549
)
549-
containers = self.ingest_client._resource_manager.get_containers()
550+
export_containers_list = self.ingest_client._resource_manager._kusto_client.execute("NetDefaultDB", ".show export containers")
551+
containers = [_ResourceUri(s["StorageRoot"]) for s in export_containers_list.primary_results[0]]
552+
553+
for c in containers:
554+
self.ingest_client._resource_manager._ranked_storage_account_set.add_storage_account(c.storage_account_name)
550555

551556
with FileDescriptor(self.json_file_path).open(False) as stream:
552557
blob_descriptor = self.ingest_client.upload_blob(

0 commit comments

Comments
 (0)