Skip to content

Commit 505f7fe

Browse files
authored
[Event Hubs] Fix a bug that sets owner level 0 (#11172)
1 parent c62ccaa commit 505f7fe

File tree

2 files changed

+4
-5
lines changed

2 files changed

+4
-5
lines changed

sdk/eventhub/azure-eventhub/azure/eventhub/_eventprocessor/event_processor.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -74,9 +74,8 @@ def __init__(
7474
self._partition_close_handler = kwargs.get(
7575
"on_partition_close", None
7676
) # type: Optional[Callable[[PartitionContext, CloseReason], None]]
77-
self._checkpoint_store = kwargs.get(
78-
"checkpoint_store"
79-
) or InMemoryCheckpointStore() # type: Optional[CheckpointStore]
77+
checkpoint_store = kwargs.get("checkpoint_store") # type: Optional[CheckpointStore]
78+
self._checkpoint_store = checkpoint_store or InMemoryCheckpointStore()
8079
self._initial_event_position = kwargs.get(
8180
"initial_event_position", "@latest"
8281
) # type: Union[str, int, datetime, Dict[str, Any]]
@@ -93,7 +92,7 @@ def __init__(
9392

9493
# Receive parameters
9594
self._owner_level = kwargs.get("owner_level", None) # type: Optional[int]
96-
if "checkpoint_store" in kwargs and self._owner_level is None:
95+
if checkpoint_store and self._owner_level is None:
9796
self._owner_level = 0
9897
self._prefetch = kwargs.get("prefetch", None) # type: Optional[int]
9998
self._track_last_enqueued_event_properties = kwargs.get(

sdk/eventhub/azure-eventhub/tests/livetest/synctests/test_auth.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@
1414
def test_client_secret_credential(aad_credential, live_eventhub):
1515
try:
1616
from azure.identity import EnvironmentCredential
17-
except ImportError as e:
17+
except:
1818
pytest.skip("No azure identity library")
1919
credential = EnvironmentCredential()
2020
producer_client = EventHubProducerClient(fully_qualified_namespace=live_eventhub['hostname'],

0 commit comments

Comments
 (0)