[Stream Analytics] az stream-analytics output: Support managed identity with the 2021-10-01-preview API

src/stream-analytics/HISTORY.rst

@@ -3,6 +3,10 @@
 Release History
 ===============
 
+1.0.6
++++++++++++++++
+* Regenerate `stream-analytics output` AAZ commands for `2021-10-01-preview` and fix managed identity outputs such as Cosmos DB.
+
 1.0.5
 +++++++++++++++
 * Migrate ``stream-analytics input`` commands to the aaz framework, upgrading the input API version to 2021-10-01-preview.

src/stream-analytics/azext_stream_analytics/aaz/latest/stream_analytics/output/_delete.py

@@ -20,9 +20,9 @@ class Delete(AAZCommand):
     """
 
     _aaz_info = {
-        "version": "2020-03-01",
+        "version": "2021-10-01-preview",
         "resources": [
-            ["mgmt-plane", "/subscriptions/{}/resourcegroups/{}/providers/microsoft.streamanalytics/streamingjobs/{}/outputs/{}", "2020-03-01"],
+            ["mgmt-plane", "/subscriptions/{}/resourcegroups/{}/providers/microsoft.streamanalytics/streamingjobs/{}/outputs/{}", "2021-10-01-preview"],
         ]
     }
 
@@ -126,7 +126,7 @@ def url_parameters(self):
         def query_parameters(self):
             parameters = {
                 **self.serialize_query_param(
-                    "api-version", "2020-03-01",
+                    "api-version", "2021-10-01-preview",
                     required=True,
                 ),
             }

src/stream-analytics/azext_stream_analytics/generated/_input_helpers.py

@@ -18,7 +18,8 @@
 
 
 def _camel_to_snake(name):
-    return re.sub(r'(?<!^)(?=[A-Z])', '_', name).lower()
+    name = re.sub(r'(.)([A-Z][a-z]+)', r'\1_\2', name)
+    return re.sub(r'([a-z0-9])([A-Z])', r'\1_\2', name).lower()
 
 
 def _normalize_key(type_name):

src/stream-analytics/azext_stream_analytics/generated/_output_helpers.py

@@ -0,0 +1,30 @@
+# --------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for
+# license information.
+# --------------------------------------------------------------------------
+"""Translate legacy output REST payloads into flattened AAZ command args."""
+
+from ._input_helpers import _normalize_key, _snake_keys
+
+
+_NESTED_DATASOURCES = {
+    "microsoft_event_hub_event_hub",
+    "microsoft_service_bus_event_hub",
+}
+
+
+def output_data_to_args(datasource=None, serialization=None):
+    args = {}
+    if datasource:
+        datasource_key = _normalize_key(datasource.get('type', ''))
+        datasource_properties = _snake_keys(datasource.get('properties', {}) or {})
+        if datasource_key in _NESTED_DATASOURCES:
+            datasource_properties = {'properties': datasource_properties}
+        args['datasource'] = {datasource_key: datasource_properties}
+    if serialization:
+        serialization_key = _normalize_key(serialization.get('type', ''))
+        args['serialization'] = {
+            serialization_key: _snake_keys(serialization.get('properties', {}) or {})
+        }
+    return args

src/stream-analytics/azext_stream_analytics/generated/custom.py

@@ -13,6 +13,7 @@
 from azure.cli.core.util import sdk_no_wait
 
 from ._input_helpers import input_properties_to_args
+from ._output_helpers import output_data_to_args
 
 
 def stream_analytics_job_list(client,
@@ -389,17 +390,17 @@ def stream_analytics_output_create(cmd,
                                    serialization=None):
     from ..aaz.latest.stream_analytics.output import Create
 
-    result = Create(cli_ctx=cmd.cli_ctx)(command_args={
+    args = {
         "resource_group": resource_group_name,
         "job_name": job_name,
         "output_name": output_name,
         "if_match": if_match,
         "if_none_match": if_none_match,
-        "datasource": datasource,
         "time_window": time_window,
-        "size_window": size_window,
-        "serialization": serialization
-    })
+        "size_window": size_window
+    }
+    args.update(output_data_to_args(datasource, serialization))
+    result = Create(cli_ctx=cmd.cli_ctx)(command_args=args)
     return _stream_analytics_output_flatten_inner_properties(result)
 
 
@@ -421,14 +422,11 @@ def stream_analytics_output_update(cmd,
     }
     if if_match is not None:
         args["if_match"] = if_match
-    if datasource is not None:
-        args["datasource"] = datasource
     if time_window is not None:
         args["time_window"] = time_window
     if size_window is not None:
         args["size_window"] = size_window
-    if serialization is not None:
-        args["serialization"] = serialization
+    args.update(output_data_to_args(datasource, serialization))
 
     result = Patch(cli_ctx=cmd.cli_ctx)(command_args=args)
     return _stream_analytics_output_flatten_inner_properties(result)
@@ -457,16 +455,16 @@ def stream_analytics_output_test(cmd,
                                  no_wait=False):
     from ..aaz.latest.stream_analytics.output import Test
 
-    return Test(cli_ctx=cmd.cli_ctx)(command_args={
+    args = {
         "resource_group": resource_group_name,
         "job_name": job_name,
         "output_name": output_name,
-        "datasource": datasource,
         "time_window": time_window,
         "size_window": size_window,
-        "serialization": serialization,
         "no_wait": no_wait
-    })
+    }
+    args.update(output_data_to_args(datasource, serialization))
+    return Test(cli_ctx=cmd.cli_ctx)(command_args=args)
 
 # endregion
 

src/stream-analytics/azext_stream_analytics/tests/latest/recordings/test_input_create_policy_violation.yaml

@@ -61,7 +61,7 @@ interactions:
       User-Agent:
       - AZURECLI/2.87.0 azsdk-python-core/1.39.0 Python/3.12.10 (Windows-11-10.0.26200-SP0)
     method: GET
-    uri: https://management.azure.com/providers/Microsoft.Authorization/policyDefinitions/ea6c4923-510a-4346-be26-1894919a5b97?api-version=2024-05-01
+    uri: https://management.azure.com/providers/Microsoft.Authorization/policyDefinitions/ea6c4923-510a-4346-be26-1894919a5b97?api-version=2025-11-01
   response:
     body:
       string: '{"properties":{"displayName":"Stream Analytics job should use managed
@@ -117,7 +117,7 @@ interactions:
       User-Agent:
       - AZURECLI/2.87.0 azsdk-python-core/1.39.0 Python/3.12.10 (Windows-11-10.0.26200-SP0)
     method: PUT
-    uri: https://management.azure.com/subscriptions/00000000-0000-0000-0000-000000000000/resourceGroups/cli_test_stream_analytics_000001/providers/Microsoft.Authorization/policyAssignments/deny-asa-mi?api-version=2024-05-01
+    uri: https://management.azure.com/subscriptions/00000000-0000-0000-0000-000000000000/resourceGroups/cli_test_stream_analytics_000001/providers/Microsoft.Authorization/policyAssignments/deny-asa-mi?api-version=2025-11-01
   response:
     body:
       string: '{"properties":{"displayName":"Deny ASA non-MI authentication","policyDefinitionId":"/providers/Microsoft.Authorization/policyDefinitions/ea6c4923-510a-4346-be26-1894919a5b97","definitionVersion":"1.*.*","scope":"/subscriptions/00000000-0000-0000-0000-000000000000/resourceGroups/cli_test_stream_analytics_000001","parameters":{"effect":{"value":"Deny"}},"metadata":{"createdBy":"73deb370-6e80-42b7-bab4-943c887d3e3c","createdOn":"2026-06-23T05:11:19.7480399Z","updatedBy":null,"updatedOn":null},"enforcementMode":"Default"},"id":"/subscriptions/00000000-0000-0000-0000-000000000000/resourceGroups/cli_test_stream_analytics_000001/providers/Microsoft.Authorization/policyAssignments/deny-asa-mi","type":"Microsoft.Authorization/policyAssignments","name":"deny-asa-mi","systemData":{"createdBy":"test@example.com","createdByType":"User","createdAt":"2026-06-23T05:11:19.6816319Z","lastModifiedBy":"test@example.com","lastModifiedByType":"User","lastModifiedAt":"2026-06-23T05:11:19.6816319Z"}}'
@@ -488,7 +488,7 @@ interactions:
       User-Agent:
       - AZURECLI/2.87.0 azsdk-python-core/1.39.0 Python/3.12.10 (Windows-11-10.0.26200-SP0)
     method: DELETE
-    uri: https://management.azure.com/subscriptions/00000000-0000-0000-0000-000000000000/resourceGroups/cli_test_stream_analytics_000001/providers/Microsoft.Authorization/policyAssignments/deny-asa-mi?api-version=2024-05-01
+    uri: https://management.azure.com/subscriptions/00000000-0000-0000-0000-000000000000/resourceGroups/cli_test_stream_analytics_000001/providers/Microsoft.Authorization/policyAssignments/deny-asa-mi?api-version=2025-11-01
   response:
     body:
       string: '{"properties":{"displayName":"Deny ASA non-MI authentication","policyDefinitionId":"/providers/Microsoft.Authorization/policyDefinitions/ea6c4923-510a-4346-be26-1894919a5b97","definitionVersion":"1.*.*","scope":"/subscriptions/00000000-0000-0000-0000-000000000000/resourceGroups/cli_test_stream_analytics_000001","parameters":{"effect":{"value":"Deny"}},"metadata":{"createdBy":"73deb370-6e80-42b7-bab4-943c887d3e3c","createdOn":"2026-06-23T05:11:19.7480399Z","updatedBy":null,"updatedOn":null},"enforcementMode":"Default"},"id":"/subscriptions/00000000-0000-0000-0000-000000000000/resourceGroups/cli_test_stream_analytics_000001/providers/Microsoft.Authorization/policyAssignments/deny-asa-mi","type":"Microsoft.Authorization/policyAssignments","name":"deny-asa-mi","systemData":{"createdBy":"test@example.com","createdByType":"User","createdAt":"2026-06-23T05:11:19.6816319Z","lastModifiedBy":"test@example.com","lastModifiedByType":"User","lastModifiedAt":"2026-06-23T05:11:19.6816319Z"}}'

src/stream-analytics/azext_stream_analytics/tests/latest/recordings/test_input_sql_create_policy_violation.yaml

@@ -61,7 +61,7 @@ interactions:
       User-Agent:
       - AZURECLI/2.87.0 azsdk-python-core/1.39.0 Python/3.12.10 (Windows-11-10.0.26200-SP0)
     method: GET
-    uri: https://management.azure.com/providers/Microsoft.Authorization/policyDefinitions/ea6c4923-510a-4346-be26-1894919a5b97?api-version=2024-05-01
+    uri: https://management.azure.com/providers/Microsoft.Authorization/policyDefinitions/ea6c4923-510a-4346-be26-1894919a5b97?api-version=2025-11-01
   response:
     body:
       string: '{"properties":{"displayName":"Stream Analytics job should use managed
@@ -117,7 +117,7 @@ interactions:
       User-Agent:
       - AZURECLI/2.87.0 azsdk-python-core/1.39.0 Python/3.12.10 (Windows-11-10.0.26200-SP0)
     method: PUT
-    uri: https://management.azure.com/subscriptions/00000000-0000-0000-0000-000000000000/resourceGroups/cli_test_stream_analytics_000001/providers/Microsoft.Authorization/policyAssignments/deny-asa-mi?api-version=2024-05-01
+    uri: https://management.azure.com/subscriptions/00000000-0000-0000-0000-000000000000/resourceGroups/cli_test_stream_analytics_000001/providers/Microsoft.Authorization/policyAssignments/deny-asa-mi?api-version=2025-11-01
   response:
     body:
       string: '{"properties":{"displayName":"Deny ASA non-MI authentication","policyDefinitionId":"/providers/Microsoft.Authorization/policyDefinitions/ea6c4923-510a-4346-be26-1894919a5b97","definitionVersion":"1.*.*","scope":"/subscriptions/00000000-0000-0000-0000-000000000000/resourceGroups/cli_test_stream_analytics_000001","parameters":{"effect":{"value":"Deny"}},"metadata":{"createdBy":"73deb370-6e80-42b7-bab4-943c887d3e3c","createdOn":"2026-06-23T05:10:49.8454662Z","updatedBy":null,"updatedOn":null},"enforcementMode":"Default"},"id":"/subscriptions/00000000-0000-0000-0000-000000000000/resourceGroups/cli_test_stream_analytics_000001/providers/Microsoft.Authorization/policyAssignments/deny-asa-mi","type":"Microsoft.Authorization/policyAssignments","name":"deny-asa-mi","systemData":{"createdBy":"test@example.com","createdByType":"User","createdAt":"2026-06-23T05:10:49.6257765Z","lastModifiedBy":"test@example.com","lastModifiedByType":"User","lastModifiedAt":"2026-06-23T05:10:49.6257765Z"}}'
@@ -343,7 +343,7 @@ interactions:
       User-Agent:
       - AZURECLI/2.87.0 azsdk-python-core/1.39.0 Python/3.12.10 (Windows-11-10.0.26200-SP0)
     method: DELETE
-    uri: https://management.azure.com/subscriptions/00000000-0000-0000-0000-000000000000/resourceGroups/cli_test_stream_analytics_000001/providers/Microsoft.Authorization/policyAssignments/deny-asa-mi?api-version=2024-05-01
+    uri: https://management.azure.com/subscriptions/00000000-0000-0000-0000-000000000000/resourceGroups/cli_test_stream_analytics_000001/providers/Microsoft.Authorization/policyAssignments/deny-asa-mi?api-version=2025-11-01
   response:
     body:
       string: '{"properties":{"displayName":"Deny ASA non-MI authentication","policyDefinitionId":"/providers/Microsoft.Authorization/policyDefinitions/ea6c4923-510a-4346-be26-1894919a5b97","definitionVersion":"1.*.*","scope":"/subscriptions/00000000-0000-0000-0000-000000000000/resourceGroups/cli_test_stream_analytics_000001","parameters":{"effect":{"value":"Deny"}},"metadata":{"createdBy":"73deb370-6e80-42b7-bab4-943c887d3e3c","createdOn":"2026-06-23T05:10:49.8454662Z","updatedBy":null,"updatedOn":null},"enforcementMode":"Default"},"id":"/subscriptions/00000000-0000-0000-0000-000000000000/resourceGroups/cli_test_stream_analytics_000001/providers/Microsoft.Authorization/policyAssignments/deny-asa-mi","type":"Microsoft.Authorization/policyAssignments","name":"deny-asa-mi","systemData":{"createdBy":"test@example.com","createdByType":"User","createdAt":"2026-06-23T05:10:49.6257765Z","lastModifiedBy":"test@example.com","lastModifiedByType":"User","lastModifiedAt":"2026-06-23T05:10:49.6257765Z"}}'

src/stream-analytics/azext_stream_analytics/tests/latest/recordings/test_job_scale.yaml

@@ -412,7 +412,7 @@ interactions:
       User-Agent:
       - AZURECLI/2.77.0 azsdk-python-core/1.35.0 Python/3.12.10 (Windows-11-10.0.26100-SP0)
     method: PUT
-    uri: https://management.azure.com/subscriptions/00000000-0000-0000-0000-000000000000/resourcegroups/cli_test_stream_analytics_000001/providers/Microsoft.StreamAnalytics/streamingjobs/job/outputs/output?api-version=2020-03-01
+    uri: https://management.azure.com/subscriptions/00000000-0000-0000-0000-000000000000/resourcegroups/cli_test_stream_analytics_000001/providers/Microsoft.StreamAnalytics/streamingjobs/job/outputs/output?api-version=2021-10-01-preview
   response:
     body:
       string: "{\r\n  \"id\": \"/subscriptions/00000000-0000-0000-0000-000000000000/resourceGroups/cli_test_stream_analytics_000001/providers/Microsoft.StreamAnalytics/streamingjobs/job/outputs/output\",\r\n

src/stream-analytics/azext_stream_analytics/tests/latest/recordings/test_output_create_policy_violation.yaml

@@ -61,7 +61,7 @@ interactions:
       User-Agent:
       - AZURECLI/2.87.0 azsdk-python-core/1.39.0 Python/3.12.10 (Windows-11-10.0.26200-SP0)
     method: GET
-    uri: https://management.azure.com/providers/Microsoft.Authorization/policyDefinitions/ea6c4923-510a-4346-be26-1894919a5b97?api-version=2024-05-01
+    uri: https://management.azure.com/providers/Microsoft.Authorization/policyDefinitions/ea6c4923-510a-4346-be26-1894919a5b97?api-version=2025-11-01
   response:
     body:
       string: '{"properties":{"displayName":"Stream Analytics job should use managed
@@ -117,7 +117,7 @@ interactions:
       User-Agent:
       - AZURECLI/2.87.0 azsdk-python-core/1.39.0 Python/3.12.10 (Windows-11-10.0.26200-SP0)
     method: PUT
-    uri: https://management.azure.com/subscriptions/00000000-0000-0000-0000-000000000000/resourceGroups/cli_test_stream_analytics_000001/providers/Microsoft.Authorization/policyAssignments/deny-asa-mi?api-version=2024-05-01
+    uri: https://management.azure.com/subscriptions/00000000-0000-0000-0000-000000000000/resourceGroups/cli_test_stream_analytics_000001/providers/Microsoft.Authorization/policyAssignments/deny-asa-mi?api-version=2025-11-01
   response:
     body:
       string: '{"properties":{"displayName":"Deny ASA non-MI authentication","policyDefinitionId":"/providers/Microsoft.Authorization/policyDefinitions/ea6c4923-510a-4346-be26-1894919a5b97","definitionVersion":"1.*.*","scope":"/subscriptions/00000000-0000-0000-0000-000000000000/resourceGroups/cli_test_stream_analytics_000001","parameters":{"effect":{"value":"Deny"}},"metadata":{"createdBy":"73deb370-6e80-42b7-bab4-943c887d3e3c","createdOn":"2026-06-23T05:11:16.8793748Z","updatedBy":null,"updatedOn":null},"enforcementMode":"Default"},"id":"/subscriptions/00000000-0000-0000-0000-000000000000/resourceGroups/cli_test_stream_analytics_000001/providers/Microsoft.Authorization/policyAssignments/deny-asa-mi","type":"Microsoft.Authorization/policyAssignments","name":"deny-asa-mi","systemData":{"createdBy":"test@example.com","createdByType":"User","createdAt":"2026-06-23T05:11:16.6937901Z","lastModifiedBy":"test@example.com","lastModifiedByType":"User","lastModifiedAt":"2026-06-23T05:11:16.6937901Z"}}'
@@ -435,7 +435,7 @@ interactions:
       User-Agent:
       - AZURECLI/2.87.0 azsdk-python-core/1.39.0 Python/3.12.10 (Windows-11-10.0.26200-SP0)
     method: PUT
-    uri: https://management.azure.com/subscriptions/00000000-0000-0000-0000-000000000000/resourcegroups/cli_test_stream_analytics_000001/providers/Microsoft.StreamAnalytics/streamingjobs/job/outputs/output?api-version=2020-03-01
+    uri: https://management.azure.com/subscriptions/00000000-0000-0000-0000-000000000000/resourcegroups/cli_test_stream_analytics_000001/providers/Microsoft.StreamAnalytics/streamingjobs/job/outputs/output?api-version=2021-10-01-preview
   response:
     body:
       string: '{"error":{"code":"RequestDisallowedByPolicy","target":"output","message":"Resource
@@ -488,7 +488,7 @@ interactions:
       User-Agent:
       - AZURECLI/2.87.0 azsdk-python-core/1.39.0 Python/3.12.10 (Windows-11-10.0.26200-SP0)
     method: DELETE
-    uri: https://management.azure.com/subscriptions/00000000-0000-0000-0000-000000000000/resourceGroups/cli_test_stream_analytics_000001/providers/Microsoft.Authorization/policyAssignments/deny-asa-mi?api-version=2024-05-01
+    uri: https://management.azure.com/subscriptions/00000000-0000-0000-0000-000000000000/resourceGroups/cli_test_stream_analytics_000001/providers/Microsoft.Authorization/policyAssignments/deny-asa-mi?api-version=2025-11-01
   response:
     body:
       string: '{"properties":{"displayName":"Deny ASA non-MI authentication","policyDefinitionId":"/providers/Microsoft.Authorization/policyDefinitions/ea6c4923-510a-4346-be26-1894919a5b97","definitionVersion":"1.*.*","scope":"/subscriptions/00000000-0000-0000-0000-000000000000/resourceGroups/cli_test_stream_analytics_000001","parameters":{"effect":{"value":"Deny"}},"metadata":{"createdBy":"73deb370-6e80-42b7-bab4-943c887d3e3c","createdOn":"2026-06-23T05:11:16.8793748Z","updatedBy":null,"updatedOn":null},"enforcementMode":"Default"},"id":"/subscriptions/00000000-0000-0000-0000-000000000000/resourceGroups/cli_test_stream_analytics_000001/providers/Microsoft.Authorization/policyAssignments/deny-asa-mi","type":"Microsoft.Authorization/policyAssignments","name":"deny-asa-mi","systemData":{"createdBy":"test@example.com","createdByType":"User","createdAt":"2026-06-23T05:11:16.6937901Z","lastModifiedBy":"test@example.com","lastModifiedByType":"User","lastModifiedAt":"2026-06-23T05:11:16.6937901Z"}}'