fix: kraft & balancer test issues

imanenami · imanenami · commit ec037851d40a · 2026-01-09T15:41:29.000+04:00
diff --git a/tests/integration/test_balancer.py b/tests/integration/test_balancer.py
@@ -109,7 +109,7 @@ def test_relate_not_enough_brokers(self, juju: jubilant.Juju):
         juju.integrate(PRODUCER_APP, APP_NAME)
 
         juju.wait(
-            lambda status: all_active_idle(
+            lambda status: jubilant.all_agents_idle(
                 status, *list({APP_NAME, CONTROLLER_NAME, self.balancer_app})
             ),
             delay=3,
diff --git a/tests/integration/test_kraft.py b/tests/integration/test_kraft.py
@@ -242,7 +242,7 @@ def test_leader_change(self, juju: jubilant.Juju):
         )
 
         # test cluster stability by adding a new controller
-        juju.add_units(self.controller_app, num_units=1)
+        juju.add_unit(self.controller_app, num_units=1)
         time.sleep(60)
 
         # ensure unit is added to dynamic quorum

Original file line number	Diff line number	Diff line change
`@@ -242,7 +242,7 @@ def test_leader_change(self, juju: jubilant.Juju):`
`242`	`242`	`)`
`243`	`243`
`244`	`244`	`# test cluster stability by adding a new controller`
`245`		`- juju.add_units(self.controller_app, num_units=1)`
	`245`	`+ juju.add_unit(self.controller_app, num_units=1)`
`246`	`246`	`time.sleep(60)`
`247`	`247`
`248`	`248`	`# ensure unit is added to dynamic quorum`