[Event Hubs] Add security token to management request message applica…

…tion properties (#9931) * Increment version * Update Development Status * Add security_token to management request message's application_properties * Fix an async call issue * use auth.token instead of get_token()
AzureSDKAutomation · Feb 24, 2020 · 690151b · 690151b
1 parent af4ffee
commit 690151b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/sdk/eventhub/azure-eventhub/azure/eventhub/_client_base.py b/sdk/eventhub/azure-eventhub/azure/eventhub/_client_base.py
@@ -228,6 +228,7 @@ def _management_request(self, mgmt_msg, op_type):
                     self._address.hostname, mgmt_auth
                 )  # pylint:disable=assignment-from-none
                 mgmt_client.open(connection=conn)
+                mgmt_msg.application_properties["security_token"] = mgmt_auth.token
                 response = mgmt_client.mgmt_request(
                     mgmt_msg,
                     constants.READ_OPERATION,

diff --git a/sdk/eventhub/azure-eventhub/azure/eventhub/aio/_client_base_async.py b/sdk/eventhub/azure-eventhub/azure/eventhub/aio/_client_base_async.py
@@ -166,6 +166,7 @@ async def _management_request_async(self, mgmt_msg: Message, op_type: bytes) ->
                 conn = await self._conn_manager_async.get_connection(
                     self._address.hostname, mgmt_auth
                 )
+                mgmt_msg.application_properties["security_token"] = mgmt_auth.token
                 await mgmt_client.open_async(connection=conn)
                 response = await mgmt_client.mgmt_request_async(
                     mgmt_msg,