@@ -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