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_MultipleMQ():
host = "localhost"
mq1 = MQStar(host)
mq2 = MQStar(host, session=mq1.session)
client, message = "c1", "HELLO"
mq1.send_server(client, message)
c, m = mq2.receive_server()
assert (c == client)
assert (eval(m) == message)
def test_MultipleMQ():
host = "localhost"
mq1 = MQStar(host)
mq2 = MQStar(host, session=mq1.session)
client, message = "c1", "HELLO"
mq1.send_server(client, message)
c, m = mq2.receive_server()
assert (c == client)
assert (eval(m) == message)
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_MQ():
host = "localhost"
mq = MQStar(host)
mq.clean()
clients = ["c1", "c2", "c3"]
mq.add_clients(clients)
for c in clients:
mq.send_server(c, "STARTED")
for i in range(len(clients)):
c, m = mq.receive_server()
assert c in clients
assert eval(m) == "STARTED", "Uncorrect value: %s" % m
mq.send_client(c, "END %s" % i)
for c in clients:
m = mq.receive_client(c)
print m
assert (m.startswith("END "))
def test_MQ():
host = "localhost"
mq = MQStar(host)
mq.clean()
clients = ["c1", "c2", "c3"]
mq.add_clients(clients)
for c in clients:
mq.send_server(c, "STARTED")
for i in range(len(clients)):
c, m = mq.receive_server()
assert c in clients
assert eval(m) == "STARTED", "Uncorrect value: %s" % m
mq.send_client(c, "END %s" % i)
for c in clients:
m = mq.receive_client(c)
print m
assert (m.startswith("END "))
