X-Git-Url: http://git.bitcoin.ninja/index.cgi?a=blobdiff_plain;f=lightning%2Fsrc%2Fln%2Freorg_tests.rs;h=91d5b3070610579474b39737b518c94932e9200d;hb=b1b36661ee21713873f07fcfa8b1bce5fdc81506;hp=f6754e8bf2dc8d6e11f0a56370117436bddd57cb;hpb=32fdeb7b4ee12011f9503c7bdc6eb64423e6d232;p=rust-lightning

diff --git a/lightning/src/ln/reorg_tests.rs b/lightning/src/ln/reorg_tests.rs
index f6754e8b..91d5b307 100644
--- a/lightning/src/ln/reorg_tests.rs
+++ b/lightning/src/ln/reorg_tests.rs
@@ -277,6 +277,9 @@ fn do_test_unconf_chan(reload_node: bool, reorg_after_reload: bool, use_funding_
 	assert_eq!(nodes[0].node.short_to_chan_info.read().unwrap().len(), 2);
 	mem::drop(channel_state);
 
+	assert_eq!(nodes[0].node.list_channels()[0].confirmations, Some(10));
+	assert_eq!(nodes[1].node.list_channels()[0].confirmations, Some(10));
+
 	if !reorg_after_reload {
 		if use_funding_unconfirmed {
 			let relevant_txids = nodes[0].node.get_relevant_txids();
@@ -287,12 +290,16 @@ fn do_test_unconf_chan(reload_node: bool, reorg_after_reload: bool, use_funding_
 			let txid = relevant_txids[0].0;
 			assert_eq!(txid, chan.3.txid());
 			nodes[0].node.transaction_unconfirmed(&txid);
+			assert_eq!(nodes[0].node.list_usable_channels().len(), 0);
 		} else if connect_style == ConnectStyle::FullBlockViaListen {
 			disconnect_blocks(&nodes[0], CHAN_CONFIRM_DEPTH - 1);
 			assert_eq!(nodes[0].node.list_usable_channels().len(), 1);
+			assert_eq!(nodes[0].node.list_channels()[0].confirmations, Some(1));
 			disconnect_blocks(&nodes[0], 1);
+			assert_eq!(nodes[0].node.list_usable_channels().len(), 0);
 		} else {
 			disconnect_all_blocks(&nodes[0]);
+			assert_eq!(nodes[0].node.list_usable_channels().len(), 0);
 		}
 
 		let relevant_txids = nodes[0].node.get_relevant_txids();
@@ -334,12 +341,16 @@ fn do_test_unconf_chan(reload_node: bool, reorg_after_reload: bool, use_funding_
 			let txid = relevant_txids[0].0;
 			assert_eq!(txid, chan.3.txid());
 			nodes[0].node.transaction_unconfirmed(&txid);
+			assert_eq!(nodes[0].node.list_channels().len(), 0);
 		} else if connect_style == ConnectStyle::FullBlockViaListen {
 			disconnect_blocks(&nodes[0], CHAN_CONFIRM_DEPTH - 1);
 			assert_eq!(nodes[0].node.list_channels().len(), 1);
+			assert_eq!(nodes[0].node.list_channels()[0].confirmations, Some(1));
 			disconnect_blocks(&nodes[0], 1);
+			assert_eq!(nodes[0].node.list_usable_channels().len(), 0);
 		} else {
 			disconnect_all_blocks(&nodes[0]);
+			assert_eq!(nodes[0].node.list_usable_channels().len(), 0);
 		}
 
 		let relevant_txids = nodes[0].node.get_relevant_txids();