Typo

canonical · May 20, 2024 · ff5fa50 · ff5fa50
1 parent 52c868c
commit ff5fa50
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/src/charm.py b/src/charm.py
@@ -1468,7 +1468,7 @@ def _restart(self, event: RunWithLock) -> None:
             return
 
         try:
-            for attempt in Retrying(wait=wait_fixed(3), stop_after_delay=stop_after_delay(300)):
+            for attempt in Retrying(wait=wait_fixed(3), stop=stop_after_delay(300)):
                 with attempt:
                     if not self._can_connect_to_postgresql:
                         assert False

diff --git a/tests/unit/test_charm.py b/tests/unit/test_charm.py
@@ -1135,6 +1135,7 @@ def test_restart(harness):
     with (
         patch("charm.Patroni.restart_postgresql") as _restart_postgresql,
         patch("charm.Patroni.are_all_members_ready") as _are_all_members_ready,
+        patch("charm.PostgresqlOperatorCharm._can_connect_to_postgresql", return_value=True),
     ):
         _are_all_members_ready.side_effect = [False, True, True]