diff --git a/rs/ledger_suite/tests/sm-tests/src/in_memory_ledger.rs b/rs/ledger_suite/tests/sm-tests/src/in_memory_ledger.rs index b43d993772f9..97ca2f27a9b0 100644 --- a/rs/ledger_suite/tests/sm-tests/src/in_memory_ledger.rs +++ b/rs/ledger_suite/tests/sm-tests/src/in_memory_ledger.rs @@ -655,6 +655,7 @@ where ledger_id: CanisterId, num_ledger_blocks: u64, ) { + let actual_total_supply = parse_metric(env, ledger_id, "ledger_total_supply"); let actual_num_approvals = parse_metric(env, ledger_id, "ledger_num_approvals"); let actual_num_balances = parse_metric(env, ledger_id, "ledger_balance_store_entries"); println!( @@ -782,6 +783,7 @@ where "allowances with no expiration: {}, expiration in future: {}, expiration in past: {}", no_expiration_count, expiration_in_future_count, expiration_in_past_count ); + assert_eq!(self.total_supply.clone().into(), candid::Nat::from(actual_total_supply)); } } diff --git a/rs/ledger_suite/tests/sm-tests/src/lib.rs b/rs/ledger_suite/tests/sm-tests/src/lib.rs index 1b3102b92e05..abe9b777c4d5 100644 --- a/rs/ledger_suite/tests/sm-tests/src/lib.rs +++ b/rs/ledger_suite/tests/sm-tests/src/lib.rs @@ -2476,6 +2476,10 @@ pub fn test_bil_migration_fix( parse_metric(&env, ledger_id, "ledger_total_transactions"); println!("ledger_total_transactions: {}", ledger_total_transactions); assert_eq!(ledger_total_transactions, 100); + let ledger_total_supply = + parse_metric(&env, ledger_id, "ledger_total_supply"); + println!("ledger_total_supply: {}", ledger_total_supply); + assert_eq!(ledger_total_supply, 0); for i in 0..15 { send_transfer( &env, @@ -2525,12 +2529,8 @@ pub fn test_bil_migration_fix( // Test if the old serialized approvals and balances are correctly deserialized println!("testing the fix upgrade"); - test_upgrade(ledger_wasm_fix.clone(), 0, &mut in_memory_ledger); - // Test the new wasm serialization - println!("testing the fix upgrade again"); - test_upgrade(ledger_wasm_fix.clone(), 0, &mut in_memory_ledger); - // Test deserializing from memory manager - test_upgrade(ledger_wasm_fix.clone(), 0, &mut in_memory_ledger); + env.upgrade_canister(ledger_id, ledger_wasm_fix.clone(), upgrade_args.clone()) + .unwrap(); // Test upgrading to the latest official release test_upgrade(ledger_wasm_v4.clone(), 0, &mut in_memory_ledger); test_upgrade(ledger_wasm_v4.clone(), 0, &mut in_memory_ledger);