refine the spl-token domain test

Author: Stevengre

program/src/entrypoint-runtime-verification.rs

@@ -277,13 +277,15 @@ fn cheatcode_is_spl_rent(_: &AccountInfo) {}
 #[inline(never)]
 fn test_spltoken_domain_data(acc: &AccountInfo, mint: &AccountInfo, rent: &AccountInfo) {
     // Mutate mint via standard unpack/pack flow; use unwraps for brevity in tests
+    cheatcode_is_spl_mint(mint);
     let mut m = Mint::unpack_unchecked(&mint.data.borrow()).unwrap();
     m.is_initialized = true;
     Mint::pack(m, &mut mint.data.borrow_mut()).unwrap();
     let m2 = Mint::unpack(&mint.data.borrow()).unwrap();
     assert!(m2.is_initialized);
 
     // Set Account.is_native in the simplest way (parity with p-token's boolean set_native(true))
+    cheatcode_is_spl_account(acc);
     let mut a = Account::unpack_unchecked(&acc.data.borrow()).unwrap();
     a.is_native = solana_program_option::COption::Some(0);
     Account::pack(a, &mut acc.data.borrow_mut()).unwrap();
@@ -295,12 +297,16 @@ fn test_spltoken_domain_data(acc: &AccountInfo, mint: &AccountInfo, rent: &Accou
     let owner = acc.owner;
     assert_eq!(acc.owner, owner);
 
-    // Compare Rent from Sysvar::get vs account; fallback if not a real rent sysvar
+    // Compare Rent behavior using the sysvar getter and the provided account
     let sysrent = solana_rent::Rent::get().unwrap();
-    let min_a = sysrent.minimum_balance(10);
+    let rent_collected = 10;
+    let (sys_burnt, sys_distributed) = sysrent.calculate_burn(rent_collected);
+    assert!(sysrent.burn_percent > 100 || (sys_burnt <= rent_collected && sys_distributed <= rent_collected));
+
+    cheatcode_is_spl_rent(rent);
     let prent = solana_rent::Rent::from_account_info(rent).unwrap_or(sysrent);
-    let min_b = prent.minimum_balance(10);
-    assert_eq!(min_a, min_b);
+    let (acct_burnt, acct_distributed) = prent.calculate_burn(rent_collected);
+    assert!(prent.burn_percent > 100 || (acct_burnt <= rent_collected && acct_distributed <= rent_collected));
 }
 
 // wrapper to ensure the test is retained in SMIR/IR outputs