[Event Hubs] Update URI used for consumer auth to include consumer group

sdk/eventhub/azure-eventhub/CHANGELOG.md

@@ -8,6 +8,8 @@
 
 ### Bugs Fixed
 
+- Fixed a bug where the correct URI was not being used for consumer authentication, causing issues when assigning roles at the consumer group level. ([#35337](https://github.com/Azure/azure-sdk-for-python/issues/35337))
+
 ### Other Changes
 
 ## 5.11.7 (2024-04-10)

sdk/eventhub/azure-eventhub/azure/eventhub/_client_base.py

@@ -329,7 +329,7 @@ def __init__(
         else:
             self._credential = credential  # type: ignore
         self._auto_reconnect = kwargs.get("auto_reconnect", True)
-        self._auth_uri = f"sb://{self._address.hostname}{self._address.path}"
+        self._auth_uri: str
         self._config = Configuration(
             amqp_transport=self._amqp_transport,
             hostname=self._address.hostname,

sdk/eventhub/azure-eventhub/azure/eventhub/_consumer_client.py

@@ -178,6 +178,7 @@ def __init__(
             network_tracing=network_tracing,
             **kwargs
         )
+        self._auth_uri = f"sb://{self._address.hostname}{self._address.path}/consumergroups/{self._consumer_group}"
         self._lock = threading.Lock()
         self._event_processors: Dict[Tuple[str, str], EventProcessor] = {}
 

sdk/eventhub/azure-eventhub/azure/eventhub/_producer_client.py

@@ -201,7 +201,7 @@ def __init__(
             network_tracing=kwargs.get("logging_enable"),
             **kwargs
         )
-
+        self._auth_uri = f"sb://{self._address.hostname}{self._address.path}"
         self._keep_alive = kwargs.get("keep_alive", None)
 
         self._producers: Dict[str, Optional[EventHubProducer]] = {

sdk/eventhub/azure-eventhub/azure/eventhub/aio/_consumer_client_async.py

@@ -191,6 +191,7 @@ def __init__(
             network_tracing=network_tracing,
             **kwargs,
         )
+        self._auth_uri = f"sb://{self._address.hostname}{self._address.path}/consumergroups/{self._consumer_group}"
         self._lock = asyncio.Lock(**self._internal_kwargs)
         self._event_processors: Dict[Tuple[str, str], EventProcessor] = {}
 

sdk/eventhub/azure-eventhub/azure/eventhub/aio/_producer_client_async.py

@@ -188,6 +188,7 @@ def __init__(
             network_tracing=kwargs.pop("logging_enable", False),
             **kwargs
         )
+        self._auth_uri = f"sb://{self._address.hostname}{self._address.path}"
         self._keep_alive = kwargs.get("keep_alive", None)
         self._producers: Dict[str, Optional[EventHubProducer]] = {
             ALL_PARTITIONS: self._create_producer()