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)