Skip to content

Commit

Permalink
Update test to support new permissions (#562)
Browse files Browse the repository at this point in the history
  • Loading branch information
AsafMah authored Jan 6, 2025
1 parent eb4b689 commit 6bbba52
Showing 1 changed file with 6 additions and 1 deletion.
7 changes: 6 additions & 1 deletion azure-kusto-ingest/tests/test_e2e_ingest.py
Original file line number Diff line number Diff line change
Expand Up @@ -33,6 +33,7 @@
StreamDescriptor,
ManagedStreamingIngestClient,
)
from azure.kusto.ingest._resource_manager import _ResourceUri


@pytest.fixture(params=["ManagedStreaming", "NormalClient"])
Expand Down Expand Up @@ -546,7 +547,11 @@ async def test_streaming_ingest_from_blob(self, is_managed_streaming):
ingestion_mapping_reference="JsonMapping",
ingestion_mapping_kind=IngestionMappingKind.JSON,
)
containers = self.ingest_client._resource_manager.get_containers()
export_containers_list = self.ingest_client._resource_manager._kusto_client.execute("NetDefaultDB", ".show export containers")
containers = [_ResourceUri(s["StorageRoot"]) for s in export_containers_list.primary_results[0]]

for c in containers:
self.ingest_client._resource_manager._ranked_storage_account_set.add_storage_account(c.storage_account_name)

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

0 comments on commit 6bbba52

Please sign in to comment.