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

use crate::db::Database;
use crate::db::models::File;
use crate::db::models::Session;
use crate::db::repository::SessionRepository;
use crate::db::repository::file::*;
use std::path::PathBuf;
use tokio;

#[tokio::test]
async fn test_file_crud() {
    let db = Database::connect_in_memory()
        .await
        .expect("Failed to create database");
    db.run_migrations().await.expect("Failed to run migrations");
    let session_repo = SessionRepository::new(db.pool().clone());
    let file_repo = FileRepository::new(db.pool().clone());

    // Create session first
    let session = Session::new(Some("Test".to_string()), Some("model".to_string()), None);
    session_repo
        .create(&session)
        .await
        .expect("Failed to create session");

    // Create file
    let file = File::new(session.id, PathBuf::from("/test/file.rs"), None);
    file_repo
        .create(&file)
        .await
        .expect("Failed to create file");

    // Read
    let found = file_repo.find_by_id(file.id).await.expect("Failed to find");
    assert!(found.is_some());
    assert_eq!(found.as_ref().unwrap().path, PathBuf::from("/test/file.rs"));

    // Update
    let mut updated = file.clone();
    updated.content = Some("Updated content".to_string());
    file_repo.update(&updated).await.expect("Failed to update");

    let found = file_repo.find_by_id(file.id).await.expect("Failed to find");
    assert_eq!(found.unwrap().content, Some("Updated content".to_string()));

    // Delete
    file_repo.delete(file.id).await.expect("Failed to delete");
    let found = file_repo.find_by_id(file.id).await.expect("Failed to find");
    assert!(found.is_none());
}

#[tokio::test]
async fn test_file_list_by_session() {
    let db = Database::connect_in_memory()
        .await
        .expect("Failed to create database");
    db.run_migrations().await.expect("Failed to run migrations");
    let session_repo = SessionRepository::new(db.pool().clone());
    let file_repo = FileRepository::new(db.pool().clone());

    let session = Session::new(Some("Test".to_string()), Some("model".to_string()), None);
    session_repo
        .create(&session)
        .await
        .expect("Failed to create session");

    // Create multiple files
    for i in 0..3 {
        let file = File::new(
            session.id,
            PathBuf::from(format!("/test/file{}.rs", i)),
            None,
        );
        file_repo
            .create(&file)
            .await
            .expect("Failed to create file");
    }

    let files = file_repo
        .list_by_session(session.id)
        .await
        .expect("Failed to list");
    assert_eq!(files.len(), 3);

    let count = file_repo
        .count_by_session(session.id)
        .await
        .expect("Failed to count");
    assert_eq!(count, 3);
}