2424)
2525
2626import zaza .model as zaza_model
27+ import zaza .charm_lifecycle .utils as lifecycle_utils
2728import zaza .openstack .charm_tests .test_utils as test_utils
2829
2930
@@ -44,11 +45,13 @@ def setUp(self):
4445 def test_configure_ips (self ):
4546 """Configure and un-configure 'service_ips' option."""
4647 config = {"service_ips" : self .VIP }
48+ test_config = lifecycle_utils .get_charm_config (fatal = False )
49+ states = test_config .get ("target_deploy_status" , {})
4750
4851 logging .info ("Configuring %s as a Service IP for %s unit." ,
4952 self .VIP , self .UNIT )
5053 zaza_model .set_application_config (self .APPLICATION , config )
51- zaza_model .wait_for_application_states ()
54+ zaza_model .wait_for_application_states (states = states )
5255
5356 for attempt in Retrying (wait = wait_fixed (2 ),
5457 retry = retry_if_exception_type (AssertionError ),
@@ -62,7 +65,7 @@ def test_configure_ips(self):
6265 self .UNIT )
6366 config ["service_ips" ] = ""
6467 zaza_model .set_application_config (self .APPLICATION , config )
65- zaza_model .wait_for_application_states ()
68+ zaza_model .wait_for_application_states (states = states )
6669
6770 for attempt in Retrying (wait = wait_fixed (2 ),
6871 retry = retry_if_exception_type (AssertionError ),
0 commit comments