Skip to content

Commit ecdacd3

Browse files
committed
f Simplify test data generation logic
1 parent c6737a9 commit ecdacd3

File tree

1 file changed

+9
-20
lines changed

1 file changed

+9
-20
lines changed

lightning-persister/src/test_utils.rs

Lines changed: 9 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -66,30 +66,19 @@ pub(crate) fn do_test_data_migration<S: MigratableKVStore, T: MigratableKVStore>
6666
source_store: &mut S, target_store: &mut T,
6767
) {
6868
// We fill the source with some bogus keys.
69-
let primary_namespace_base = "testspace".to_string();
70-
let secondary_namespace_base = "testsubspace".to_string();
71-
let key_base = "testkey".to_string();
72-
let alphabet_len = KVSTORE_NAMESPACE_KEY_ALPHABET.len();
73-
7469
let dummy_data = [42u8; 32];
75-
76-
let num_primary_namespaces = 5;
77-
let num_secondary_namespaces = 5;
78-
let num_keys = 10;
70+
let num_primary_namespaces = 2;
71+
let num_secondary_namespaces = 2;
72+
let num_keys = 3;
7973
for i in 0..num_primary_namespaces {
80-
let primary_ext = KVSTORE_NAMESPACE_KEY_ALPHABET.chars().nth(i % alphabet_len).unwrap();
81-
let mut primary_namespace = primary_namespace_base.clone();
82-
primary_namespace.push(primary_ext);
74+
let primary_namespace =
75+
format!("testspace{}", KVSTORE_NAMESPACE_KEY_ALPHABET.chars().nth(i).unwrap());
8376
for j in 0..num_secondary_namespaces {
84-
let secondary_ext =
85-
KVSTORE_NAMESPACE_KEY_ALPHABET.chars().nth(j % alphabet_len).unwrap();
86-
let mut secondary_namespace = secondary_namespace_base.clone();
87-
secondary_namespace.push(secondary_ext);
77+
let secondary_namespace =
78+
format!("testsubspace{}", KVSTORE_NAMESPACE_KEY_ALPHABET.chars().nth(j).unwrap());
8879
for k in 0..num_keys {
89-
let key_ext = KVSTORE_NAMESPACE_KEY_ALPHABET.chars().nth(k % alphabet_len).unwrap();
90-
let mut key = key_base.clone();
91-
key.push(key_ext);
92-
80+
let key =
81+
format!("testkey{}", KVSTORE_NAMESPACE_KEY_ALPHABET.chars().nth(k).unwrap());
9382
source_store
9483
.write(&primary_namespace, &secondary_namespace, &key, &dummy_data)
9584
.unwrap();

0 commit comments

Comments
 (0)