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): 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() self.os_application_version_set.assert_called_with( utils.VERSION_PACKAGE)
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() self.os_application_version_set.assert_called_with( utils.VERSION_PACKAGE )
def test_assess_status(self, services, mock_is_paused): services.return_value = 's1' mock_is_paused.return_value = False 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', 's1') callee.assert_called_once_with() self.os_application_version_set.assert_called_with( utils.VERSION_PACKAGE)
def test_assess_status(self, services, mock_is_paused): services.return_value = 's1' mock_is_paused.return_value = False 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', 's1') callee.assert_called_once_with() self.os_application_version_set.assert_called_with( utils.VERSION_PACKAGE )
def test_assess_status_paused(self, services, mock_is_paused, mock_os_release): services.return_value = ['qemu-kvm', 'libvirtd', 'nova-compute'] mock_is_paused.return_value = True mock_os_release.return_value = 'pike' 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', ['nova-compute', 'qemu-kvm']) callee.assert_called_once_with() self.os_application_version_set.assert_called_with( utils.VERSION_PACKAGE)
def test_assess_status_paused(self, services, mock_is_paused, mock_os_release): services.return_value = ['qemu-kvm', 'libvirtd', 'nova-compute'] mock_is_paused.return_value = True mock_os_release.return_value = 'pike' 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', ['nova-compute', 'qemu-kvm']) callee.assert_called_once_with() self.os_application_version_set.assert_called_with( utils.VERSION_PACKAGE )
def main(): try: hooks.execute(sys.argv) except UnregisteredHookError as e: log('Unknown hook {} - skipping.'.format(e)) assess_status(CONFIGS)
def update_status(): log('Updating status.') assess_status(CONFIGS)
'amqp-relation-departed') def relation_broken(): proxy = get_proxy() @restart_on_change(restart_map(), proxy.restart_service) def write_config(): CONFIGS.write_all() write_config() proxy.commit() @hooks.hook('upgrade-charm') def upgrade_charm(): for r_id in relation_ids('amqp'): amqp_joined(relation_id=r_id) @hooks.hook('update-status') def update_status(): log('Updating status.') assess_status(CONFIGS) if __name__ == '__main__': try: hooks.execute(sys.argv) except UnregisteredHookError as e: log('Unknown hook {} - skipping.'.format(e)) assess_status(CONFIGS)