Unverified Commit 0efeaf80 authored by Andrew Jones's avatar Andrew Jones Committed by GitHub

[examples] Add test for runtime storage (#369)

parent 2b796bd6
Pipeline #84680 passed with stages
in 9 minutes and 12 seconds
...@@ -34,7 +34,7 @@ mod runtime { ...@@ -34,7 +34,7 @@ mod runtime {
/// All balance information for an account, mirroring the structure defined in the runtime. /// All balance information for an account, mirroring the structure defined in the runtime.
/// Copied from [substrate](https://github.com/paritytech/substrate/blob/2c87fe171bc341755a43a3b32d67560469f8daac/frame/system/src/lib.rs#L307) /// Copied from [substrate](https://github.com/paritytech/substrate/blob/2c87fe171bc341755a43a3b32d67560469f8daac/frame/system/src/lib.rs#L307)
#[derive(Decode)] #[derive(Encode, Decode)]
pub struct AccountData { pub struct AccountData {
free: Balance, free: Balance,
_reserved: Balance, _reserved: Balance,
...@@ -44,7 +44,7 @@ mod runtime { ...@@ -44,7 +44,7 @@ mod runtime {
/// Information of an account, mirroring the structure defined in the runtime /// Information of an account, mirroring the structure defined in the runtime
/// Copied from [substrate](https://github.com/paritytech/substrate/blob/2c87fe171bc341755a43a3b32d67560469f8daac/frame/system/src/lib.rs#L307) /// Copied from [substrate](https://github.com/paritytech/substrate/blob/2c87fe171bc341755a43a3b32d67560469f8daac/frame/system/src/lib.rs#L307)
#[derive(Decode)] #[derive(Encode, Decode)]
pub struct AccountInfo { pub struct AccountInfo {
_nonce: u32, _nonce: u32,
_refcount: u8, _refcount: u8,
...@@ -111,6 +111,7 @@ mod runtime { ...@@ -111,6 +111,7 @@ mod runtime {
#[cfg(all(test))] #[cfg(all(test))]
mod tests { mod tests {
use super::*; use super::*;
use ink_core::env;
#[test] #[test]
fn non_existent_account_returns_zero() { fn non_existent_account_returns_zero() {
...@@ -118,5 +119,41 @@ mod runtime { ...@@ -118,5 +119,41 @@ mod runtime {
let account: AccountId = [0u8; 32].into(); let account: AccountId = [0u8; 32].into();
assert_eq!(contract.get_balance(account), 0); assert_eq!(contract.get_balance(account), 0);
} }
#[test]
fn returns_account_balance_from_storage() {
let contract = RuntimeStorage::new();
let account: AccountId = [0u8; 32].into();
let balance = 1_000_000;
let account_info = AccountInfo {
data: AccountData {
free: balance,
_reserved: 0,
_fee_frozen: 0,
_misc_frozen: 0,
},
_nonce: 0,
_refcount: 0,
};
let encoded_account = &account.encode();
let mut key = vec![
// Precomputed: Twox128("System")
38, 170, 57, 78, 234, 86, 48, 224, 124, 72, 174, 12, 149, 88, 206, 247,
// Precomputed: Twox128("Account")
185, 157, 136, 14, 198, 129, 121, 156, 12, 243, 14, 136, 134, 55, 29, 169,
];
let mut blake2_128 = Blake2x128::from(Vec::new());
let hashed_account = blake2_128.hash_raw(&encoded_account);
key.extend_from_slice(&hashed_account);
key.extend_from_slice(&encoded_account);
env::test::set_runtime_storage(&key, account_info);
assert_eq!(contract.get_balance(account), balance);
}
} }
} }
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment