particle-os/debian-koji
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

test mode for messagebus plugin

Browse source
This commit is contained in:
Mike McLean 2017-08-03 17:02:57 -04:00
parent 2a6c04635f
commit 78162b5543
1 changed files with 11 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

15
plugins/hub/messagebus.py
View file

@ -248,14 +248,21 @@ def prep_message(cbtype, *args, **kws):
def send_messages(cbtype, *args, **kws):
'''Send the messages cached by prep_message'''
logger = logging.getLogger('koji.plugin.messagebus')
config = get_config()
messages = getattr(context, 'messagebus_plugin_messages', [])
if not messages:
return
sender = get_sender()
for message in messages:
sender.send(message, sync=False, timeout=config.getfloat('broker', 'timeout'))
sender.close(timeout=config.getfloat('broker', 'timeout'))
if config.getboolean('broker', 'test_mode'):
logger.debug('test mode: skipping broker connection')
for message in messages:
logger.debug('test mode: skipping message: %r', message)
else:
sender = get_sender()
for message in messages:
sender.send(message, sync=False,
timeout=config.getfloat('broker', 'timeout'))
sender.close(timeout=config.getfloat('broker', 'timeout'))
# koji should do this for us, but just in case...
del context.messagebus_plugin_messages