def test_assess_status(self): with patch.object(utils, 'assess_status_func') as asf: callee = MagicMock() asf.return_value = callee utils.assess_status('test-config') asf.assert_called_once_with('test-config') callee.assert_called_once_with()
def test_assess_status(self, status_set, is_paused): """test_assess_status: verify that it does pick the right status""" # check that paused status does the right thing is_paused.return_value = True utils.assess_status(None) status_set.assert_called_with( "maintenance", "Paused. Use 'resume' action to resume normal service.") # if it isn't paused, the assess_status() calls # set_os_workload_status() is_paused.return_value = False with patch.object(utils, 'set_os_workload_status') \ as set_os_workload_status: utils.assess_status("TEST CONFIG") set_os_workload_status.assert_called_with( "TEST CONFIG", utils.REQUIRED_INTERFACES, charm_func=utils.check_optional_relations, services=['haproxy', 'keystone', 'apache2'], ports=[5000, 35357])
def main(): try: hooks.execute(sys.argv) except UnregisteredHookError as e: log('Unknown hook {} - skipping.'.format(e)) assess_status(CONFIGS)