Fix test, reducing coverage slightly.

Author: tnull

lightning/src/ln/peer_handler.rs

@@ -1878,11 +1878,11 @@ mod tests {
 		// RoutingMessageHandlers that intentionally return more channel messages
 		// than can fit into a peer's buffer).
 		let (mut fd_a, mut fd_b) = establish_connection(&peers[0], &peers[1]);
+		peers[0].process_events();
 
 		// Make each peer to read the messages that the other peer just wrote to them. Note that
 		// due to the max-message-before-ping limits this may take a few iterations to complete.
 		for _ in 0..150/super::BUFFER_DRAIN_MSGS_PER_TICK + 1 {
-			peers[0].process_events();
 			let b_read_data = fd_a.outbound_data.lock().unwrap().split_off(0);
 			assert!(!b_read_data.is_empty());
 
@@ -1891,12 +1891,16 @@ mod tests {
 
 			let a_read_data = fd_b.outbound_data.lock().unwrap().split_off(0);
 			assert!(!a_read_data.is_empty());
-			peers[0].read_event(&mut fd_a, &a_read_data).unwrap();
 
-			peers[1].process_events();
-			assert_eq!(fd_b.outbound_data.lock().unwrap().len(), 0, "Until B receives data, it shouldn't send more messages");
+			peers[0].read_event(&mut fd_a, &a_read_data).unwrap();
+			peers[0].process_events();
 		}
 
+		let b_remaining_data = fd_a.outbound_data.lock().unwrap().split_off(0);
+		peers[1].read_event(&mut fd_b, &b_remaining_data).unwrap();
+		peers[1].process_events();
+		assert_eq!(fd_b.outbound_data.lock().unwrap().len(), 0, "Until B receives data, it shouldn't send more messages");
+
 		// Check that each peer has received the expected number of channel updates and channel
 		// announcements.
 		assert_eq!(cfgs[0].routing_handler.chan_upds_recvd.load(Ordering::Acquire), 100);