Skip to content

Commit 7882d04

Browse files
committed
Clean up some peer_connected call rustfmt verticality
`rustfmt` loves to make expressions vertical by putting each method parameter on its own line. In cases where each method parameter is actually doing something, this can be fine, but in some cases we have a few method parameters that are straightforward and shouldn't be the readers focus, mixed with one or two parameters which are key. Here, we clean up a regular instance of this in calls to `peer_connected` in `priv_short_conf_tests.rs` and `shutdown_tests.rs`. In these tests, the fact that we're reconnecting is important, but the specific features in the `Init` message are not. Thus, condensing the calls onto two lines by breaking the init message out into a variable enables the reader to more easily skim past useless details.
1 parent 7ee4adf commit 7882d04

File tree

2 files changed

+29
-96
lines changed

2 files changed

+29
-96
lines changed

lightning/src/ln/priv_short_conf_tests.rs

Lines changed: 20 additions & 48 deletions
Original file line numberDiff line numberDiff line change
@@ -134,61 +134,33 @@ fn test_priv_forwarding_rejection() {
134134
nodes_1_deserialized
135135
);
136136

137-
nodes[0]
138-
.node
139-
.peer_connected(
140-
node_b_id,
141-
&msgs::Init {
142-
features: nodes[1].node.init_features(),
143-
networks: None,
144-
remote_network_address: None,
145-
},
146-
true,
147-
)
148-
.unwrap();
149-
nodes[1]
150-
.node
151-
.peer_connected(
152-
node_a_id,
153-
&msgs::Init {
154-
features: nodes[0].node.init_features(),
155-
networks: None,
156-
remote_network_address: None,
157-
},
158-
false,
159-
)
160-
.unwrap();
137+
let bs_init_msg = msgs::Init {
138+
features: nodes[1].node.init_features(),
139+
networks: None,
140+
remote_network_address: None,
141+
};
142+
nodes[0].node.peer_connected(node_b_id, &bs_init_msg, true).unwrap();
143+
144+
let as_init_msg = msgs::Init {
145+
features: nodes[0].node.init_features(),
146+
networks: None,
147+
remote_network_address: None,
148+
};
149+
nodes[1].node.peer_connected(node_a_id, &as_init_msg, false).unwrap();
161150
let as_reestablish = get_chan_reestablish_msgs!(nodes[0], nodes[1]).pop().unwrap();
162151
let bs_reestablish = get_chan_reestablish_msgs!(nodes[1], nodes[0]).pop().unwrap();
163152
nodes[1].node.handle_channel_reestablish(node_a_id, &as_reestablish);
164153
nodes[0].node.handle_channel_reestablish(node_b_id, &bs_reestablish);
165154
get_event_msg!(nodes[0], MessageSendEvent::SendChannelUpdate, node_b_id);
166155
get_event_msg!(nodes[1], MessageSendEvent::SendChannelUpdate, node_a_id);
167156

168-
nodes[1]
169-
.node
170-
.peer_connected(
171-
node_c_id,
172-
&msgs::Init {
173-
features: nodes[2].node.init_features(),
174-
networks: None,
175-
remote_network_address: None,
176-
},
177-
true,
178-
)
179-
.unwrap();
180-
nodes[2]
181-
.node
182-
.peer_connected(
183-
node_b_id,
184-
&msgs::Init {
185-
features: nodes[1].node.init_features(),
186-
networks: None,
187-
remote_network_address: None,
188-
},
189-
false,
190-
)
191-
.unwrap();
157+
let cs_init_msg = msgs::Init {
158+
features: nodes[2].node.init_features(),
159+
networks: None,
160+
remote_network_address: None,
161+
};
162+
nodes[1].node.peer_connected(node_c_id, &cs_init_msg, true).unwrap();
163+
nodes[2].node.peer_connected( node_b_id, &bs_init_msg, false).unwrap();
192164
let bs_reestablish = get_chan_reestablish_msgs!(nodes[1], nodes[2]).pop().unwrap();
193165
let cs_reestablish = get_chan_reestablish_msgs!(nodes[2], nodes[1]).pop().unwrap();
194166
nodes[2].node.handle_channel_reestablish(node_b_id, &bs_reestablish);

lightning/src/ln/shutdown_tests.rs

Lines changed: 9 additions & 48 deletions
Original file line numberDiff line numberDiff line change
@@ -683,31 +683,14 @@ fn do_test_shutdown_rebroadcast(recv_count: u8) {
683683
nodes[0].node.peer_disconnected(node_b_id);
684684
nodes[1].node.peer_disconnected(node_a_id);
685685

686-
nodes[0]
687-
.node
688-
.peer_connected(
689-
node_b_id,
690-
&msgs::Init {
691-
features: nodes[1].node.init_features(),
692-
networks: None,
693-
remote_network_address: None,
694-
},
695-
true,
696-
)
697-
.unwrap();
686+
let init_msg = msgs::Init {
687+
features: nodes[1].node.init_features(),
688+
networks: None,
689+
remote_network_address: None,
690+
};
691+
nodes[0].node.peer_connected(node_b_id, &init_msg, true).unwrap();
698692
let node_0_reestablish = get_chan_reestablish_msgs!(nodes[0], nodes[1]).pop().unwrap();
699-
nodes[1]
700-
.node
701-
.peer_connected(
702-
node_a_id,
703-
&msgs::Init {
704-
features: nodes[0].node.init_features(),
705-
networks: None,
706-
remote_network_address: None,
707-
},
708-
false,
709-
)
710-
.unwrap();
693+
nodes[1].node.peer_connected(node_a_id, &init_msg, false).unwrap();
711694
let node_1_reestablish = get_chan_reestablish_msgs!(nodes[1], nodes[0]).pop().unwrap();
712695

713696
nodes[1].node.handle_channel_reestablish(node_a_id, &node_0_reestablish);
@@ -772,31 +755,9 @@ fn do_test_shutdown_rebroadcast(recv_count: u8) {
772755
nodes[0].node.peer_disconnected(node_b_id);
773756
nodes[1].node.peer_disconnected(node_a_id);
774757

775-
nodes[1]
776-
.node
777-
.peer_connected(
778-
node_a_id,
779-
&msgs::Init {
780-
features: nodes[0].node.init_features(),
781-
networks: None,
782-
remote_network_address: None,
783-
},
784-
true,
785-
)
786-
.unwrap();
758+
nodes[1].node.peer_connected(node_a_id, &init_msg, true).unwrap();
787759
let node_1_2nd_reestablish = get_chan_reestablish_msgs!(nodes[1], nodes[0]).pop().unwrap();
788-
nodes[0]
789-
.node
790-
.peer_connected(
791-
node_b_id,
792-
&msgs::Init {
793-
features: nodes[1].node.init_features(),
794-
networks: None,
795-
remote_network_address: None,
796-
},
797-
false,
798-
)
799-
.unwrap();
760+
nodes[0].node.peer_connected(node_b_id, &init_msg, false).unwrap();
800761
if recv_count == 0 {
801762
// If all closing_signeds weren't delivered we can just resume where we left off...
802763
let node_0_2nd_reestablish = get_chan_reestablish_msgs!(nodes[0], nodes[1]).pop().unwrap();

0 commit comments

Comments
 (0)