cryochamber 0.1.2

A hibernation chamber for AI agents — schedule, wake, and manage long-running agent tasks
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

// tests/daemon_tests.rs
use cryochamber::daemon::RetryState;

#[test]
fn test_retry_state_provider_rotation_advances() {
    let mut retry = RetryState::new(5, 3); // 3 providers
    assert_eq!(retry.provider_index, 0);

    let wrapped = retry.rotate_provider();
    assert!(!wrapped);
    assert_eq!(retry.provider_index, 1);

    let wrapped = retry.rotate_provider();
    assert!(!wrapped);
    assert_eq!(retry.provider_index, 2);
}

#[test]
fn test_retry_state_provider_rotation_wraps() {
    let mut retry = RetryState::new(5, 3);

    retry.rotate_provider(); // 0 -> 1
    retry.rotate_provider(); // 1 -> 2
    let wrapped = retry.rotate_provider(); // 2 -> 0
    assert_eq!(retry.provider_index, 0);
    assert!(wrapped); // signals cycle complete
}

#[test]
fn test_retry_state_rotation_resets_attempt() {
    let mut retry = RetryState::new(5, 3);
    retry.record_failure();
    retry.record_failure();
    assert_eq!(retry.attempt, 2);

    retry.rotate_provider();
    assert_eq!(retry.attempt, 0); // reset on rotation
    assert_eq!(retry.provider_index, 1);
}

#[test]
fn test_retry_state_reset_clears_provider_index() {
    let mut retry = RetryState::new(5, 3);
    retry.rotate_provider();
    retry.rotate_provider();
    assert_eq!(retry.provider_index, 2);

    retry.reset();
    assert_eq!(retry.attempt, 0);
    assert_eq!(retry.provider_index, 0);
}

#[test]
fn test_retry_state_single_provider_no_rotation() {
    let mut retry = RetryState::new(5, 1);
    let wrapped = retry.rotate_provider();
    assert_eq!(retry.provider_index, 0);
    assert!(wrapped); // immediately wraps with 1 provider
}

#[test]
fn test_retry_state_no_providers() {
    let mut retry = RetryState::new(5, 0);
    let wrapped = retry.rotate_provider();
    assert_eq!(retry.provider_index, 0);
    assert!(wrapped);
}