Skip to content

Commit eff02f8

Browse files
committed
Fix and run all tests under cfg(vss_test)
.. we previously avoided running some tests which turned out to be broken.
1 parent db79781 commit eff02f8

File tree

3 files changed

+11
-11
lines changed

3 files changed

+11
-11
lines changed

.github/workflows/vss-integration.yml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -44,4 +44,5 @@ jobs:
4444
run: |
4545
cd ldk-node
4646
export TEST_VSS_BASE_URL="http://localhost:8080/vss"
47+
RUSTFLAGS="--cfg vss_test" cargo test io::vss_store
4748
RUSTFLAGS="--cfg vss_test" cargo test --test integration_tests_vss

src/builder.rs

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -710,8 +710,7 @@ impl NodeBuilder {
710710

711711
let vss_seed_bytes: [u8; 32] = vss_xprv.private_key.secret_bytes();
712712

713-
let vss_store =
714-
VssStore::new(vss_url, store_id, vss_seed_bytes, header_provider, Arc::clone(&runtime));
713+
let vss_store = VssStore::new(vss_url, store_id, vss_seed_bytes, header_provider);
715714
build_with_store_internal(
716715
config,
717716
self.chain_data_source_config.as_ref(),

src/io/vss_store.rs

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,6 @@ use vss_client::util::retry::{
3535
use vss_client::util::storable_builder::{EntropySource, StorableBuilder};
3636

3737
use crate::io::utils::check_namespace_key_validity;
38-
use crate::runtime::Runtime;
3938

4039
type 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

6058
impl 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

Comments
 (0)