def get_redis(self):
'''Creates and returns a redis manager'''
if hasattr(self, 'redis'):
returnValue(self.redis)
persistencehelper = PersistenceHelper()
yield persistencehelper.setup()
self.redis = yield persistencehelper.get_redis_manager()
self.addCleanup(persistencehelper.cleanup)
returnValue(self.redis)
def get_redis(self):
'''Creates and returns a redis manager'''
if hasattr(self, 'redis'):
returnValue(self.redis)
persistencehelper = PersistenceHelper()
yield persistencehelper.setup()
self.redis = yield persistencehelper.get_redis_manager()
self.addCleanup(persistencehelper.cleanup)
returnValue(self.redis)
class JunebugTestBase(TestCase):
'''Base test case that all junebug tests inherit from. Contains useful
helper functions'''
default_channel_properties = {
'type': 'telnet',
'config': {
'twisted_endpoint': 'tcp:0',
'worker_name': 'unnamed',
},
'mo_url': 'http://foo.bar',
}
default_channel_config = {
'ttl': 60,
'amqp': {},
}
def patch_logger(self):
''' Patches the logger with an in-memory logger, which is acccessable
at "self.logging_handler".'''
self.logging_handler = logging.handlers.MemoryHandler(100)
logging.getLogger().addHandler(self.logging_handler)
self.addCleanup(self._cleanup_logging_patch)
def _cleanup_logging_patch(self):
self.logging_handler.close()
logging.getLogger().removeHandler(self.logging_handler)
def create_channel_properties(self, **kw):
properties = deepcopy(self.default_channel_properties)
properties.update(kw)
return properties
@inlineCallbacks
def create_channel_config(self, **kw):
self.persistencehelper = PersistenceHelper()
yield self.persistencehelper.setup()
self.addCleanup(self.persistencehelper.cleanup)
config = deepcopy(self.default_channel_config)
config.update(kw)
channel_config = self.persistencehelper.mk_config(config)
channel_config['redis'] = channel_config['redis_manager']
returnValue(JunebugConfig(channel_config))
@inlineCallbacks
def create_channel(
self, service, redis, transport_class,
properties=default_channel_properties, id=None):
'''Creates and starts, and saves a channel, with a
TelnetServerTransport transport'''
properties = deepcopy(properties)
config = yield self.create_channel_config()
channel = Channel(
redis, config, properties, id=id)
properties['config']['transport_name'] = channel.id
yield channel.start(self.service)
yield channel.save()
self.addCleanup(channel.stop)
returnValue(channel)
@inlineCallbacks
def create_channel_from_id(self, redis, config, id, service):
'''Creates an existing channel given the channel id'''
config = yield self.create_channel_config(**config)
channel = yield Channel.from_id(redis, config, id, service)
returnValue(channel)
@inlineCallbacks
def get_redis(self):
'''Creates and returns a redis manager'''
if hasattr(self, 'redis'):
returnValue(self.redis)
persistencehelper = PersistenceHelper()
yield persistencehelper.setup()
self.redis = yield persistencehelper.get_redis_manager()
self.addCleanup(persistencehelper.cleanup)
returnValue(self.redis)
@inlineCallbacks
def start_server(self):
'''Starts a junebug server. Stores the service to "self.service", and
the url at "self.url"'''
config = yield self.create_channel_config()
self.service = JunebugService(config)
self.api = JunebugApi(
self.service, config)
redis = yield self.persistencehelper.get_redis_manager()
self.api = JunebugApi(self.service, config)
yield self.api.setup(redis, self.get_message_sender())
self.config = self.api.config
self.redis = self.api.redis
self.inbounds = self.api.inbounds
self.outbounds = self.api.outbounds
self.message_sender = self.api.message_sender
port = reactor.listenTCP(
0, Site(self.api.app.resource()),
interface='127.0.0.1')
self.addCleanup(port.stopListening)
addr = port.getHost()
self.url = "http://%s:%s" % (addr.host, addr.port)
def get_message_sender(self):
'''Creates a new MessageSender object, with a fake amqp client'''
message_sender = MessageSender('amqp-spec-0-8.xml', None)
spec = get_spec(vumi_resource_path('amqp-spec-0-8.xml'))
client = FakeAmqpClient(spec)
message_sender.client = client
return message_sender
def get_dispatched_messages(self, queue):
'''Gets all messages that have been dispatched to the amqp broker.
Should only be called after start_server, as it looks in the api for
the amqp client'''
amqp_client = self.api.message_sender.client
return amqp_client.broker.get_messages(
'vumi', queue)
