def notest_ProtocolEval():
host = "localhost"
mq = MQStar(host)
mq.clean()
c = "client1"
mq.add_client(c)
commands = [
"BEGIN", ("EVAL_SERVER", "dir()"), ("EVAL_SERVER", "locals()"),
("EVAL_SERVER", "__import__('os').getcwd()"), ("END", None, None)
]
procedure = Procedure("PROC", commands)
p = Protocol(mq, c, procedure)
while p.send_next_command():
logging.debug("sent command")
exit = False
while not exit:
rec = mq.receive_server(blocking=True, timeout=10)
if rec is not None:
logging.debug("- SERVER RECEIVED %s %s" % (rec, type(rec)))
c, msg = rec
answer = p.receive_answer(c, msg)
logging.debug("- SERVER RECEIVED ANSWER: ", answer.success)
if answer.name == "END" or not answer.success:
logging.debug("- SERVER RECEIVE END")
#if answer.success:
a = """('client1', ('EVAL_SERVER', True, {'self': <Command_EVAL_SERVER.Command_EVAL_SERVER object at 0x10931f810>, 'args': 'locals()'}))""" # p.send_next_command()
else:
logging.debug("- SERVER RECEIVED empty")
exit = True
def test_ProtocolCall():
host = "localhost"
mq = MQStar(host)
mq.clean()
c = "client1"
mq.add_client(c)
yaml = """
HELLO:
- EVAL_SERVER: 3+2
BASIC:
- EVAL_SERVER: dir()
- CALL: HELLO
CALLER:
- BEGIN
- CALL: BASIC
- EVAL_SERVER: locals()
- END
"""
procedures = Procedure.load_from_yaml(yaml)
caller = Procedure.procedures["CALLER"]
basic = Procedure.procedures["BASIC"]
class D:
pass
d = D()
d.mq = mq
p = Protocol(d, c, caller)
while p.send_next_command():
logging.debug("sent command")
exit = False
answers =0
while not exit:
rec = mq.receive_server(blocking=True, timeout=10)
if rec is not None:
logging.debug("- SERVER RECEIVED %s %s" % (rec, type(rec)))
c, msg = rec
command_unserialize = command.unserialize(msg)
answer = p.receive_answer(c, command_unserialize)
logging.debug("- SERVER RECEIVED ANSWER: %s" % answer.success)
if answer.success:
answers += 1
if answer.name == "END" or not answer.success:
logging.debug("- SERVER RECEIVE END")
#if answer.success:
exit = True
else:
logging.debug("- SERVER RECEIVED empty")
exit = True
assert answers == 7, "wrong answers: %s" % answers
def test_ProtocolEval():
host = "localhost"
mq = MQStar(host)
mq.clean()
c = "client1"
mq.add_client(c)
commands = ["BEGIN", ("EVAL_SERVER", None, "dir()"),
("EVAL_SERVER", None, "locals()"),
("EVAL_SERVER", None, "__import__('os').getcwd()"),
("END", None, None)]
procedure = Procedure("PROC", commands)
class D:
pass
d = D()
d.mq = mq
p = Protocol(d, c, procedure)
while p.send_next_command():
logging.debug("sent command: %s" % p.last_command)
print("---- START RECEIVING ----")
exit = False
while not exit:
rec = mq.receive_server(blocking=True, timeout=10)
if rec is not None:
logging.debug("- SERVER RECEIVED %s %s" % (rec, type(rec)))
c, msg = rec
command_unserialize = command.unserialize(msg)
answer = p.receive_answer(c, command_unserialize)
logging.debug("- SERVER RECEIVED ANSWER: %s" % answer.success)
if answer.name == "END" or not answer.success:
logging.debug("- SERVER RECEIVE END")
#if answer.success:
a = """('client1', ('EVAL_SERVER', True, {'self': <Command_EVAL_SERVER.Command_EVAL_SERVER object at 0x10931f810>, 'args': 'locals()'}))"""# p.send_next_command()
else:
logging.debug("- SERVER RECEIVED empty")
exit = True
print("---- STOP RECEIVING ----")
def test_blockingMQ():
global received
host = "localhost"
mq1 = MQStar(host)
mq2 = MQStar(host, session=mq1.session)
c = "client1"
mq1.add_client(c)
thread1 = threading.Thread(target=server, args=(mq1, ))
thread1.start()
mq2.send_server(c, "WORKS")
mq2.send_server(c, "FINE TO THE")
time.sleep(1)
mq2.send_server(c, "STOP")
time.sleep(6)
print "RECEIVED: ", received
assert len(received) == 3, "len: %s" % len(received)
def test_blockingMQ():
global received
host = "localhost"
mq1 = MQStar(host)
mq2 = MQStar(host, session=mq1.session)
c = "client1"
mq1.add_client(c)
thread1 = threading.Thread(target=server, args=(mq1,))
thread1.start()
mq2.send_server(c, "WORKS")
mq2.send_server(c, "FINE TO THE")
time.sleep(1)
mq2.send_server(c, "STOP")
time.sleep(6)
print "RECEIVED: ", received
assert len(received) == 3, "len: %s" % len(received)
def start_agent(self, mq=None, procedure=None, force=False):
if not force and check_running(self.vm):
logging.fatal("already running")
exit = True
return False
class D:
pass
d = D()
if not mq:
mq = MQStar(self.host, self.session)
d.mq = mq
self.pc = Protocol(d, self.vm)
else:
assert procedure
d.mq = mq
self.pc = Protocol(d, self.vm, procedure=procedure)
mq.protocol = self.pc
logging.debug("mq: %s pc:%s" % (mq.protocol.procedure, self.pc.procedure))
mq.add_client(self.vm)
mq.notify_connection(self.vm)
logging.info("start receiving commands")
exit = False
while not exit:
logging.debug("- CLIENT %s LISTENING" % self.vm)
received = self.pc.receive_command()
logging.debug("- CLIENT %s EXECUTED: %s" % (self.vm, received))
if received.name == 'STOP_AGENT':
exit = True
logging.info("stop receiving commands")
remove_running(self.vm)
