Skip to content

Commit 8326a20

Browse files
authored
[SB] Fix the Scope In SB Management Client (#36545)
* use the right scope for token cred * add sastoken cred
1 parent 071ae03 commit 8326a20

File tree

2 files changed

+8
-0
lines changed

2 files changed

+8
-0
lines changed

sdk/servicebus/azure-servicebus/azure/servicebus/aio/management/_management_client_async.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -231,6 +231,10 @@ async def _create_forward_to_header_tokens(self, entity, kwargs):
231231
kwargs["headers"] = kwargs.get("headers", {})
232232

233233
async def _populate_header_within_kwargs(uri, header):
234+
if not isinstance(
235+
self._credential,
236+
(ServiceBusSASTokenCredential, ServiceBusSharedKeyCredential)):
237+
uri = JWT_TOKEN_SCOPE
234238
token = (await self._credential.get_token(uri)).token
235239
if not isinstance(
236240
self._credential,

sdk/servicebus/azure-servicebus/azure/servicebus/management/_management_client.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -229,6 +229,10 @@ def _create_forward_to_header_tokens(self, entity, kwargs):
229229
kwargs["headers"] = kwargs.get("headers", {})
230230

231231
def _populate_header_within_kwargs(uri, header):
232+
if not isinstance(
233+
self._credential,
234+
(ServiceBusSASTokenCredential, ServiceBusSharedKeyCredential)):
235+
uri = JWT_TOKEN_SCOPE
232236
token = self._credential.get_token(uri).token
233237
if not isinstance(
234238
self._credential,

0 commit comments

Comments
 (0)