Skip to content

Commit 07fba55

Browse files
committed
refactor: adapt cardano_immutable_digester tests to test list_immutable_files_to_process function directly
1 parent be89d59 commit 07fba55

File tree

1 file changed

+10
-31
lines changed

1 file changed

+10
-31
lines changed

mithril-common/src/digesters/cardano_immutable_digester.rs

Lines changed: 10 additions & 31 deletions
Original file line numberDiff line numberDiff line change
@@ -288,20 +288,15 @@ mod tests {
288288
#[tokio::test]
289289
async fn fail_if_no_file_in_folder() {
290290
let immutable_db = db_builder("fail_if_no_file_in_folder").build();
291-
let digester =
292-
CardanoImmutableDigester::new("devnet".to_string(), None, TestLogger::stdout());
293-
let beacon = CardanoDbBeacon::new(1, 1);
294291

295-
let result = digester
296-
.compute_digest(&immutable_db.dir, &beacon)
297-
.await
298-
.expect_err("compute_digest should have failed");
292+
let result = list_immutable_files_to_process(&immutable_db.dir, 1)
293+
.expect_err("list_immutable_files_to_process should have failed");
299294

300295
assert_eq!(
301296
format!(
302297
"{:?}",
303298
ImmutableDigesterError::NotEnoughImmutable {
304-
expected_number: beacon.immutable_file_number,
299+
expected_number: 1,
305300
found_number: None,
306301
db_dir: immutable_db.dir,
307302
}
@@ -315,35 +310,24 @@ mod tests {
315310
let immutable_db = db_builder("fail_if_no_immutable_exist")
316311
.with_non_immutables(&["not_immutable"])
317312
.build();
318-
let digester =
319-
CardanoImmutableDigester::new("devnet".to_string(), None, TestLogger::stdout());
320-
let beacon = CardanoDbBeacon::new(1, 1);
321313

322-
assert!(digester
323-
.compute_digest(&immutable_db.dir, &beacon)
324-
.await
325-
.is_err());
314+
assert!(list_immutable_files_to_process(&immutable_db.dir, 1).is_err());
326315
}
327316

328317
#[tokio::test]
329318
async fn fail_if_theres_only_the_uncompleted_immutable_trio() {
330319
let immutable_db = db_builder("fail_if_theres_only_the_uncompleted_immutable_trio")
331320
.append_immutable_trio()
332321
.build();
333-
let digester =
334-
CardanoImmutableDigester::new("devnet".to_string(), None, TestLogger::stdout());
335-
let beacon = CardanoDbBeacon::new(1, 1);
336322

337-
let result = digester
338-
.compute_digest(&immutable_db.dir, &beacon)
339-
.await
340-
.expect_err("compute_digest should've failed");
323+
let result = list_immutable_files_to_process(&immutable_db.dir, 1)
324+
.expect_err("list_immutable_files_to_process should've failed");
341325

342326
assert_eq!(
343327
format!(
344328
"{:?}",
345329
ImmutableDigesterError::NotEnoughImmutable {
346-
expected_number: beacon.immutable_file_number,
330+
expected_number: 1,
347331
found_number: None,
348332
db_dir: immutable_db.dir,
349333
}
@@ -358,20 +342,15 @@ mod tests {
358342
.with_immutables(&[1, 2, 3, 4, 5])
359343
.append_immutable_trio()
360344
.build();
361-
let digester =
362-
CardanoImmutableDigester::new("devnet".to_string(), None, TestLogger::stdout());
363-
let beacon = CardanoDbBeacon::new(1, 10);
364345

365-
let result = digester
366-
.compute_digest(&immutable_db.dir, &beacon)
367-
.await
368-
.expect_err("compute_digest should've failed");
346+
let result = list_immutable_files_to_process(&immutable_db.dir, 10)
347+
.expect_err("list_immutable_files_to_process should've failed");
369348

370349
assert_eq!(
371350
format!(
372351
"{:?}",
373352
ImmutableDigesterError::NotEnoughImmutable {
374-
expected_number: beacon.immutable_file_number,
353+
expected_number: 10,
375354
found_number: Some(5),
376355
db_dir: immutable_db.dir,
377356
}

0 commit comments

Comments
 (0)