def test_etcd_server_start_stop(self):
server = EtcdServer()
server.start()
try:
port = server.get_port()
host = server.get_host()
self.assertGreater(port, 0)
self.assertEqual("localhost", host)
self.assertEqual(f"{host}:{port}", server.get_endpoint())
self.assertIsNotNone(server.get_client().version)
finally:
server.stop()
def test_etcd_server_with_rendezvous(self):
server = EtcdServer()
server.start()
client = etcd.Client(server.get_host(), server.get_port())
rdzv = EtcdRendezvous(
client=client,
prefix="test",
run_id=1,
num_min_workers=1,
num_max_workers=1,
timeout=60,
last_call_timeout=30,
)
rdzv_handler = EtcdRendezvousHandler(rdzv)
store, rank, world_size = rdzv_handler.next_rendezvous()
self.assertIsNotNone(store)
self.assertEqual(0, rank)
self.assertEqual(1, world_size)