@@ -300,31 +300,35 @@ fn onchain_spend_receive() {
300300 expect_channel_ready_event ! ( node_a, node_b. node_id( ) ) ;
301301 expect_channel_ready_event ! ( node_b, node_a. node_id( ) ) ;
302302
303- let node_a_balance = premine_amount_sat - reserve_amount_sat;
304- let node_b_balance_lower = premine_amount_sat - channel_amount_sat - reserve_amount_sat - 1000 ;
305- let node_b_balance_upper = premine_amount_sat - channel_amount_sat - reserve_amount_sat;
306- assert_eq ! ( node_a. list_balances( ) . spendable_onchain_balance_sats, node_a_balance) ;
307- assert ! ( node_b. list_balances( ) . spendable_onchain_balance_sats > node_b_balance_lower) ;
308- assert ! ( node_b. list_balances( ) . spendable_onchain_balance_sats < node_b_balance_upper) ;
303+ let onchain_fee_buffer_sat = 1000 ;
304+ let expected_node_a_balance = premine_amount_sat - reserve_amount_sat;
305+ let expected_node_b_balance_lower =
306+ premine_amount_sat - channel_amount_sat - reserve_amount_sat - onchain_fee_buffer_sat;
307+ let expected_node_b_balance_upper =
308+ premine_amount_sat - channel_amount_sat - reserve_amount_sat;
309+ assert_eq ! ( node_a. list_balances( ) . spendable_onchain_balance_sats, expected_node_a_balance) ;
310+ assert ! ( node_b. list_balances( ) . spendable_onchain_balance_sats > expected_node_b_balance_lower) ;
311+ assert ! ( node_b. list_balances( ) . spendable_onchain_balance_sats < expected_node_b_balance_upper) ;
309312
310313 assert_eq ! (
311314 Err ( NodeError :: InsufficientFunds ) ,
312- node_a. onchain_payment( ) . send_to_address( & addr_b, node_a_balance + 1 )
315+ node_a. onchain_payment( ) . send_to_address( & addr_b, expected_node_a_balance + 1 )
313316 ) ;
314317
315- let txid = node_b. onchain_payment ( ) . send_to_address ( & addr_a, 1000 ) . unwrap ( ) ;
318+ let amount_to_send_sats = 1000 ;
319+ let txid = node_b. onchain_payment ( ) . send_to_address ( & addr_a, amount_to_send_sats) . unwrap ( ) ;
316320 generate_blocks_and_wait ( & bitcoind. client , & electrsd. client , 6 ) ;
317321 wait_for_tx ( & electrsd. client , txid) ;
318322
319323 node_a. sync_wallets ( ) . unwrap ( ) ;
320324 node_b. sync_wallets ( ) . unwrap ( ) ;
321325
322- let node_a_balance = node_a_balance + 1000 ;
323- let node_b_balance_lower = node_b_balance_lower - 1000 ;
324- let node_b_balance_upper = node_b_balance_upper - 1000 ;
325- assert_eq ! ( node_a. list_balances( ) . spendable_onchain_balance_sats, node_a_balance ) ;
326- assert ! ( node_b. list_balances( ) . spendable_onchain_balance_sats > node_b_balance_lower ) ;
327- assert ! ( node_b. list_balances( ) . spendable_onchain_balance_sats < node_b_balance_upper ) ;
326+ let expected_node_a_balance = expected_node_a_balance + amount_to_send_sats ;
327+ let expected_node_b_balance_lower = expected_node_b_balance_lower - amount_to_send_sats ;
328+ let expected_node_b_balance_upper = expected_node_b_balance_upper - amount_to_send_sats ;
329+ assert_eq ! ( node_a. list_balances( ) . spendable_onchain_balance_sats, expected_node_a_balance ) ;
330+ assert ! ( node_b. list_balances( ) . spendable_onchain_balance_sats > expected_node_b_balance_lower ) ;
331+ assert ! ( node_b. list_balances( ) . spendable_onchain_balance_sats < expected_node_b_balance_upper ) ;
328332
329333 let addr_b = node_b. onchain_payment ( ) . new_address ( ) . unwrap ( ) ;
330334 let txid = node_a. onchain_payment ( ) . send_all_to_address ( & addr_b, true ) . unwrap ( ) ;
@@ -334,13 +338,13 @@ fn onchain_spend_receive() {
334338 node_a. sync_wallets ( ) . unwrap ( ) ;
335339 node_b. sync_wallets ( ) . unwrap ( ) ;
336340
337- let node_b_balance_lower = node_b_balance_lower + node_a_balance ;
338- let node_b_balance_upper = node_b_balance_upper + node_a_balance ;
339- let node_a_balance = 0 ;
340- assert_eq ! ( node_a. list_balances( ) . spendable_onchain_balance_sats, node_a_balance ) ;
341+ let expected_node_b_balance_lower = expected_node_b_balance_lower + expected_node_a_balance ;
342+ let expected_node_b_balance_upper = expected_node_b_balance_upper + expected_node_a_balance ;
343+ let expected_node_a_balance = 0 ;
344+ assert_eq ! ( node_a. list_balances( ) . spendable_onchain_balance_sats, expected_node_a_balance ) ;
341345 assert_eq ! ( node_a. list_balances( ) . total_onchain_balance_sats, reserve_amount_sat) ;
342- assert ! ( node_b. list_balances( ) . spendable_onchain_balance_sats > node_b_balance_lower ) ;
343- assert ! ( node_b. list_balances( ) . spendable_onchain_balance_sats < node_b_balance_upper ) ;
346+ assert ! ( node_b. list_balances( ) . spendable_onchain_balance_sats > expected_node_b_balance_lower ) ;
347+ assert ! ( node_b. list_balances( ) . spendable_onchain_balance_sats < expected_node_b_balance_upper ) ;
344348
345349 let addr_b = node_b. onchain_payment ( ) . new_address ( ) . unwrap ( ) ;
346350 let txid = node_a. onchain_payment ( ) . send_all_to_address ( & addr_b, false ) . unwrap ( ) ;
@@ -350,14 +354,14 @@ fn onchain_spend_receive() {
350354 node_a. sync_wallets ( ) . unwrap ( ) ;
351355 node_b. sync_wallets ( ) . unwrap ( ) ;
352356
353- let node_b_balance_lower = node_b_balance_lower + reserve_amount_sat;
354- let node_b_balance_upper = node_b_balance_upper + reserve_amount_sat;
355- let node_a_balance = 0 ;
357+ let expected_node_b_balance_lower = expected_node_b_balance_lower + reserve_amount_sat;
358+ let expected_node_b_balance_upper = expected_node_b_balance_upper + reserve_amount_sat;
359+ let expected_node_a_balance = 0 ;
356360
357- assert_eq ! ( node_a. list_balances( ) . spendable_onchain_balance_sats, node_a_balance ) ;
358- assert_eq ! ( node_a. list_balances( ) . total_onchain_balance_sats, node_a_balance ) ;
359- assert ! ( node_b. list_balances( ) . spendable_onchain_balance_sats > node_b_balance_lower ) ;
360- assert ! ( node_b. list_balances( ) . spendable_onchain_balance_sats < node_b_balance_upper ) ;
361+ assert_eq ! ( node_a. list_balances( ) . spendable_onchain_balance_sats, expected_node_a_balance ) ;
362+ assert_eq ! ( node_a. list_balances( ) . total_onchain_balance_sats, expected_node_a_balance ) ;
363+ assert ! ( node_b. list_balances( ) . spendable_onchain_balance_sats > expected_node_b_balance_lower ) ;
364+ assert ! ( node_b. list_balances( ) . spendable_onchain_balance_sats < expected_node_b_balance_upper ) ;
361365}
362366
363367#[ test]
0 commit comments