Skip to content

Commit fb40fe2

Browse files
committed
build fix
1 parent c7eda70 commit fb40fe2

File tree

1 file changed

+8
-5
lines changed

1 file changed

+8
-5
lines changed

rs/rosetta-api/icrc1/src/common/storage/storage_operations_test.rs

Lines changed: 8 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -419,23 +419,26 @@ fn test_fee_collector_resolution_and_repair() -> anyhow::Result<()> {
419419

420420
// Test fee collector resolution
421421
assert_eq!(
422-
get_fee_collector_from_block(&block1, &connection)?,
422+
get_107_fee_collector_or_legacy(&block1, &connection, None)?,
423423
Some(fee_collector_account)
424424
);
425425
assert_eq!(
426-
get_fee_collector_from_block(&block2, &connection)?,
426+
get_107_fee_collector_or_legacy(&block2, &connection, None)?,
427427
Some(fee_collector_account)
428428
);
429-
assert_eq!(get_fee_collector_from_block(&block3, &connection)?, None);
429+
assert_eq!(
430+
get_107_fee_collector_or_legacy(&block3, &connection, None)?,
431+
None
432+
);
430433

431434
// Test error cases (without storing invalid blocks in DB to avoid repair conflicts)
432435
let mut invalid_block = create_test_rosetta_block(999, 1000000003, &principal1, 400);
433436
invalid_block.block.fee_collector_block_index = Some(999); // Non-existent
434-
assert!(get_fee_collector_from_block(&invalid_block, &connection).is_err());
437+
assert!(get_107_fee_collector_or_legacy(&invalid_block, &connection, None).is_err());
435438

436439
let mut invalid_block2 = create_test_rosetta_block(998, 1000000004, &principal1, 500);
437440
invalid_block2.block.fee_collector_block_index = Some(3); // Block with no fee collector
438-
assert!(get_fee_collector_from_block(&invalid_block2, &connection).is_err());
441+
assert!(get_107_fee_collector_or_legacy(&invalid_block2, &connection, None).is_err());
439442

440443
// Test 2: Repair functionality with broken state simulation
441444
// Manually create broken balances (missing fee collector credits for block 2)

0 commit comments

Comments
 (0)