use cw_orch::{
environment::TxHandler,
prelude::{ContractWrapper, Uploadable},
};
use mock_contract::{ExecuteMsg, InstantiateMsg, MigrateMsg, QueryMsg};
use cosmwasm_std::Event;
use cw_orch::prelude::{
ContractInstance, CwOrchExecute, CwOrchInstantiate, CwOrchMigrate, CwOrchQuery, CwOrchUpload,
Mock,
};
use cw_orch::interface;
#[interface(
InstantiateMsg,
ExecuteMsg,
QueryMsg,
MigrateMsg,
id = "test:mock_contract"
)]
pub struct MockContract;
impl<Chain> Uploadable for MockContract<Chain> {
fn wrapper() -> <Mock as TxHandler>::ContractSource {
Box::new(
ContractWrapper::new_with_empty(
mock_contract::execute,
mock_contract::instantiate,
mock_contract::query,
)
.with_migrate(mock_contract::migrate),
)
}
}
#[test]
fn test_instantiate() {
let contract = MockContract::new(Mock::new("Ghazshag"));
contract.upload().unwrap();
contract.instantiate(&InstantiateMsg {}, None, &[]).unwrap();
}
#[test]
fn test_execute() {
let contract = MockContract::new(Mock::new("Ghazshag"));
contract.upload().unwrap();
contract.instantiate(&InstantiateMsg {}, None, &[]).unwrap();
let response = contract.execute(&ExecuteMsg::FirstMessage {}, &[]).unwrap();
response.has_event(
&Event::new("wasm")
.add_attribute("_contract_addr", "contract0")
.add_attribute("action", "first message passed"),
);
contract
.execute(&ExecuteMsg::SecondMessage { t: "".to_string() }, &[])
.unwrap_err();
}
#[test]
fn test_query() {
let contract = MockContract::new(Mock::new("Ghazshag"));
contract.upload().unwrap();
contract.instantiate(&InstantiateMsg {}, None, &[]).unwrap();
let response: String = contract.query(&QueryMsg::FirstQuery {}).unwrap();
assert_eq!(response, "first query passed");
contract
.query::<String>(&QueryMsg::SecondQuery { t: "".to_string() })
.unwrap_err();
}
#[test]
fn test_migrate() {
let chain = Mock::new("Ghazshag");
let contract = MockContract::new(chain.clone());
contract.upload().unwrap();
contract
.instantiate(&InstantiateMsg {}, Some(&chain.sender_addr()), &[])
.unwrap();
contract
.migrate(
&MigrateMsg {
t: "error".to_string(),
},
contract.code_id().unwrap(),
)
.unwrap_err();
let response = contract
.migrate(
&MigrateMsg {
t: "success".to_string(),
},
contract.code_id().unwrap(),
)
.unwrap();
assert_eq!(response.events.len(), 1);
}