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

use super::{create_temp_dir, create_test_file};
use kota::kota_code::tools::read_file::{ReadFileArgs, ReadFileTool};
use kota::kota_code::tools::FileToolError;
use rig::tool::Tool;

#[tokio::test]
async fn test_read_existing_file() {
    let temp_dir = create_temp_dir();
    let content = "Hello, world!\nThis is a test file.";
    let file_path = create_test_file(temp_dir.path(), "test.txt", content);

    let tool = ReadFileTool;
    let args = ReadFileArgs {
        file_path: file_path.clone(),
    };

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

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

#[tokio::test]
async fn test_read_nonexistent_file() {
    let tool = ReadFileTool;
    let args = ReadFileArgs {
        file_path: "nonexistent_file.txt".to_string(),
    };

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

    assert!(result.is_err());
    match result.unwrap_err() {
        FileToolError::FileNotFound(path) => {
            assert_eq!(path, "nonexistent_file.txt");
        }
        _ => panic!("Expected FileNotFound error"),
    }
}