def test_telnetclient_connect(self, mocked):
""" Test connect() method """
telnetclientconf={"host":"host","port":20000}
telnetclient = TelnetClient(telnetclientconf)
telnetclient.connect()
mocked.assert_called_with()
telnetclient.client.open.assert_called_with(telnetclientconf["host"], telnetclientconf["port"])
def test_telnetclient_disconnect(self, mocked):
""" Test disconnect() method """
telnetclientconf={"host":"host","port":20000}
telnetclient = TelnetClient(telnetclientconf)
# Call
telnetclient.connect()
telnetclient.disconnect()
telnetclient.client.close.assert_called_with()
def test_telnetclient_disconnect_exception(self, mocked):
""" Test disconnect() method with exception raised """
mocked.side_effect=Exception("In your face !")
telnetclientconf={"host":"host","port":20000}
telnetclient = TelnetClient(telnetclientconf)
with self.assertRaises(TelnetClientException):
telnetclient.disconnect()
class TelnetRmqAgent(RabbitMqAgent):
"""
Telnet/RabbitMq client handler
"""
def __init__(self, config, telnetconf={}, rmqconf={}):
RabbitMqAgent.__init__(self, config, rmqconf)
self.telnetclient = TelnetClient(telnetconf)
def before_execute(self):
RabbitMqAgent.before_execute(self)
self.telnetclient.connect()
def after_execute(self):
RabbitMqAgent.after_execute(self)
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
finally:
def __init__(self, config, telnetconf={}, rmqconf={}):
RabbitMqAgent.__init__(self, config, rmqconf)
self.telnetclient = TelnetClient(telnetconf)
