canic 0.19.4

Canic — a canister orchestration and management toolkit for the Internet Computer
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35

// Category C - Artifact / deployment test (embedded config).
// This test relies on embedded production config by design.

use canic::{
    Error,
    cdk::types::Principal,
    dto::{topology::SubnetRegistryEntry, topology::SubnetRegistryResponse},
    protocol,
};
use canic_internal::canister::SCALE;
use canic_testkit::pic::Pic;

/// Create a worker canister via the given hub canister.
pub fn create_worker(pic: &Pic, hub_pid: Principal) -> Result<Principal, Error> {
    let result: Result<Result<Principal, Error>, Error> =
        pic.update_call(hub_pid, "create_worker", ());

    result.map_err(|err| Error::internal(format!("create_worker transport failed: {err}")))?
}

/// Count worker canisters registered under a given parent.
pub fn count_workers(pic: &Pic, root_id: Principal, parent_pid: Principal) -> usize {
    let registry: Result<SubnetRegistryResponse, Error> = pic
        .query_call(root_id, protocol::CANIC_SUBNET_REGISTRY, ())
        .expect("query subnet registry transport");
    let SubnetRegistryResponse(registry): SubnetRegistryResponse =
        registry.expect("query subnet registry application");

    registry
        .iter()
        .filter(|entry: &&SubnetRegistryEntry| {
            entry.role == SCALE && entry.record.parent_pid == Some(parent_pid)
        })
        .count()
}