def test_two_tasks(self):
r1 = random_int()
r2 = random_int()
f1 = self.test_wm.submit(identity, (r1, ), {})
f2 = self.test_wm.submit(identity, (r2, ), {})
assert f1.result == r1
assert f2.result == r2
def test_task_send(self):
r = random_int()
self.test_wm.submit(identity, (r, ))
with self.rr_socket() as s:
self.test_core.send_message(s, Message.TASK_REQUEST)
msg = self.test_core.recv_message(s)
assert msg.message == Message.TASK
assert isinstance(msg.payload, Task)
assert msg.payload.args == (r, )
def test_result_return(self):
r = random_int()
future = self.test_wm.submit(identity, (r, ))
with self.rr_socket() as s:
self.test_core.send_message(s, Message.TASK_REQUEST)
msg = self.test_core.recv_message(s)
task = msg.payload
result = task.execute()
self.test_core.send_message(s, Message.RESULT, result)
assert future.result == r
def test_task(self):
r = random_int()
future = self.test_wm.submit(identity,(r,),{})
assert future.get_result() == r
def test_task_avail_beacon(self):
with self.expect_announcement(Message.TASKS_AVAILABLE):
self.test_wm.submit(identity, (random_int(), ))
time.sleep(BEACON_WAIT)
def test_worker_processes_task(self):
r = random_int()
task = Task(identity, (r,), {})
rsl = self.roundtrip_task(task)
assert rsl.result == r