1717import os
1818
1919import zaza .model as zaza_model
20+ import zaza .charm_lifecycle .utils as lifecycle_utils
2021import zaza .openstack .charm_tests .test_utils as test_utils
2122
2223
@@ -37,11 +38,13 @@ def setUp(self):
3738 def test_configure_ips (self ):
3839 """Configure and un-configure 'service_ips' option."""
3940 config = {"service_ips" : self .VIP }
41+ test_config = lifecycle_utils .get_charm_config (fatal = False )
42+ states = test_config .get ("target_deploy_status" , {})
4043
4144 logging .info ("Configuring %s as a Service IP for %s unit." ,
4245 self .VIP , self .UNIT )
4346 zaza_model .set_application_config (self .APPLICATION , config )
44- zaza_model .wait_for_application_states ()
47+ zaza_model .wait_for_application_states (states = states )
4548
4649 configured_ips = zaza_model .run_on_unit (self .UNIT , "ip addr" )
4750 self .assertIn (self .VIP , configured_ips ["Stdout" ])
@@ -50,7 +53,7 @@ def test_configure_ips(self):
5053 self .UNIT )
5154 config ["service_ips" ] = ""
5255 zaza_model .set_application_config (self .APPLICATION , config )
53- zaza_model .wait_for_application_states ()
56+ zaza_model .wait_for_application_states (states = states )
5457
5558 configured_ips = zaza_model .run_on_unit (self .UNIT , "ip addr" )
5659 self .assertNotIn (self .VIP , configured_ips ["Stdout" ])
0 commit comments