crabtalk 0.0.21

Crabtalk library
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

//! MCP CRUD surface — Storage::{list,load,upsert,delete}_mcp on FsStorage.

use crabtalk::storage::FsStorage;
use std::collections::BTreeMap;
use wcore::{McpServerConfig, storage::Storage};

fn fs_storage(config_dir: std::path::PathBuf) -> FsStorage {
    let sessions = config_dir.join("sessions");
    FsStorage::new(config_dir, sessions, Vec::new())
}

fn mcp(name: &str, command: &str) -> McpServerConfig {
    McpServerConfig {
        name: name.to_owned(),
        command: command.to_owned(),
        args: vec!["--flag".into()],
        env: BTreeMap::from([("KEY".into(), "value".into())]),
        auto_restart: true,
        url: None,
        auth: false,
    }
}

#[test]
fn empty_storage_lists_no_mcps() {
    let dir = tempfile::tempdir().unwrap();
    let storage = fs_storage(dir.path().to_path_buf());
    assert!(storage.list_mcps().unwrap().is_empty());
    assert!(storage.load_mcp("nonexistent").unwrap().is_none());
}

#[test]
fn upsert_then_load_round_trip() {
    let dir = tempfile::tempdir().unwrap();
    let storage = fs_storage(dir.path().to_path_buf());

    let server = mcp("playwright", "npx");
    storage.upsert_mcp(&server).unwrap();

    let loaded = storage.load_mcp("playwright").unwrap().unwrap();
    assert_eq!(loaded.name, "playwright");
    assert_eq!(loaded.command, "npx");
    assert_eq!(loaded.args, vec!["--flag".to_string()]);
    assert_eq!(loaded.env.get("KEY").map(String::as_str), Some("value"));
}

#[test]
fn upsert_overwrites_existing_entry() {
    let dir = tempfile::tempdir().unwrap();
    let storage = fs_storage(dir.path().to_path_buf());

    storage.upsert_mcp(&mcp("srv", "v1")).unwrap();
    storage.upsert_mcp(&mcp("srv", "v2")).unwrap();
    assert_eq!(storage.load_mcp("srv").unwrap().unwrap().command, "v2");
}

#[test]
fn upsert_rejects_invalid_names() {
    let dir = tempfile::tempdir().unwrap();
    let storage = fs_storage(dir.path().to_path_buf());

    for bad in [
        "",
        "with.dot",
        "br[acket",
        "br]acket",
        "qu\"ote",
        "ctrl\nchar",
    ] {
        let cfg = mcp(bad, "cmd");
        assert!(
            storage.upsert_mcp(&cfg).is_err(),
            "upsert_mcp accepted bad name: {bad:?}"
        );
    }
}

#[test]
fn delete_returns_existed_flag() {
    let dir = tempfile::tempdir().unwrap();
    let storage = fs_storage(dir.path().to_path_buf());

    storage.upsert_mcp(&mcp("srv", "cmd")).unwrap();
    assert!(storage.delete_mcp("srv").unwrap());
    assert!(!storage.delete_mcp("srv").unwrap());
    assert!(storage.load_mcp("srv").unwrap().is_none());
}

#[test]
fn list_returns_all_mcps_sorted() {
    let dir = tempfile::tempdir().unwrap();
    let storage = fs_storage(dir.path().to_path_buf());

    storage.upsert_mcp(&mcp("zeta", "z")).unwrap();
    storage.upsert_mcp(&mcp("alpha", "a")).unwrap();

    let names: Vec<_> = storage.list_mcps().unwrap().into_keys().collect();
    assert_eq!(names, vec!["alpha".to_string(), "zeta".to_string()]);
}

#[test]
fn mcps_persist_across_storage_instances() {
    let dir = tempfile::tempdir().unwrap();
    {
        let storage = fs_storage(dir.path().to_path_buf());
        storage.upsert_mcp(&mcp("persisted", "cmd")).unwrap();
    }
    let storage = fs_storage(dir.path().to_path_buf());
    let loaded = storage.load_mcp("persisted").unwrap().unwrap();
    assert_eq!(loaded.command, "cmd");
}

#[test]
fn mcp_serialized_under_mcps_table() {
    let dir = tempfile::tempdir().unwrap();
    let storage = fs_storage(dir.path().to_path_buf());

    storage.upsert_mcp(&mcp("srv", "cmd")).unwrap();

    let body = std::fs::read_to_string(dir.path().join("local").join("settings.toml")).unwrap();
    assert!(body.contains("[mcps.srv]"));
}