opencrabs 0.3.58

The autonomous, self-improving AI agent. Single Rust binary. Every channel. Install with: cargo install opencrabs
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

use crate::brain::tools::Tool;
use crate::brain::tools::ToolCapability;
use crate::brain::tools::ToolExecutionContext;
use crate::brain::tools::write::*;
use tempfile::TempDir;
use tokio;
use uuid::Uuid;

#[tokio::test]
async fn test_write_file() {
    let temp_dir = TempDir::new().unwrap();
    let file_path = temp_dir.path().join("test.txt");

    let tool = WriteTool;
    let session_id = Uuid::new_v4();
    let context =
        ToolExecutionContext::new(session_id).with_working_directory(temp_dir.path().to_path_buf());

    let input = serde_json::json!({
        "path": "test.txt",
        "content": "Hello, World!"
    });

    let result = tool.execute(input, &context).await.unwrap();
    assert!(result.success);

    // Verify file was written
    let contents = tokio::fs::read_to_string(&file_path).await.unwrap();
    assert_eq!(contents, "Hello, World!");
}

#[tokio::test]
async fn test_write_file_with_create_dirs() {
    let temp_dir = TempDir::new().unwrap();
    let file_path = temp_dir.path().join("subdir").join("test.txt");

    let tool = WriteTool;
    let session_id = Uuid::new_v4();
    let context =
        ToolExecutionContext::new(session_id).with_working_directory(temp_dir.path().to_path_buf());

    let input = serde_json::json!({
        "path": "subdir/test.txt",
        "content": "Nested file",
        "create_dirs": true
    });

    let result = tool.execute(input, &context).await.unwrap();
    assert!(result.success);

    // Verify file was written
    let contents = tokio::fs::read_to_string(&file_path).await.unwrap();
    assert_eq!(contents, "Nested file");
}

#[tokio::test]
async fn test_write_file_missing_parent_dir() {
    let temp_dir = TempDir::new().unwrap();

    let tool = WriteTool;
    let session_id = Uuid::new_v4();
    let context =
        ToolExecutionContext::new(session_id).with_working_directory(temp_dir.path().to_path_buf());

    let input = serde_json::json!({
        "path": "nonexistent/test.txt",
        "content": "Should fail",
        "create_dirs": false
    });

    let result = tool.execute(input, &context).await.unwrap();
    assert!(!result.success);
    assert!(result.error.is_some());
}

#[test]
fn test_write_tool_schema() {
    let tool = WriteTool;
    assert_eq!(tool.name(), "write_file");
    assert!(tool.requires_approval());

    let capabilities = tool.capabilities();
    assert!(capabilities.contains(&ToolCapability::WriteFiles));
    assert!(capabilities.contains(&ToolCapability::SystemModification));
}

#[tokio::test]
async fn test_overwrite_existing_file() {
    let temp_dir = TempDir::new().unwrap();
    let file_path = temp_dir.path().join("test.txt");

    // Write initial content
    tokio::fs::write(&file_path, "Initial content")
        .await
        .unwrap();

    let tool = WriteTool;
    let session_id = Uuid::new_v4();
    let context =
        ToolExecutionContext::new(session_id).with_working_directory(temp_dir.path().to_path_buf());

    let input = serde_json::json!({
        "path": "test.txt",
        "content": "New content"
    });

    let result = tool.execute(input, &context).await.unwrap();
    assert!(result.success);

    // Verify file was overwritten
    let contents = tokio::fs::read_to_string(&file_path).await.unwrap();
    assert_eq!(contents, "New content");
}