Check total supply

dfinity · Jan 18, 2025 · 522bb48 · 522bb48
1 parent 58c4dd1
commit 522bb48
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 6 deletions.
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
@@ -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
@@ -2476,6 +2476,10 @@ pub fn test_bil_migration_fix<Tokens>(
                         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<Tokens>(
 
                 // 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);