WIP: Maybe fix integration tests

canonical · Oct 21, 2024 · d0e4f54 · d0e4f54
1 parent 304b4f3
commit d0e4f54
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/core/models.py b/src/core/models.py
@@ -17,6 +17,7 @@
 )
 from charms.zookeeper.v0.client import QuorumLeaderNotFoundError, ZooKeeperManager
 from kazoo.client import AuthFailedError, NoNodeError
+from kazoo.exceptions import NoAuthError
 from lightkube.resources.core_v1 import Node, Pod
 from ops.model import Application, Relation, Unit
 from tenacity import retry, retry_if_result, stop_after_attempt, wait_fixed
@@ -713,7 +714,7 @@ def broker_active(self) -> bool:
         zk = ZooKeeperManager(hosts=hosts, username=self.username, password=self.password)
         try:
             brokers = zk.leader_znodes(path=path)
-        except (NoNodeError, AuthFailedError, QuorumLeaderNotFoundError) as e:
+        except (NoNodeError, AuthFailedError, QuorumLeaderNotFoundError, NoAuthError) as e:
             logger.debug(str(e))
             brokers = set()