Function restart_network

tools/debugging/stress_test_transfers.py:318–343

Stop all `RunningNode`s and start them again under the nursery, returns None if the script is shuting down.

(
    nursery: Nursery,
    port_generator: Iterator[Port],
    running_nodes: List[RunningNode],
    retry_timeout: int,
)

def restart_network(
    nursery: Nursery,
    port_generator: Iterator[Port],
    running_nodes: List[RunningNode],
    retry_timeout: int,
) -> Optional[List[RunningNode]]:
    """Stop all `RunningNode`s and start them again under the nursery, returns
    None if the script is shuting down.
    """
    greenlets = set(
        nursery.spawn_under_watch(
            restart_and_wait_for_server, nursery, port_generator, node, retry_timeout
        )
        for node in running_nodes
    )

    all_running_nodes = []
    for g in gevent.joinall(greenlets, raise_error=True):
        running_node = g.get()

        if running_node is None:
            return None

        all_running_nodes.append(running_node)

    return all_running_nodes


def transfer_and_assert_successful(