canic-host 0.100.3

//! Module: component_topology::tests
//!
//! Responsibility: verify host finalization of immutable Fleet root topology bindings.
//! Does not own: network discovery, root creation, release-set construction, or installation.
//! Boundary: exercises canonicalization and Fleet-scoped admission/placement invariants.

use canic_core::{
    bootstrap::{compiled::ConfigModel, parse_config_model},
    cdk::types::Cycles,
    ids::{
        AppId, CanonicalNetworkId, CyclesFundingBudget, FleetBinding, FleetCoordinatorBinding,
        FleetId, FleetKey, FleetRegistryAuthority, FleetSubnetRootLimits, SubnetId,
    },
};

use super::*;

const CONFIG: &str = r#"
[app]
name = "toko"

[roles.root]
kind = "root"
package = "root"

[roles.database]
kind = "canister"
package = "database"

[roles.user_hub]
kind = "canister"
package = "user_hub"

[component_specs.database]
component_role = "database"
maximum_instances = 2

[component_specs.users]
component_role = "user_hub"
maximum_instances = 3
"#;

fn config() -> ConfigModel {
    parse_config_model(CONFIG).expect("valid topology config")
}

fn authority(fleet_byte: u8) -> FleetRegistryAuthority {
    FleetRegistryAuthority {
        binding: FleetCoordinatorBinding {
            fleet: FleetBinding {
                fleet: FleetKey {
                    canonical_network_id: CanonicalNetworkId::public_ic(),
                    fleet_id: FleetId::from_generated_bytes([fleet_byte; 32]),
                },
                app: AppId::from("toko"),
            },
            coordinator_subnet: SubnetId::from_principal(Principal::from_slice(&[1; 29])),
            coordinator: Principal::from_slice(&[2; 29]),
        },
        epoch: 1,
    }
}

fn limits() -> FleetSubnetRootLimits {
    FleetSubnetRootLimits {
        maximum_component_instances: 10,
        maximum_managed_canisters: 1_000,
        maximum_registry_bytes: 4_194_304,
        maximum_wasm_store_bytes: 40_000_000,
        cycles_funding: CyclesFundingBudget {
            window_secs: 3_600,
            maximum_cycles: Cycles::new(1_000_000_000_000_000),
        },
    }
}

fn admission(component_spec: &str, maximum_root_instances: u32) -> RootComponentAdmissionInput {
    RootComponentAdmissionInput {
        component_spec: component_spec.parse().expect("Component Spec ID"),
        maximum_root_instances,
    }
}

fn root(
    root_byte: u8,
    subnet_byte: u8,
    component_admissions: Vec<RootComponentAdmissionInput>,
) -> FleetSubnetRootTopologyInput {
    FleetSubnetRootTopologyInput {
        placement_subnet: SubnetId::from_principal(Principal::from_slice(&[subnet_byte; 29])),
        fleet_subnet_root: Principal::from_slice(&[root_byte; 29]),
        component_admissions,
        limits: limits(),
    }
}

#[test]
fn planner_derives_hashes_and_root_digests_from_canonical_config() {
    let plan = plan_fleet_topology(
        &config(),
        authority(3),
        vec![
            root(4, 5, vec![admission("users", 1), admission("database", 1)]),
            root(6, 7, vec![admission("users", 2), admission("database", 1)]),
        ],
    )
    .expect("valid Fleet topology plan");

    assert_eq!(plan.fleet_subnet_roots.len(), 2);
    for root in &plan.fleet_subnet_roots {
        assert_eq!(
            root.component_admissions
                .iter()
                .map(|admission| admission.component_spec.as_str())
                .collect::<Vec<_>>(),
            vec!["database", "users"],
        );
        let projection = plan
            .component_topology
            .project_for_admissions(&root.component_admissions)
            .expect("root projection");
        assert_eq!(
            root.component_topology_digest,
            projection.digest().expect("root digest"),
        );
    }
}

#[test]
fn planner_rejects_duplicate_unknown_zero_and_excess_admissions() {
    let config = config();

    std::assert_matches!(
        plan_fleet_topology(
            &config,
            authority(3),
            vec![root(
                4,
                5,
                vec![admission("users", 1), admission("users", 1)],
            )],
        ),
        Err(FleetTopologyPlanError::DuplicateAdmission { .. })
    );

    std::assert_matches!(
        plan_fleet_topology(
            &config,
            authority(3),
            vec![root(4, 5, vec![admission("unknown", 1)])],
        ),
        Err(FleetTopologyPlanError::UnknownComponentSpec { .. })
    );

    std::assert_matches!(
        plan_fleet_topology(
            &config,
            authority(3),
            vec![root(
                4,
                5,
                vec![admission("database", 1), admission("users", 0)],
            )],
        ),
        Err(FleetTopologyPlanError::Topology(
            canic_core::bootstrap::compiled::ComponentTopologyError::ZeroRootAdmission { .. }
        ))
    );

    std::assert_matches!(
        plan_fleet_topology(
            &config,
            authority(3),
            vec![root(
                4,
                5,
                vec![admission("database", 1), admission("users", 4)],
            )],
        ),
        Err(FleetTopologyPlanError::Topology(
            canic_core::bootstrap::compiled::ComponentTopologyError::RootAdmissionExceedsFleetMaximum { .. }
        ))
    );
}

#[test]
fn one_fleet_rejects_duplicate_subnet_but_different_fleets_may_reuse_it() {
    let config = config();
    let first = root(4, 5, vec![admission("database", 1), admission("users", 1)]);
    let second_same_subnet = root(6, 5, vec![admission("database", 1), admission("users", 2)]);

    std::assert_matches!(
        plan_fleet_topology(
            &config,
            authority(3),
            vec![first, second_same_subnet],
        ),
        Err(FleetTopologyPlanError::Topology(
            canic_core::bootstrap::compiled::ComponentTopologyError::DuplicateFleetSubnetRootSubnet { .. }
        ))
    );

    for fleet_byte in [3, 8] {
        plan_fleet_topology(
            &config,
            authority(fleet_byte),
            vec![root(
                fleet_byte + 10,
                5,
                vec![admission("database", 2), admission("users", 3)],
            )],
        )
        .expect("different Fleet may independently use the same physical Subnet");
    }
}

#[test]
fn planner_binds_the_authority_to_the_configured_app() {
    let mut wrong = authority(3);
    wrong.binding.fleet.app = AppId::from("another_app");

    std::assert_matches!(
        plan_fleet_topology(
            &config(),
            wrong,
            vec![root(
                4,
                5,
                vec![admission("database", 2), admission("users", 3)],
            )],
        ),
        Err(FleetTopologyPlanError::AppMismatch { .. })
    );
}