refactor: move password_events to events.py

canonical · Nov 7, 2024 · f57d1e6 · f57d1e6
1 parent bde92a5
commit f57d1e6
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 113 deletions.
diff --git a/src/events/broker.py b/src/events/broker.py
@@ -88,7 +88,8 @@ def __init__(self, charm) -> None:
                 **DEPENDENCIES  # pyright: ignore[reportArgumentType]
             ),
         )
-        self.password_action_events = ActionEvents(self)
+        self.action_events = ActionEvents(self)
+
         if not self.charm.state.runs_controller:
             self.zookeeper = ZooKeeperHandler(self)
 

diff --git a/src/events/password_actions.py b/src/events/password_actions.py
diff --git a/tests/unit/test_upgrade.py b/tests/unit/test_upgrade.py
@@ -111,7 +111,7 @@ def test_run_password_rotation_while_upgrading(harness: Harness[KafkaCharm], upg
         ),
         patch("managers.auth.AuthManager.add_user"),
     ):
-        harness.charm.broker.password_action_events._set_password_action(mock_event)
+        harness.charm.broker.action_events._set_password_action(mock_event)
 
     if not upgrade_stack:
         mock_event.set_results.assert_called()