def test_participant_rendezvous_later(participant_stub):
server = grpc.server(futures.ThreadPoolExecutor(max_workers=1))
coordinator_pb2_grpc.add_CoordinatorServicer_to_server(
Coordinator(Participants()), server)
server.add_insecure_port("localhost:50051")
server.start()
reply = participant_stub.Rendezvous(coordinator_pb2.RendezvousRequest())
server.stop(0)
assert reply.response == coordinator_pb2.RendezvousResponse.LATER
def rendezvous(channel):
stub = coordinator_pb2_grpc.CoordinatorStub(channel)
response = coordinator_pb2.RendezvousResponse.LATER
while response == coordinator_pb2.RendezvousResponse.LATER:
reply = stub.Rendezvous(coordinator_pb2.RendezvousRequest())
if reply.response == coordinator_pb2.RendezvousResponse.ACCEPT:
print("Participant received: ACCEPT")
elif reply.response == coordinator_pb2.RendezvousResponse.LATER:
print(f"Participant received: LATER. Retrying in {RETRY_TIMEOUT}s")
time.sleep(RETRY_TIMEOUT)
response = reply.response
def test_participant_rendezvous_accept(participant_stub, coordinator_service):
reply = participant_stub.Rendezvous(coordinator_pb2.RendezvousRequest())
assert reply.response == coordinator_pb2.RendezvousResponse.ACCEPT
def test_heartbeat(participant_stub, coordinator_service):
# first we need to rendezvous so that the participant is added to the list of participants
_ = participant_stub.Rendezvous(coordinator_pb2.RendezvousRequest())
reply = participant_stub.Heartbeat(coordinator_pb2.HeartbeatRequest())
assert reply == coordinator_pb2.HeartbeatReply()