def wait_for_peers(self) -> None:
if self.node_count < 2:
return
for cl_node in self.cl_nodes:
wait_for_peers_count_at_least(
cl_node.node, self.node_count - 1, cl_node.config.command_timeout
)
def create_cl_network(
self, node_count: int = 3, network_connections: List[List[int]] = None
) -> None:
"""
Allow creation of a network where all nodes are not connected to node-0's network and therefore each other.
Ex: Star Network
node3
|
node1 -- node0 -- node2
create_cl_network(node_count=4, network_connections=[[0, 1], [0, 2], [0, 3]])
:param node_count: Number of nodes to create.
:param network_connections: A list of lists of node indexes that should be joined.
"""
self.network_names = {}
kp = self.get_key()
config = DockerConfig(
self.docker_client,
node_private_key=kp.private_key,
node_public_key=kp.public_key,
network=self.create_docker_network(),
node_account=kp,
)
self.add_bootstrap(config)
for i in range(1, node_count):
kp = self.get_key()
config = DockerConfig(
self.docker_client,
node_private_key=kp.private_key,
network=self.create_docker_network(),
node_account=kp,
)
self.add_cl_node(config, network_with_bootstrap=False)
for network_members in network_connections:
self.connect(network_members)
for node_number in range(1, node_count):
self.wait_method(
wait_for_approved_block_received_handler_state, node_number
)
# Calculate how many peers should be connected to each node.
# Assumes duplicate network connections are not given, else wait_for_peers will time out.
peer_counts = [0] * node_count
for network in network_connections:
for node_num in network:
peer_counts[node_num] += len(network) - 1
timeout = self.docker_nodes[0].timeout
for node_num, peer_count in enumerate(peer_counts):
if peer_count > 0:
wait_for_peers_count_at_least(
self.docker_nodes[node_num], peer_count, timeout
)
