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

use crate::prelude::*;
use serde::{Deserialize, Serialize};

#[derive(Serialize, Deserialize, Clone, Debug, PartialEq, JsonSchema)]
pub struct MockEnv {
    pub(crate) sender: HumanAddr,
    pub(crate) contract: ContractLink<HumanAddr>,
    pub(crate) sent_funds: Vec<Coin>
}

impl MockEnv {
    /// Default values are what `cosmwasm_std::testing::mock_env` returns.
    pub fn new(sender: impl Into<HumanAddr>, contract: ContractLink<HumanAddr>) -> Self {
        Self {
            sender: sender.into(),
            contract,
            sent_funds: vec![]
        }
    }

    pub fn sent_funds(mut self, funds: Vec<Coin>) -> Self {
        self.sent_funds = funds;

        self
    }
}