Merge branch 'feature/hub_messaging_test_fixes' of https://github.com…

…/Azure/azure-iot-cli-extension into feature/hub_messaging_test_fixes
Azure · Dec 13, 2024 · 8355556 · 8355556
2 parents 40e4fe6 + 7142703
commit 8355556
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 1 deletion.
diff --git a/azext_iot/central/models/v2022_06_30_preview/template.py b/azext_iot/central/models/v2022_06_30_preview/template.py
@@ -91,6 +91,7 @@ def _extract_modules(self, template: dict) -> dict:
                     interfaces = self._extract_interfaces(template)
                     schema_names = self._extract_schema_names(interfaces)
                     components = self._extract_components(template)
+                    component_schema_names = []
                     if components:
                         component_schema_names = self._extract_schema_names(components)
 

diff --git a/azext_iot/tests/iothub/conftest.py b/azext_iot/tests/iothub/conftest.py
@@ -707,6 +707,7 @@ def _cosmos_db_provisioner():
         "cosmosdb": cosmos_obj,
         "database": database_obj,
         "container": container_obj,
+        "partitionKey": partition_key_path,
         "connectionString": _cosmos_db_get_cstring(account_name)
     }
 

diff --git a/azext_iot/tests/iothub/message_endpoint/test_iothub_message_endpoint_int.py b/azext_iot/tests/iothub/message_endpoint/test_iothub_message_endpoint_int.py
@@ -918,7 +918,7 @@ def test_iot_cosmos_endpoint_lifecycle(provisioned_cosmosdb_with_identity_module
     endpoint_names = generate_ep_names(3)
     partition_template = "{iothub}-{device_id}-{DD}-{MM}-{YYYY}"
     partition_template_default = "{deviceid}-{YYYY}-{MM}"
-    partition_path = "example"
+    partition_path = cosmosdb_obj["partitionKey"]
     # use connection string - no pkn or pkt
     cli.invoke(
         "iot hub message-endpoint create cosmosdb-container -n {} -g {} --en {} --erg {} -c {} --container {} "