def rabbit_config(request, rabbit_manager):
from kombu import pools
from nameko.testing.utils import (
reset_rabbit_vhost,
reset_rabbit_connections,
get_rabbit_connections,
get_rabbit_config,
)
amqp_uri = request.config.getoption("AMQP_URI")
conf = get_rabbit_config(amqp_uri)
reset_rabbit_connections(conf["vhost"], rabbit_manager)
reset_rabbit_vhost(conf["vhost"], conf["username"], rabbit_manager)
yield conf
pools.reset() # close connections in pools
# raise a runtime error if the test leaves any connections lying around
connections = get_rabbit_connections(conf["vhost"], rabbit_manager)
open_connections = [conn for conn in connections if conn["state"] != "closed"]
if open_connections:
count = len(open_connections)
raise RuntimeError("{} rabbit connection(s) left open.".format(count))
def rabbit_config(request, rabbit_manager):
from kombu import pools
from nameko.testing.utils import (reset_rabbit_vhost,
reset_rabbit_connections,
get_rabbit_connections,
get_rabbit_config)
amqp_uri = request.config.getoption('AMQP_URI')
conf = get_rabbit_config(amqp_uri)
reset_rabbit_connections(conf['vhost'], rabbit_manager)
reset_rabbit_vhost(conf['vhost'], conf['username'], rabbit_manager)
yield conf
pools.reset() # close connections in pools
# raise a runtime error if the test leaves any connections lying around
connections = get_rabbit_connections(conf['vhost'], rabbit_manager)
open_connections = [
conn for conn in connections if conn['state'] != "closed"
]
if open_connections:
count = len(open_connections)
raise RuntimeError("{} rabbit connection(s) left open.".format(count))
def test_reset_rabbit_vhost(rabbit_config, rabbit_manager):
vhost = rabbit_config['vhost']
username = rabbit_config['username']
def get_active_vhosts():
return [vhost_data['name'] for
vhost_data in rabbit_manager.get_all_vhosts()]
reset_rabbit_vhost(vhost, username, rabbit_manager)
assert vhost in get_active_vhosts()
rabbit_manager.delete_vhost(vhost)
assert vhost not in get_active_vhosts()
reset_rabbit_vhost(vhost, username, rabbit_manager)
assert vhost in get_active_vhosts()
def rabbit_config(request, rabbit_manager):
amqp_uri = request.config.getoption('AMQP_URI')
conf = get_rabbit_config(amqp_uri)
reset_rabbit_connections(conf['vhost'], rabbit_manager)
reset_rabbit_vhost(conf['vhost'], conf['username'], rabbit_manager)
yield conf
pools.reset() # close connections in pools
# raise a runtime error if the test leaves any connections lying around
connections = get_rabbit_connections(conf['vhost'], rabbit_manager)
if connections:
count = len(connections)
raise RuntimeError("{} rabbit connection(s) left open.".format(count))
def rabbit_config(request, rabbit_manager):
amqp_uri = request.config.getoption('AMQP_URI')
conf = get_rabbit_config(amqp_uri)
reset_rabbit_connections(conf['vhost'], rabbit_manager)
reset_rabbit_vhost(conf['vhost'], conf['username'], rabbit_manager)
yield conf
pools.reset() # close connections in pools
# raise a runtime error if the test leaves any connections lying around
connections = get_rabbit_connections(conf['vhost'], rabbit_manager)
if connections:
count = len(connections)
raise RuntimeError("{} rabbit connection(s) left open.".format(count))
def test_reset_rabbit_vhost_errors():
rabbit_manager = Mock()
# 500 error
error_500 = HTTPError({'reason': 'error'}, status=500)
rabbit_manager.delete_vhost.side_effect = error_500
with pytest.raises(HTTPError):
reset_rabbit_vhost("vhost", "username", rabbit_manager)
# 404 error
error_404 = HTTPError({'reason': 'error'}, status=404)
rabbit_manager.delete_vhost.side_effect = error_404
# does not raise
reset_rabbit_vhost("vhost", "username", rabbit_manager)
assert rabbit_manager.create_vhost.call_args == call("vhost")
assert rabbit_manager.set_vhost_permissions.call_args == call(
"vhost", "username", '.*', '.*', '.*')
def test_reset_rabbit_vhost_errors():
rabbit_manager = Mock()
# 500 error
error_500 = HTTPError({'reason': 'error'}, status=500)
rabbit_manager.delete_vhost.side_effect = error_500
with pytest.raises(HTTPError):
reset_rabbit_vhost("vhost", "username", rabbit_manager)
# 404 error
error_404 = HTTPError({'reason': 'error'}, status=404)
rabbit_manager.delete_vhost.side_effect = error_404
# does not raise
reset_rabbit_vhost("vhost", "username", rabbit_manager)
assert rabbit_manager.create_vhost.call_args == call("vhost")
assert rabbit_manager.set_vhost_permissions.call_args == call(
"vhost", "username", '.*', '.*', '.*')
def test_reset_rabbit_vhost_errors():
rabbit_manager = Mock()
# 500 error
response_500 = Response()
response_500.status_code = 50
error_500 = HTTPError(response=response_500)
rabbit_manager.delete_vhost.side_effect = error_500
with pytest.raises(HTTPError):
reset_rabbit_vhost("vhost", "username", rabbit_manager)
# 404 error
response_404 = Response()
response_404.status_code = 404
error_404 = HTTPError(response=response_404)
rabbit_manager.delete_vhost.side_effect = error_404
# does not raise
reset_rabbit_vhost("vhost", "username", rabbit_manager)
assert rabbit_manager.create_vhost.call_args == call("vhost")
assert rabbit_manager.set_vhost_permissions.call_args == call(
"vhost", "username", '.*', '.*', '.*')
def rabbit_config(request, rabbit_manager):
amqp_uri = request.config.getoption('AMQP_URI')
conf = {'AMQP_URI': amqp_uri}
uri = urlparse(amqp_uri)
vhost = uri.path[1:].replace('/', '%2F')
username = uri.username
conf['vhost'] = vhost
conf['username'] = username
reset_rabbit_connections(vhost, rabbit_manager)
reset_rabbit_vhost(vhost, username, rabbit_manager)
yield conf
pools.reset() # close connections in pools
# raise a runtime error if the test leaves any connections lying around
connections = get_rabbit_connections(vhost, rabbit_manager)
if connections:
count = len(connections)
raise RuntimeError("{} rabbit connection(s) left open.".format(count))
def rabbit_config(request, rabbit_manager):
amqp_uri = request.config.getoption('AMQP_URI')
conf = {'AMQP_URI': amqp_uri}
uri = urlparse(amqp_uri)
vhost = uri.path[1:].replace('/', '%2F')
username = uri.username
conf['vhost'] = vhost
conf['username'] = username
reset_rabbit_connections(vhost, rabbit_manager)
reset_rabbit_vhost(vhost, username, rabbit_manager)
yield conf
pools.reset() # close connections in pools
# raise a runtime error if the test leaves any connections lying around
connections = get_rabbit_connections(vhost, rabbit_manager)
if connections:
count = len(connections)
raise RuntimeError("{} rabbit connection(s) left open.".format(count))