ic-query 0.89.0

use super::NnsInventoryCacheRequest;
use super::NnsInventoryError;
use super::NnsInventoryListRequest;
use super::NnsNodeOperatorListReport;
use super::NnsNodeOperatorRow;
use super::NnsNodeOperatorSource;
use super::build_nns_node_operator_list_report_with_source_and_progress;
use super::resolve_node_operator;
use crate::nns::LiveNnsSource;
use crate::report::CollectionSourceRequest;
use crate::subnet_catalog::MAINNET_NETWORK;
use crate::subnet_catalog::MAINNET_REGISTRY_CANISTER_ID;
use crate::test_support::temp_dir;

#[tokio::test(flavor = "current_thread")]
async fn live_node_operator_source_rejects_non_mainnet_before_agent_construction() {
    let request = CollectionSourceRequest::new(
        "local",
        "not a valid replica endpoint",
        "2026-07-29T00:00:00Z",
        "test",
    );

    let error = LiveNnsSource::default()
        .fetch_node_operator_list_report(&request)
        .await
        .expect_err("unsupported network");

    assert!(matches!(
        error,
        NnsInventoryError::UnsupportedNetwork { collection: crate::nns::NnsInventoryCollection::NodeOperator, network } if network == "local"
    ));
}

#[tokio::test(flavor = "current_thread")]
async fn node_operator_report_uses_live_registry_source() {
    let request = NnsInventoryListRequest {
        cache: test_cache_request(MAINNET_NETWORK, "uses-live-source"),
        source_endpoint: "https://icp-api.io".to_string(),
        now_unix_secs: 1_780_531_200,
    };
    let report = build_nns_node_operator_list_report_with_source_and_progress(
        &request,
        &FixtureNodeOperatorSource {
            node_operators: vec![NnsNodeOperatorRow {
                node_operator_principal: "aaaaa-aa".to_string(),
                node_provider_principal: "ryjl3-tyaaa-aaaaa-aaaba-cai".to_string(),
                node_allowance: 4,
                data_center_id: "dc1".to_string(),
                node_count: Some(3),
            }],
        },
        &mut |_| {},
    )
    .await
    .expect("node operator report");

    assert_eq!(report.schema_version, 1);
    assert_eq!(report.provenance.network, MAINNET_NETWORK);
    assert_eq!(
        report.provenance.registry_canister_id,
        MAINNET_REGISTRY_CANISTER_ID
    );
    assert_eq!(report.provenance.registry_version, 42);
    assert_eq!(report.provenance.fetched_at, "2026-06-04T00:00:00Z");
    assert_eq!(report.node_operators.len(), 1);
    assert_eq!(report.node_operators[0].node_operator_principal, "aaaaa-aa");
    assert_eq!(
        report.node_operators[0].node_provider_principal,
        "ryjl3-tyaaa-aaaaa-aaaba-cai"
    );
    assert_eq!(report.node_operators[0].node_allowance, 4);
    assert_eq!(report.node_operators[0].data_center_id, "dc1");
    assert_eq!(report.node_operators[0].node_count, Some(3));
}

#[test]
fn node_operator_info_resolves_unique_prefix() {
    let report = node_operator_report_fixture();

    let (index, resolved_from) = resolve_node_operator(&report, "ryjl").expect("prefix resolves");
    let operator = &report.node_operators[index];

    assert_eq!(resolved_from, "node_operator_principal_prefix");
    assert_eq!(
        operator.node_operator_principal,
        "ryjl3-tyaaa-aaaaa-aaaba-cai"
    );
}

fn node_operator_report_fixture() -> NnsNodeOperatorListReport {
    NnsNodeOperatorListReport {
        schema_version: 1,
        provenance: crate::nns::registry::NnsRegistryInventoryProvenance {
            network: MAINNET_NETWORK.to_string(),
            registry_canister_id: MAINNET_REGISTRY_CANISTER_ID.to_string(),
            registry_version: 42,
            fetched_at: "2026-06-04T00:00:00Z".to_string(),
            source_endpoint: "https://icp-api.io".to_string(),
            fetched_by: "test".to_string(),
        },
        node_operators: vec![NnsNodeOperatorRow {
            node_operator_principal: "ryjl3-tyaaa-aaaaa-aaaba-cai".to_string(),
            node_provider_principal: "aaaaa-aa".to_string(),
            node_allowance: 7,
            data_center_id: "dc1".to_string(),
            node_count: Some(13),
        }],
    }
}

fn test_cache_request(network: &str, name: &str) -> NnsInventoryCacheRequest {
    NnsInventoryCacheRequest {
        cache_root: temp_dir(&format!("ic-query-nns-node-operator-{name}")),
        network: network.to_string(),
    }
}

struct FixtureNodeOperatorSource {
    node_operators: Vec<NnsNodeOperatorRow>,
}

impl NnsNodeOperatorSource for FixtureNodeOperatorSource {
    fn fetch_node_operator_list_report<'a>(
        &'a self,
        request: &'a CollectionSourceRequest,
    ) -> crate::report::SourceFuture<'a, Result<NnsNodeOperatorListReport, NnsInventoryError>> {
        Box::pin(async move {
            let node_operators = self.node_operators.clone();
            Ok(NnsNodeOperatorListReport {
                schema_version: 1,
                provenance: crate::nns::registry::NnsRegistryInventoryProvenance {
                    network: MAINNET_NETWORK.to_string(),
                    registry_canister_id: MAINNET_REGISTRY_CANISTER_ID.to_string(),
                    registry_version: 42,
                    fetched_at: request.fetched_at.clone(),
                    source_endpoint: request.endpoint.clone(),
                    fetched_by: request.fetched_by.clone(),
                },
                node_operators,
            })
        })
    }
}