@@ -11,7 +11,7 @@ use bitcoin::blockdata::constants::genesis_block;
1111use bitcoin:: network:: constants:: Network ;
1212use electrsd:: bitcoind:: bitcoincore_rpc:: bitcoincore_rpc_json:: AddressType ;
1313use bitcoind:: bitcoincore_rpc:: RpcApi ;
14- use electrum_client:: ElectrumApi ;
14+ use electrsd :: electrum_client:: ElectrumApi ;
1515
1616use std:: env;
1717use std:: sync:: Mutex ;
@@ -43,21 +43,22 @@ pub fn generate_blocks_and_wait(bitcoind: &BitcoinD, electrsd: &ElectrsD, num: u
4343 let address = bitcoind
4444 . client
4545 . get_new_address ( Some ( "test" ) , Some ( AddressType :: Legacy ) )
46- . expect ( "failed to get new address" ) ;
46+ . expect ( "failed to get new address" )
47+ . assume_checked ( ) ;
4748 // TODO: expect this Result once the WouldBlock issue is resolved upstream.
4849 let _block_hashes_res = bitcoind. client . generate_to_address ( num as u64 , & address) ;
4950 wait_for_block ( electrsd, cur_height as usize + num) ;
5051}
5152
5253pub fn wait_for_block ( electrsd : & ElectrsD , min_height : usize ) {
53- let mut header = match electrsd. client . block_headers_subscribe ( ) {
54+ let mut header = match electrsd. client . block_headers_subscribe_raw ( ) {
5455 Ok ( header) => header,
5556 Err ( _) => {
5657 // While subscribing should succeed the first time around, we ran into some cases where
5758 // it didn't. Since we can't proceed without subscribing, we try again after a delay
5859 // and panic if it still fails.
5960 std:: thread:: sleep ( Duration :: from_secs ( 1 ) ) ;
60- electrsd. client . block_headers_subscribe ( ) . expect ( "failed to subscribe to block headers" )
61+ electrsd. client . block_headers_subscribe_raw ( ) . expect ( "failed to subscribe to block headers" )
6162 }
6263 } ;
6364 loop {
@@ -67,7 +68,7 @@ pub fn wait_for_block(electrsd: &ElectrsD, min_height: usize) {
6768 header = exponential_backoff_poll ( || {
6869 electrsd. trigger ( ) . expect ( "failed to trigger electrsd" ) ;
6970 electrsd. client . ping ( ) . expect ( "failed to ping electrsd" ) ;
70- electrsd. client . block_headers_pop ( ) . expect ( "failed to pop block header" )
71+ electrsd. client . block_headers_pop_raw ( ) . expect ( "failed to pop block header" )
7172 } ) ;
7273 }
7374}
@@ -177,7 +178,7 @@ fn test_esplora_syncs() {
177178 assert_eq ! ( events. len( ) , 1 ) ;
178179
179180 // Check registered confirmed transactions are marked confirmed
180- let new_address = bitcoind. client . get_new_address ( Some ( "test" ) , Some ( AddressType :: Legacy ) ) . unwrap ( ) ;
181+ let new_address = bitcoind. client . get_new_address ( Some ( "test" ) , Some ( AddressType :: Legacy ) ) . unwrap ( ) . assume_checked ( ) ;
181182 let txid = bitcoind. client . send_to_address ( & new_address, Amount :: from_sat ( 5000 ) , None , None , None , None , None , None ) . unwrap ( ) ;
182183 tx_sync. register_tx ( & txid, & new_address. payload . script_pubkey ( ) ) ;
183184
@@ -260,7 +261,7 @@ async fn test_esplora_syncs() {
260261 assert_eq ! ( events. len( ) , 1 ) ;
261262
262263 // Check registered confirmed transactions are marked confirmed
263- let new_address = bitcoind. client . get_new_address ( Some ( "test" ) , Some ( AddressType :: Legacy ) ) . unwrap ( ) ;
264+ let new_address = bitcoind. client . get_new_address ( Some ( "test" ) , Some ( AddressType :: Legacy ) ) . unwrap ( ) . assume_checked ( ) ;
264265 let txid = bitcoind. client . send_to_address ( & new_address, Amount :: from_sat ( 5000 ) , None , None , None , None , None , None ) . unwrap ( ) ;
265266 tx_sync. register_tx ( & txid, & new_address. payload . script_pubkey ( ) ) ;
266267
0 commit comments