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);
}