def configure_nodename(): '''Set RABBITMQ_NODENAME to something that's resolvable by my peers''' nodename = rabbit.get_local_nodename() log('configuring nodename', level=INFO) if (nodename and rabbit.get_node_name() != 'rabbit@%s' % nodename): log('forcing nodename=%s' % nodename, level=INFO) # would like to have used the restart_on_change decorator, but # need to stop it under current nodename prior to updating env log('Stopping rabbitmq-server.') service_stop('rabbitmq-server') rabbit.update_rmq_env_conf(hostname='rabbit@%s' % nodename, ipv6=config('prefer-ipv6')) log('Starting rabbitmq-server.') service_restart('rabbitmq-server') rabbit.wait_app()
def update_cookie(leaders_cookie=None): # sync cookie if leaders_cookie: cookie = leaders_cookie else: cookie = peer_retrieve('cookie') cookie_local = None with open(rabbit.COOKIE_PATH, 'r') as f: cookie_local = f.read().strip() if cookie_local == cookie: log('Cookie already synchronized with peer.') return service_stop('rabbitmq-server') with open(rabbit.COOKIE_PATH, 'wb') as out: out.write(cookie) service_restart('rabbitmq-server') rabbit.wait_app()
def update_cookie(leaders_cookie=None): # sync cookie if leaders_cookie: cookie = leaders_cookie else: cookie = peer_retrieve('cookie') cookie_local = None with open(rabbit.COOKIE_PATH, 'r') as f: cookie_local = f.read().strip() if cookie_local == cookie: log('Cookie already synchronized with peer.') return service_stop('rabbitmq-server') with open(rabbit.COOKIE_PATH, 'wb') as out: out.write(cookie) if not is_unit_paused_set(): service_restart('rabbitmq-server') rabbit.wait_app()
def update_cookie(leaders_cookie=None): # sync cookie if leaders_cookie: cookie = leaders_cookie else: cookie = peer_retrieve('cookie') cookie_local = None with open(rabbit.COOKIE_PATH, 'r') as f: cookie_local = f.read().strip() if cookie_local == cookie: log('Cookie already synchronized with peer.') return elif not is_restart_permitted(): raise Exception("rabbitmq-server must be restarted but not permitted") service_stop('rabbitmq-server') with open(rabbit.COOKIE_PATH, 'wb') as out: out.write(cookie.encode('ascii')) if not is_unit_paused_set(): service_restart('rabbitmq-server') rabbit.wait_app()
def test_rabbitmqctl_wait_success(self, check_call): check_call.return_value = 0 self.assertTrue(rabbit_utils.wait_app())
def test_rabbitmqctl_wait_fail(self, check_call): check_call.side_effect = (rabbit_utils.subprocess.CalledProcessError( 1, 'cmd')) with self.assertRaises(rabbit_utils.subprocess.CalledProcessError): rabbit_utils.wait_app()
def test_rabbitmqctl_wait_fail(self, check_call): check_call.side_effect = (rabbit_utils.subprocess. CalledProcessError(1, 'cmd')) with self.assertRaises(rabbit_utils.subprocess.CalledProcessError): rabbit_utils.wait_app()
def test_rabbitmqctl_wait_success(self, check_call, local_nodename): check_call.return_value = 0 local_nodename.return_value = 'rabbitmq-server-0' self.assertTrue(rabbit_utils.wait_app())
def test_rabbitmqctl_wait_fail(self, check_call, local_nodename): check_call.side_effect = (rabbit_utils.subprocess.CalledProcessError( 1, 'cmd')) local_nodename.return_value = 'rabbitmq-server-0' with self.assertRaises(rabbit_utils.subprocess.CalledProcessError): rabbit_utils.wait_app()