kota 0.1.3

A lightweight, highly extensible ai code agent, built in Rust.
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

use super::create_temp_dir;
use kota::kota_code::tools::write_file::{WriteFileArgs, WriteFileTool};
use kota::kota_code::tools::FileToolError;
use rig::tool::Tool;
use std::fs;

#[tokio::test]
async fn test_write_new_file() {
    let temp_dir = create_temp_dir();
    let file_path = temp_dir
        .path()
        .join("new_file.txt")
        .to_string_lossy()
        .to_string();
    let content = "Hello, world!\nThis is new content.";

    let tool = WriteFileTool;
    let args = WriteFileArgs {
        file_path: file_path.clone(),
        content: content.to_string(),
    };

    let result: Result<_, FileToolError> = tool.call(args).await;

    assert!(result.is_ok());
    let output = result.unwrap();
    assert!(output.success);
    assert_eq!(output.file_path, file_path);
    assert_eq!(output.bytes_written, content.len() as u64);
    assert!(output.message.contains("Successfully wrote"));

    // Verify file was actually written
    let written_content = fs::read_to_string(&file_path).unwrap();
    assert_eq!(written_content, content);
}

#[tokio::test]
async fn test_overwrite_existing_file() {
    let temp_dir = create_temp_dir();
    let file_path = temp_dir
        .path()
        .join("existing.txt")
        .to_string_lossy()
        .to_string();

    // Create initial file
    fs::write(&file_path, "Original content").unwrap();

    let new_content = "New content that replaces the old";
    let tool = WriteFileTool;
    let args = WriteFileArgs {
        file_path: file_path.clone(),
        content: new_content.to_string(),
    };

    let result: Result<_, FileToolError> = tool.call(args).await;

    assert!(result.is_ok());
    let output = result.unwrap();
    assert!(output.success);
    assert_eq!(output.bytes_written, new_content.len() as u64);

    // Verify file was overwritten
    let written_content = fs::read_to_string(&file_path).unwrap();
    assert_eq!(written_content, new_content);
}