ink_sandbox 5.1.1

Sandbox runtime environment for ink! e2e tests
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82

use crate::{
    AccountIdFor,
    Sandbox,
};
use frame_support::{
    sp_runtime::DispatchError,
    traits::fungible::Mutate,
};

type BalanceOf<R> = <R as pallet_balances::Config>::Balance;

/// Balance API for the sandbox.
pub trait BalanceAPI<T: Sandbox>
where
    T: Sandbox,
    T::Runtime: pallet_balances::Config,
{
    /// Mint tokens to an account.
    ///
    /// # Arguments
    ///
    /// * `address` - The address of the account to add tokens to.
    /// * `amount` - The number of tokens to add.
    fn mint_into(
        &mut self,
        address: &AccountIdFor<T::Runtime>,
        amount: BalanceOf<T::Runtime>,
    ) -> Result<BalanceOf<T::Runtime>, DispatchError>;

    /// Return the free balance of an account.
    ///
    /// # Arguments
    ///
    /// * `address` - The address of the account to query.
    fn free_balance(
        &mut self,
        address: &AccountIdFor<T::Runtime>,
    ) -> BalanceOf<T::Runtime>;
}

impl<T> BalanceAPI<T> for T
where
    T: Sandbox,
    T::Runtime: pallet_balances::Config,
{
    fn mint_into(
        &mut self,
        address: &AccountIdFor<T::Runtime>,
        amount: BalanceOf<T::Runtime>,
    ) -> Result<BalanceOf<T::Runtime>, DispatchError> {
        self.execute_with(|| {
            pallet_balances::Pallet::<T::Runtime>::mint_into(address, amount)
        })
    }

    fn free_balance(
        &mut self,
        address: &AccountIdFor<T::Runtime>,
    ) -> BalanceOf<T::Runtime> {
        self.execute_with(|| pallet_balances::Pallet::<T::Runtime>::free_balance(address))
    }
}

#[cfg(test)]
mod test {
    use super::*;
    use crate::DefaultSandbox;
    #[test]
    fn mint_works() {
        let mut sandbox = DefaultSandbox::default();
        let balance = sandbox.free_balance(&DefaultSandbox::default_actor());

        sandbox
            .mint_into(&DefaultSandbox::default_actor(), 100)
            .unwrap();

        assert_eq!(
            sandbox.free_balance(&DefaultSandbox::default_actor()),
            balance + 100
        );
    }
}