Skip to content

Commit 491bc6c

Browse files
reduce duplicate code
1 parent 909701c commit 491bc6c

1 file changed

Lines changed: 34 additions & 44 deletions

File tree

chain/src/application.rs

Lines changed: 34 additions & 44 deletions
Original file line numberDiff line numberDiff line change
@@ -246,6 +246,33 @@ mod tests {
246246
async fn retain(&mut self, _: impl Fn(&Self::Key) -> bool + Send + 'static) {}
247247
}
248248

249+
fn test_archive_config<C>(
250+
prefix: &str,
251+
label: &str,
252+
page_cache: CacheRef,
253+
codec_config: C,
254+
) -> immutable::Config<C> {
255+
immutable::Config {
256+
metadata_partition: format!("{prefix}-{label}-metadata"),
257+
freezer_table_partition: format!("{prefix}-{label}-freezer-table"),
258+
freezer_table_initial_size: 64,
259+
freezer_table_resize_frequency: 10,
260+
freezer_table_resize_chunk_size: 10,
261+
freezer_key_partition: format!("{prefix}-{label}-key"),
262+
freezer_key_page_cache: page_cache,
263+
freezer_value_partition: format!("{prefix}-{label}-value"),
264+
freezer_value_target_size: 1024,
265+
freezer_value_compression: None,
266+
ordinal_partition: format!("{prefix}-{label}-ordinal"),
267+
items_per_section: NZU64!(10),
268+
codec_config,
269+
replay_buffer: NZUsize!(1024),
270+
freezer_key_write_buffer: NZUsize!(1024),
271+
freezer_value_write_buffer: NZUsize!(1024),
272+
ordinal_write_buffer: NZUsize!(1024),
273+
}
274+
}
275+
249276
async fn init_mailbox(
250277
context: deterministic::Context,
251278
scheme: Scheme,
@@ -262,55 +289,18 @@ mod tests {
262289

263290
let finalizations_by_height = immutable::Archive::init(
264291
context.with_label("finalizations_by_height"),
265-
immutable::Config {
266-
metadata_partition: format!("{partition_prefix}-finalizations-by-height-metadata"),
267-
freezer_table_partition: format!(
268-
"{partition_prefix}-finalizations-by-height-freezer-table"
269-
),
270-
freezer_table_initial_size: 64,
271-
freezer_table_resize_frequency: 10,
272-
freezer_table_resize_chunk_size: 10,
273-
freezer_key_partition: format!("{partition_prefix}-finalizations-by-height-key"),
274-
freezer_key_page_cache: page_cache.clone(),
275-
freezer_value_partition: format!(
276-
"{partition_prefix}-finalizations-by-height-value"
277-
),
278-
freezer_value_target_size: 1024,
279-
freezer_value_compression: None,
280-
ordinal_partition: format!("{partition_prefix}-finalizations-by-height-ordinal"),
281-
items_per_section: NZU64!(10),
282-
codec_config: Scheme::certificate_codec_config_unbounded(),
283-
replay_buffer: NZUsize!(1024),
284-
freezer_key_write_buffer: NZUsize!(1024),
285-
freezer_value_write_buffer: NZUsize!(1024),
286-
ordinal_write_buffer: NZUsize!(1024),
287-
},
292+
test_archive_config(
293+
partition_prefix,
294+
"finalizations-by-height",
295+
page_cache.clone(),
296+
Scheme::certificate_codec_config_unbounded(),
297+
),
288298
)
289299
.await
290300
.expect("failed to initialize finalizations archive");
291301
let finalized_blocks = immutable::Archive::init(
292302
context.with_label("finalized_blocks"),
293-
immutable::Config {
294-
metadata_partition: format!("{partition_prefix}-finalized-blocks-metadata"),
295-
freezer_table_partition: format!(
296-
"{partition_prefix}-finalized-blocks-freezer-table"
297-
),
298-
freezer_table_initial_size: 64,
299-
freezer_table_resize_frequency: 10,
300-
freezer_table_resize_chunk_size: 10,
301-
freezer_key_partition: format!("{partition_prefix}-finalized-blocks-key"),
302-
freezer_key_page_cache: page_cache.clone(),
303-
freezer_value_partition: format!("{partition_prefix}-finalized-blocks-value"),
304-
freezer_value_target_size: 1024,
305-
freezer_value_compression: None,
306-
ordinal_partition: format!("{partition_prefix}-finalized-blocks-ordinal"),
307-
items_per_section: NZU64!(10),
308-
codec_config: (),
309-
replay_buffer: NZUsize!(1024),
310-
freezer_key_write_buffer: NZUsize!(1024),
311-
freezer_value_write_buffer: NZUsize!(1024),
312-
ordinal_write_buffer: NZUsize!(1024),
313-
},
303+
test_archive_config(partition_prefix, "finalized-blocks", page_cache.clone(), ()),
314304
)
315305
.await
316306
.expect("failed to initialize finalized blocks archive");

0 commit comments

Comments
 (0)