@@ -35,7 +35,6 @@ use vss_client::util::retry::{
3535use vss_client:: util:: storable_builder:: { EntropySource , StorableBuilder } ;
3636
3737use crate :: io:: utils:: check_namespace_key_validity;
38- use crate :: runtime:: Runtime ;
3938
4039type CustomRetryPolicy = FilteredRetryPolicy <
4140 JitteredRetryPolicy <
@@ -53,14 +52,13 @@ pub struct VssStore {
5352 // Version counter to ensure that writes are applied in the correct order. It is assumed that read and list
5453 // operations aren't sensitive to the order of execution.
5554 next_version : AtomicU64 ,
56- runtime : Arc < Runtime > ,
5755 internal_runtime : Option < tokio:: runtime:: Runtime > ,
5856}
5957
6058impl VssStore {
6159 pub ( crate ) fn new (
6260 base_url : String , store_id : String , vss_seed : [ u8 ; 32 ] ,
63- header_provider : Arc < dyn VssHeaderProvider > , runtime : Arc < Runtime > ,
61+ header_provider : Arc < dyn VssHeaderProvider > ,
6462 ) -> Self {
6563 let inner = Arc :: new ( VssStoreInner :: new ( base_url, store_id, vss_seed, header_provider) ) ;
6664 let next_version = AtomicU64 :: new ( 1 ) ;
@@ -78,7 +76,7 @@ impl VssStore {
7876 . unwrap ( ) ,
7977 ) ;
8078
81- Self { inner, next_version, runtime , internal_runtime }
79+ Self { inner, next_version, internal_runtime }
8280 }
8381
8482 // Same logic as for the obfuscated keys below, but just for locking, using the plaintext keys
@@ -598,38 +596,40 @@ mod tests {
598596
599597 use rand:: distributions:: Alphanumeric ;
600598 use rand:: { thread_rng, Rng , RngCore } ;
601- use tokio:: runtime;
602599 use vss_client:: headers:: FixedHeaders ;
603600
604601 use super :: * ;
605602 use crate :: io:: test_utils:: do_read_write_remove_list_persist;
603+ use crate :: logger:: Logger ;
606604
607605 #[ test]
608606 fn vss_read_write_remove_list_persist ( ) {
609- let runtime = Arc :: new ( Runtime :: new ( ) . unwrap ( ) ) ;
610607 let vss_base_url = std:: env:: var ( "TEST_VSS_BASE_URL" ) . unwrap ( ) ;
611608 let mut rng = thread_rng ( ) ;
612609 let rand_store_id: String = ( 0 ..7 ) . map ( |_| rng. sample ( Alphanumeric ) as char ) . collect ( ) ;
613610 let mut vss_seed = [ 0u8 ; 32 ] ;
614611 rng. fill_bytes ( & mut vss_seed) ;
615612 let header_provider = Arc :: new ( FixedHeaders :: new ( HashMap :: new ( ) ) ) ;
613+ let logger = Arc :: new ( Logger :: new_log_facade ( ) ) ;
614+ let runtime = Arc :: new ( Runtime :: new ( logger) . unwrap ( ) ) ;
616615 let vss_store =
617- VssStore :: new ( vss_base_url, rand_store_id, vss_seed, header_provider, runtime) . unwrap ( ) ;
616+ VssStore :: new ( vss_base_url, rand_store_id, vss_seed, header_provider, runtime) ;
618617
619618 do_read_write_remove_list_persist ( & vss_store) ;
620619 }
621620
622621 #[ tokio:: test( flavor = "multi_thread" , worker_threads = 1 ) ]
623622 async fn vss_read_write_remove_list_persist_in_runtime_context ( ) {
624- let runtime = Arc :: new ( Runtime :: new ( ) . unwrap ( ) ) ;
625623 let vss_base_url = std:: env:: var ( "TEST_VSS_BASE_URL" ) . unwrap ( ) ;
626624 let mut rng = thread_rng ( ) ;
627625 let rand_store_id: String = ( 0 ..7 ) . map ( |_| rng. sample ( Alphanumeric ) as char ) . collect ( ) ;
628626 let mut vss_seed = [ 0u8 ; 32 ] ;
629627 rng. fill_bytes ( & mut vss_seed) ;
630628 let header_provider = Arc :: new ( FixedHeaders :: new ( HashMap :: new ( ) ) ) ;
629+ let logger = Arc :: new ( Logger :: new_log_facade ( ) ) ;
630+ let runtime = Arc :: new ( Runtime :: new ( logger) . unwrap ( ) ) ;
631631 let vss_store =
632- VssStore :: new ( vss_base_url, rand_store_id, vss_seed, header_provider, runtime) . unwrap ( ) ;
632+ VssStore :: new ( vss_base_url, rand_store_id, vss_seed, header_provider, runtime) ;
633633
634634 do_read_write_remove_list_persist ( & vss_store) ;
635635 drop ( vss_store)
0 commit comments