@@ -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