diff --git a/azure-kusto-ingest/tests/test_e2e_ingest.py b/azure-kusto-ingest/tests/test_e2e_ingest.py index 22ad8639..c9eb743d 100644 --- a/azure-kusto-ingest/tests/test_e2e_ingest.py +++ b/azure-kusto-ingest/tests/test_e2e_ingest.py @@ -33,6 +33,7 @@ StreamDescriptor, ManagedStreamingIngestClient, ) +from azure.kusto.ingest._resource_manager import _ResourceUri @pytest.fixture(params=["ManagedStreaming", "NormalClient"]) @@ -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(