bvs_vault_router/
testing.rs

1#![cfg(not(target_arch = "wasm32"))]
2
3use crate::msg::{ExecuteMsg, InstantiateMsg, QueryMsg};
4use bvs_library::testing::TestingContract;
5use cosmwasm_std::{Addr, Empty, Env};
6use cw_multi_test::{App, Contract, ContractWrapper};
7use serde::{Deserialize, Serialize};
8
9#[derive(Serialize, Deserialize, Clone, Debug, PartialEq)]
10pub struct VaultRouterContract {
11    pub addr: Addr,
12    pub init: InstantiateMsg,
13}
14
15impl TestingContract<InstantiateMsg, ExecuteMsg, QueryMsg> for VaultRouterContract {
16    fn wrapper() -> Box<dyn Contract<Empty>> {
17        Box::new(ContractWrapper::new(
18            crate::contract::execute,
19            crate::contract::instantiate,
20            crate::contract::query,
21        ))
22    }
23
24    fn default_init(app: &mut App, _env: &Env) -> InstantiateMsg {
25        InstantiateMsg {
26            owner: app.api().addr_make("owner").to_string(),
27            registry: Self::get_contract_addr(app, "registry").to_string(),
28            pauser: Self::get_contract_addr(app, "pauser").to_string(),
29            guardrail: Self::get_contract_addr(app, "guardrail").to_string(),
30        }
31    }
32
33    fn new(app: &mut App, env: &Env, msg: Option<InstantiateMsg>) -> Self {
34        let init = msg.unwrap_or(Self::default_init(app, env));
35        let code_id = Self::store_code(app);
36        let addr = Self::instantiate(app, code_id, "vault_router", &init);
37        Self { addr, init }
38    }
39
40    fn addr(&self) -> &Addr {
41        &self.addr
42    }
43}