use abstract_integration_tests::{create_default_account, AResult};
use abstract_interface::{Abstract, AccountI, AccountQueryFns};
use abstract_std::IBC_CLIENT;
use anyhow::bail;
use cw_orch::prelude::*;

pub fn ibc_client_installed<Chain: CwEnv>(account: &AccountI<Chain>) -> AResult {
    let ibc_addr = account.module_addresses(vec![IBC_CLIENT.to_string()])?;
    if ibc_addr.modules.is_empty() {
        bail!("IBC client not installed")
    }
    Ok(())
}

#[test]
fn throws_if_enabling_when_already_enabled() -> AResult {
    let chain = MockBech32::new("mock");
    let sender = chain.sender_addr();
    let abstr = Abstract::deploy_on_mock(chain.clone())?;
    let account = create_default_account(&sender, &abstr)?;

    account.set_ibc_status(true)?;
    let res = account.set_ibc_status(true);

    assert!(res.is_err());

    Ok(())
}

#[test]
fn throws_if_disabling_without_ibc_client_installed() -> AResult {
    let chain = MockBech32::new("mock");
    let sender = chain.sender_addr();
    let abstr = Abstract::deploy_on_mock(chain.clone())?;
    let account = create_default_account(&sender, &abstr)?;

    let res = account.set_ibc_status(false);

    assert!(res.is_err());

    Ok(())
}

#[test]
fn can_update_ibc_settings() -> AResult {
    let chain = MockBech32::new("mock");
    let sender = chain.sender_addr();
    let abstr = Abstract::deploy_on_mock(chain.clone())?;
    let account = create_default_account(&sender, &abstr)?;

    ibc_client_installed(&account).unwrap_err();
    account.set_ibc_status(true)?;
    ibc_client_installed(&account)?;
    account.set_ibc_status(false)?;
    ibc_client_installed(&account).unwrap_err();

    Ok(())
}