def test_get_rabbitmq_location(self):
flexmock(file_io) \
.should_receive("read").and_return("192.168.0.1")
flexmock(file_io) \
.should_receive("write").and_return(None)
expected = "amqp://*****:*****@192.168.0.1:5672//"
self.assertEquals(rabbitmq.get_connection_string(), expected)
def __broker_location(self, broker):
""" Gets the broker location connection string.
Args:
broker: The broker enum value.
Returns:
A broker connection string.
Raises:
NotImplementedError: If the broker is not implemented
"""
if broker == self.RABBITMQ:
from brokers import rabbitmq
return rabbitmq.get_connection_string()
else:
raise NotImplementedError(
"The given broker of code %d is not implemented" % broker)
def __broker_location(self, broker):
""" Gets the broker location connection string.
Args:
broker: The broker enum value.
Returns:
A broker connection string.
Raises:
NotImplementedError: If the broker is not implemented
"""
if broker == self.RABBITMQ:
from brokers import rabbitmq
return rabbitmq.get_connection_string()
else:
raise NotImplementedError(
"The given broker of code %d is not implemented" % broker)
