ic-query 0.25.4

Internet Computer query library for NNS, SNS, ICRC, system canisters, and public network metadata
Documentation
use super::{fixtures::*, *};

#[test]
fn sns_canister_report_resolves_inventory_health_and_typed_gaps() {
    let report =
        build_sns_canister_report_with_source(&info_request("1"), &FixtureSnsCanisterSource)
            .expect("sns canister report");
    let text = sns_canister_report_text(&report);

    assert_eq!(report.schema_version, SNS_CANISTER_REPORT_SCHEMA_VERSION);
    assert_eq!(report.id, 1);
    assert_eq!(report.root_canister_id, ROOT_A);
    assert_eq!(report.inventory_method, "list_sns_canisters");
    assert_eq!(report.health_method, "get_sns_canisters_summary");
    assert_eq!(report.health_call_type, "ingress_update");
    assert!(!report.health_update_canister_list);
    assert!(!report.point_in_time_guaranteed);
    assert_eq!(report.canister_count, 2);
    assert_eq!(report.health_status_count, 1);
    assert_eq!(report.gap_count, 1);
    assert_eq!(report.canisters[0].role, SnsCanisterRole::Root);
    assert_eq!(report.canisters[1].role, SnsCanisterRole::Extension);
    assert_eq!(report.gaps[0].kind, SnsCanisterGapKind::HealthUnsupported);
    assert!(text.contains("health_call_type: ingress_update"));
    assert!(text.contains("health_update_canister_list: no"));
    assert!(text.contains("health_unsupported"));
}

#[test]
fn live_sns_canister_source_rejects_non_mainnet_before_agent_construction() {
    let request = SnsSourceRequest::new(
        "local",
        "not a valid endpoint",
        "2026-07-30T00:00:00Z",
        "test",
    );
    let sns = fixture_sns_a();

    let error = LiveSnsSource
        .fetch_sns_canisters(&request, &sns)
        .expect_err("non-mainnet must fail");

    assert!(matches!(
        error,
        SnsHostError::UnsupportedNetwork { network } if network == "local"
    ));
}

#[test]
fn sns_canister_report_rejects_invalid_custom_source_evidence() {
    for (mutate, expected_reason) in [
        (
            wrong_inventory_method as fn(&mut MainnetSnsCanisterInventory),
            "inventory_method",
        ),
        (
            enable_health_inventory_update,
            "health_update_canister_list",
        ),
        (invalidate_controller, "controller"),
    ] {
        let error = build_sns_canister_report_with_source(
            &info_request("1"),
            &MutatingSnsCanisterSource(mutate),
        )
        .expect_err("invalid custom Root inventory must fail");

        assert!(matches!(
            error,
            SnsHostError::InvalidSourceData {
                capability: "SNS Root canister inventory",
                reason,
            } if reason.contains(expected_reason)
        ));
    }
}

struct MutatingSnsCanisterSource(fn(&mut MainnetSnsCanisterInventory));

delegate_sns_discovery!(MutatingSnsCanisterSource);

impl SnsCanisterSource for MutatingSnsCanisterSource {
    fn fetch_sns_canisters(
        &self,
        request: &SnsSourceRequest,
        sns: &MainnetSns,
    ) -> Result<MainnetSnsCanisterInventory, SnsHostError> {
        let mut inventory = FixtureSnsCanisterSource.fetch_sns_canisters(request, sns)?;
        self.0(&mut inventory);
        Ok(inventory)
    }
}

fn wrong_inventory_method(inventory: &mut MainnetSnsCanisterInventory) {
    inventory.inventory_method = "wrong_method".to_string();
}

const fn enable_health_inventory_update(inventory: &mut MainnetSnsCanisterInventory) {
    inventory.health_update_canister_list = true;
}

fn invalidate_controller(inventory: &mut MainnetSnsCanisterInventory) {
    inventory.canisters[1].controllers = vec!["not a principal".to_string()];
}