ic-query 0.55.4

use super::{
    NnsDataCenterHostError, NnsDataCenterListReport, NnsDataCenterRow, NnsDataCenterSource,
    NnsInventoryCacheRequest, NnsInventoryListRequest,
    build_nns_data_center_list_report_with_source, nns_data_center_list_report_text,
    nns_data_center_list_report_verbose_text, resolve_data_center,
};
use crate::ic_registry::MainnetDataCenter;
use crate::nns::{LiveNnsSource, NnsSourceRequest};
use crate::subnet_catalog::{MAINNET_NETWORK, MAINNET_REGISTRY_CANISTER_ID};
use crate::test_support::temp_dir;

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

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

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

#[tokio::test(flavor = "current_thread")]
async fn data_center_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_data_center_list_report_with_source(
        &request,
        &FixtureDataCenterSource {
            data_centers: vec![data_center_fixture()],
        },
    )
    .await
    .expect("data-center report");

    assert_eq!(report.schema_version, 1);
    assert_eq!(report.network, MAINNET_NETWORK);
    assert_eq!(report.registry_canister_id, MAINNET_REGISTRY_CANISTER_ID);
    assert_eq!(report.registry_version, 42);
    assert_eq!(report.fetched_at, "2026-06-04T00:00:00Z");
    assert_eq!(report.data_center_count, 1);
    assert_eq!(report.data_centers[0].data_center_id, "dc1");
}

#[test]
fn data_center_text_renders_relation_counts() {
    let report = data_center_report_fixture();

    let text = nns_data_center_list_report_text(&report);
    let verbose_text = nns_data_center_list_report_verbose_text(&report);

    assert!(text.contains("data_centers: ic count 1"));
    assert!(text.contains("OPS"));
    assert!(text.contains("PROVIDERS"));
    assert!(text.contains("NODES"));
    assert!(text.contains("dc1"));
    assert!(text.contains("fetched_at 2026-06-04T00:00:00Z\n\nDC"));
    assert!(verbose_text.contains("fetched_by: test\n\nDC"));
}

#[test]
fn data_center_info_resolves_unique_prefix() {
    let report = data_center_report_fixture();

    let (data_center, resolved_from) = resolve_data_center(&report, "dc").expect("prefix resolves");

    assert_eq!(resolved_from, "data_center_id_prefix");
    assert_eq!(data_center.data_center_id, "dc1");
}

fn data_center_report_fixture() -> NnsDataCenterListReport {
    NnsDataCenterListReport {
        schema_version: 1,
        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(),
        data_center_count: 1,
        data_centers: vec![NnsDataCenterRow {
            data_center_id: "dc1".to_string(),
            region: "eu-west".to_string(),
            owner: "example owner".to_string(),
            latitude: Some(48.8566),
            longitude: Some(2.3522),
            node_operator_count: 2,
            node_provider_count: 1,
            node_count: 13,
        }],
    }
}

fn data_center_fixture() -> MainnetDataCenter {
    MainnetDataCenter {
        id: "dc1".to_string(),
        region: "eu-west".to_string(),
        owner: "example owner".to_string(),
        latitude: Some(48.8566),
        longitude: Some(2.3522),
        node_operator_count: 2,
        node_provider_count: 1,
        node_count: 13,
    }
}

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

struct FixtureDataCenterSource {
    data_centers: Vec<MainnetDataCenter>,
}

impl NnsDataCenterSource for FixtureDataCenterSource {
    fn fetch_data_center_list_report<'a>(
        &'a self,
        request: &'a NnsSourceRequest,
    ) -> crate::report::SourceFuture<'a, Result<NnsDataCenterListReport, NnsDataCenterHostError>>
    {
        Box::pin(async move {
            let data_centers = self
                .data_centers
                .iter()
                .map(|data_center| NnsDataCenterRow {
                    data_center_id: data_center.id.clone(),
                    region: data_center.region.clone(),
                    owner: data_center.owner.clone(),
                    latitude: data_center.latitude,
                    longitude: data_center.longitude,
                    node_operator_count: data_center.node_operator_count,
                    node_provider_count: data_center.node_provider_count,
                    node_count: data_center.node_count,
                })
                .collect::<Vec<_>>();
            Ok(NnsDataCenterListReport {
                schema_version: 1,
                network: MAINNET_NETWORK.to_string(),
                registry_canister_id: MAINNET_REGISTRY_CANISTER_ID.to_string(),
                registry_version: 42,
                fetched_at: request.fetched_at.clone(),
                fetched_by: request.fetched_by.clone(),
                source_endpoint: request.endpoint.clone(),
                data_center_count: data_centers.len(),
                data_centers,
            })
        })
    }
}