class Test_CmdServer(unittest.TestCase):
def janitor(self, *args, **kwargs):
pass
def setUp(self):
self.cfg = Mock()
self.cfg.command_port = DATA_PORT
self.cp = CmdProcessor()
self.ds = CmdServer(('127.0.0.1', DATA_PORT), self.cp.processCmds, self.janitor)
def test_CmdServer(self):
rxcalled = []
def rx(*args, **kwargs):
rxcalled.append(True)
def rx2(val, *args, **kwargs):
rxcalled.append(True)
self.assertEquals(val, 123)
self.cp.setCmd('cmd', None, rx)
self.cp.setCmd('cmd2', int, rx2)
def tx():
try:
self.ds.start()
sock = create_connection(('127.0.0.1', DATA_PORT), timeout=2)
with closing(sock):
sock.sendall('cmd\ncmd2 123\n')
sleep(0)
finally:
self.ds.stop()
txg = spawn(tx)
wait()
self.assertTrue(rxcalled[0])
self.assertTrue(rxcalled[1])
def state_startup(self):
# start cmdserver; err if not cmd port
spawn(self.heartbeat_timer).link_exception(self.janitor)
self.cmdserver = CmdServer((LISTENING_ADDR, self.cfg.command_port),
self.cmdproc.processCmds, self.janitor)
self.cmdserver.start()
return self.state_unconfigured
def setUp(self):
self.cfg = Mock()
self.cfg.command_port = DATA_PORT
self.cp = CmdProcessor()
self.ds = CmdServer(('127.0.0.1', DATA_PORT), self.cp.processCmds, self.janitor)
class PortAgent(Greenlet):
def __init__(self, cfg, cmdproc):
super(PortAgent, self).__init__()
self.cfg = cfg
self.cmdproc = cmdproc
self.heartbeat_event = Event()
self.cmdproc.setCmd('get_state', None, self.get_state)
self.cmdproc.setCmd('ping', None, self.ping)
self.cmdproc.setCmd('shutdown', None, self.shutdown)
self.states = {
self.state_startup: 'STARTUP',
self.state_unconfigured: 'UNCONFIGURED',
self.state_configured: 'CONFIGURED',
self.state_connected: 'CONNECTED',
}
@property
def state(self):
return self._state
@state.setter
def state(self, v):
self._state = v
log.debug("Transitioning to state %s" % v)
def janitor(self, src):
log.debug("Janitor, cleanup aisle 12")
self.kill(src.exception)
def heartbeat_timer(self):
try:
while True:
self.cfg.heartbeatactive.wait()
sleep(self.cfg.heartbeat_interval)
self.heartbeat_event.set()
self.heartbeat_event.clear()
except Exception:
log.critical("heartbeat_timer terminating due to exception", exc_info=True)
raise
def _run(self):
try:
self.state = self.state_startup
while True:
self.state = self.state()
except Exception:
log.critical("PortAgent terminating due to exception", exc_info=True)
raise
finally:
log.debug("Killing orbpkt src")
try: self.orbpktsrc.kill()
except: pass
log.debug("Stopping dataserver")
try: self.dataserver.stop()
except: pass
log.debug("Stopping cmdserver")
try: self.cmdserver.stop()
except: pass
log.debug("PortAgent finally done")
def state_startup(self):
# start cmdserver; err if not cmd port
spawn(self.heartbeat_timer).link_exception(self.janitor)
self.cmdserver = CmdServer((LISTENING_ADDR, self.cfg.command_port),
self.cmdproc.processCmds, self.janitor)
self.cmdserver.start()
return self.state_unconfigured
def state_unconfigured(self):
self.cfg.configuredevent.wait()
return self.state_configured
def state_configured(self):
# spawn orbreapthr
# link to it? what if it dies?
self.orbpktsrc = OrbPktSrc(
srcname = self.cfg.antelope_orb_name,
select = self.cfg.antelope_orb_select,
reject = self.cfg.antelope_orb_reject,
after = self.cfg.antelope_orb_after,
transformation = transform
)
self.orbpktsrc.link_exception(self.janitor)
self.orbpktsrc.start()
# spawn data server
self.dataserver = DataServer(
(LISTENING_ADDR, self.cfg.data_port),
self.orbpktsrc.subscription,
self.heartbeat_event, self.janitor)
self.dataserver.start()
# spawn state_connected
return self.state_connected
def state_connected(self):
# on dataserver config update event
self.cfg.dataserverconfigupdate.wait()
self.orbpktsrc.kill()
self.dataserver.stop()
return self.state_configured
def get_state(self, val, sock):
statestr = self.states[self.state]
sock.sendall(makepacket(PacketType.PORT_AGENT_STATUS, ntp.now(), statestr))
def ping(self, val, sock):
msg = "pong. version: " + __version__
sock.sendall(makepacket(PacketType.PORT_AGENT_STATUS, ntp.now(), msg))
def shutdown(self, val, sock):
self.kill()