def before_scenario(self, context, scenario):
context.peer = utils.mock_peer(context._connection, '127.0.0.1', 1234)
context.add_recv_packet, context.received_packets = \
utils.mock_connection_recv(context._connection)
context.sent_packets = utils.mock_connection_send(context._connection)
time_sleep_patcher = mock.patch('time.sleep')
time_sleep_patcher.start()
context.time_sleep_patcher = time_sleep_patcher
def step_impl(context):
context.peer_manager.connected_peers = []
for ip, port, node_id in context.peer_addresses:
peer = utils.mock_peer(utils.mock_connection(), ip, port)
peer.node_id = node_id
context.peer_manager.connected_peers.append(peer)
def start_peer_side_effect(connection, ip, port):
peer = utils.mock_peer(connection, ip, port)
return peer
