def ensure_after_execute(self):
LOGGER.debug("ensure_after_execute")
try:
RabbitMqAgent.ensure_after_execute(self)
except Exception, ex:
LOGGER.error("Error on error calling parent ensure_after_execute")
raise MpdRmqException, MpdRmqException(ex), sys.exc_info()[2] # keep stacktrace
def test_rabbitmqagent_ensure_after_execute_call_super(self, method):
rmqagent = RabbitMqAgent({},{})
rmq = Mock()
rmqagent.rmqclient = rmq
rmqagent.ensure_after_execute()
method.assert_called_once_with(rmqagent)
def test_rabbitmqagent_ensure_after_execute(self):
rmqagent = RabbitMqAgent({},{})
rmq = Mock()
rmqagent.rmqclient = rmq
rmqagent.ensure_after_execute()
rmq.disconnect.assert_called_once_with()
def test_rabbitmqagent_basicpublish(self):
import json
conf={"channel":{"exchange":"ex","type":"type"},"messages":{"routing_key":"rt"}}
rmqagent = RabbitMqAgent(conf, {})
rmq = Mock()
rmqagent.rmqclient = rmq
obj={"aa":1, "bb":{"cc":2}}
rmqagent.basic_publish_json(obj)
expected = {}
expected["id"] = rmqagent.id
expected["payload"] = obj
message=json.dumps(expected)
rmq.basic_publish_json.assert_called_once_with(conf["channel"]["exchange"], conf["channel"]["type"], conf["messages"]["routing_key"], expected)
def after_execute(self):
LOGGER.debug("after_execute")
RabbitMqAgent.after_execute(self)
def before_execute(self):
LOGGER.debug("before_execute")
self.mpdclient.connect()
RabbitMqAgent.before_execute(self)
def __init__(self, config={}, mpdconf={}, rmqconf={}):
RabbitMqAgent.__init__(self, config, rmqconf)
self.mpdclient = MpdClient(mpdconf)
def ensure_after_execute(self):
try:
RabbitMqAgent.ensure_after_execute(self)
except Exception, ex:
LOGGER.error("Error calling RabbitMqAgent.ensure_after_execute")
raise TelnetRmqAgenException, TelnetRmqAgenException(ex), sys.exc_info()[2] # keep stacktrace
def after_execute(self): RabbitMqAgent.after_execute(self)
def before_execute(self): RabbitMqAgent.before_execute(self) self.telnetclient.connect()
def __init__(self, config, telnetconf={}, rmqconf={}):
RabbitMqAgent.__init__(self, config, rmqconf)
self.telnetclient = TelnetClient(telnetconf)
def __init__(self, config, xbmcconf={}, rmqconf={}):
RabbitMqAgent.__init__(self,config,rmqconf)
self.xbmc=XbmcClient(xbmcconf)
def before_execute(self): RabbitMqAgent.before_execute(self)