codex-helper-core 0.15.0

Core library for codex-helper.
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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126

use super::*;

#[test]
fn rename_route_node_updates_references_and_entry_name() {
    let mut routing = RoutingConfigV4 {
        entry: "main".to_string(),
        routes: BTreeMap::from([
            (
                "main".to_string(),
                RoutingNodeV4 {
                    strategy: RoutingPolicyV4::OrderedFailover,
                    children: vec!["pool".to_string(), "paygo".to_string()],
                    ..RoutingNodeV4::default()
                },
            ),
            (
                "pool".to_string(),
                RoutingNodeV4 {
                    strategy: RoutingPolicyV4::OrderedFailover,
                    children: vec!["alpha".to_string()],
                    ..RoutingNodeV4::default()
                },
            ),
        ]),
        ..RoutingConfigV4::default()
    };

    routing
        .rename_route_node("pool", "monthly_pool".to_string())
        .expect("rename should succeed");

    assert_eq!(routing.entry, "main");
    assert!(routing.routes.contains_key("monthly_pool"));
    assert!(!routing.routes.contains_key("pool"));
    assert_eq!(
        routing.entry_node().map(|node| node.children.as_slice()),
        Some(&["monthly_pool".to_string(), "paygo".to_string()][..])
    );
}

#[test]
fn delete_route_node_rejects_entry_and_referenced_nodes() {
    let mut routing = RoutingConfigV4 {
        entry: "main".to_string(),
        routes: BTreeMap::from([
            (
                "main".to_string(),
                RoutingNodeV4 {
                    strategy: RoutingPolicyV4::OrderedFailover,
                    children: vec!["pool".to_string()],
                    ..RoutingNodeV4::default()
                },
            ),
            (
                "pool".to_string(),
                RoutingNodeV4 {
                    strategy: RoutingPolicyV4::OrderedFailover,
                    children: vec!["alpha".to_string()],
                    ..RoutingNodeV4::default()
                },
            ),
            (
                "unused".to_string(),
                RoutingNodeV4 {
                    strategy: RoutingPolicyV4::OrderedFailover,
                    children: vec!["alpha".to_string()],
                    ..RoutingNodeV4::default()
                },
            ),
        ]),
        ..RoutingConfigV4::default()
    };

    let err = routing
        .delete_route_node("pool")
        .expect_err("referenced node should not delete");
    assert!(err.to_string().contains("referenced"));

    let err = routing
        .delete_route_node("main")
        .expect_err("entry node should not delete");
    assert!(err.to_string().contains("entry route node"));

    routing
        .delete_route_node("unused")
        .expect("unreferenced node should delete");
    assert!(!routing.routes.contains_key("unused"));
}

#[test]
fn sync_graph_from_compat_ignores_default_compat_fields_for_existing_graph() {
    let mut routing = RoutingConfigV4 {
        entry: "main".to_string(),
        routes: BTreeMap::from([(
            "main".to_string(),
            RoutingNodeV4 {
                strategy: RoutingPolicyV4::TagPreferred,
                children: vec!["monthly".to_string(), "paygo".to_string()],
                prefer_tags: vec![BTreeMap::from([(
                    "billing".to_string(),
                    "monthly".to_string(),
                )])],
                on_exhausted: RoutingExhaustedActionV4::Stop,
                ..RoutingNodeV4::default()
            },
        )]),
        ..RoutingConfigV4::default()
    };

    routing.sync_graph_from_compat();

    let entry = routing.entry_node().expect("entry node should remain");
    assert_eq!(entry.strategy, RoutingPolicyV4::TagPreferred);
    assert_eq!(
        entry.children,
        vec!["monthly".to_string(), "paygo".to_string()]
    );
    assert_eq!(entry.on_exhausted, RoutingExhaustedActionV4::Stop);
    assert_eq!(
        entry.prefer_tags,
        vec![BTreeMap::from([(
            "billing".to_string(),
            "monthly".to_string()
        )])]
    );
}