ic-memory 0.36.3

Durable stable-memory allocation governance for Internet Computer canisters
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
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79

use ic_memory::RuntimeMemory;
use ic_stable_structures::DefaultMemoryImpl;
use std::cell::RefCell;

const MACRO_AUTHORITY: &str = "runtime_macros";

fn snapshot() -> ic_memory::SealedDeclarationSnapshot {
    ic_memory::SealedDeclarationSnapshot::new(vec![
        ic_memory::MemoryRequest::new(
            MACRO_AUTHORITY,
            "macro.logical.rows.v1",
            ic_memory::SchemaMetadata::default(),
        )
        .unwrap(),
        ic_memory::MemoryRequest::new(
            MACRO_AUTHORITY,
            "macro.integration.users.v1",
            ic_memory::SchemaMetadata::default(),
        )
        .unwrap(),
    ])
    .unwrap()
}

thread_local! {
    static MACRO_MEMORY: RefCell<Option<RuntimeMemory<DefaultMemoryImpl>>> = {
        assert!(
            ic_memory::is_default_memory_manager_bootstrapped()
                .expect("default runtime lifecycle")
        );
        RefCell::new(Some(ic_memory::open_default_memory_manager_memory("macro.integration.users.v1")
        .expect("committed macro memory")))
    };
}

fn bootstrap_and_require_thread_local_ledger() {
    let validated =
        ic_memory::bootstrap_default_memory_manager(&snapshot(), &pool()).expect("bootstrap");
    ic_memory::default_memory_manager_diagnostic_export()
        .expect("bootstrapped runtime should expose its local ledger");

    assert!(
        validated
            .declarations()
            .iter()
            .any(|declaration| declaration.stable_key().as_str() == "macro.integration.users.v1")
    );
    let logical = ic_memory::open_default_memory_manager_memory("macro.logical.rows.v1")
        .expect("key-only host adoption");
    assert_eq!(ic_stable_structures::Memory::size(&logical), 0);
    assert_eq!(
        validated
            .slot_for(&ic_memory::StableKey::parse("macro.logical.rows.v1").unwrap())
            .unwrap()
            .id(),
        11
    );
    MACRO_MEMORY.with(|memory| assert!(memory.borrow().is_some()));
    ic_memory::open_default_memory_manager_memory("macro.integration.users.v1")
        .expect("open macro memory");
}

#[test]
fn first_libtest_default_runtime_bootstraps_its_own_memory() {
    bootstrap_and_require_thread_local_ledger();
}

#[test]
fn second_libtest_default_runtime_bootstraps_its_own_memory() {
    bootstrap_and_require_thread_local_ledger();
}

fn pool() -> ic_memory::MemoryAllocationPool {
    ic_memory::MemoryAllocationPool::new(
        vec![ic_memory::MemoryAuthority::new("runtime_macros", "macro.").unwrap()],
        vec![],
    )
    .unwrap()
}