jorm 0.1.1

A lightweight DAG execution engine with natural language processing
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

use super::generator::NlpProcessor;

#[tokio::test]
async fn test_nlp_processor_creation() {
    let processor = NlpProcessor::new().await;
    assert!(processor.is_ok());
}

#[tokio::test]
async fn test_simple_shell_command_generation() {
    let processor = NlpProcessor::new().await.unwrap();
    let description = "run command \"echo hello world\"";

    let result = processor.generate_dag(description).await;
    assert!(result.is_ok());

    let dag_content = result.unwrap();
    assert!(dag_content.contains("task_1"));
    assert!(dag_content.contains("type: shell"));
    assert!(dag_content.contains("echo hello world"));
}

#[tokio::test]
async fn test_file_copy_generation() {
    let processor = NlpProcessor::new().await.unwrap();
    let description = "copy file1.txt to file2.txt";

    let result = processor.generate_dag(description).await;
    assert!(result.is_ok());

    let dag_content = result.unwrap();
    assert!(dag_content.contains("type: file_copy"));
    assert!(dag_content.contains("file1.txt"));
    assert!(dag_content.contains("file2.txt"));
}

#[tokio::test]
async fn test_dag_preview_generation() {
    let processor = NlpProcessor::new().await.unwrap();
    let description = "run command \"ls -la\" then copy file1.txt to backup/file1.txt";

    let result = processor.generate_dag_with_preview(description).await;
    assert!(result.is_ok());

    let preview = result.unwrap();
    assert_eq!(preview.original_description, description);
    // The task count might be 1 if only one task is parsed correctly
    assert!(preview.task_count >= 1);
    assert!(preview.task_names.contains(&"task_1".to_string()));
}

#[tokio::test]
async fn test_validation_of_generated_dag() {
    let processor = NlpProcessor::new().await.unwrap();
    let valid_dag = r#"
task test_task {
    type: shell
    command: "echo test"
}
"#;

    let result = processor.validate_generated_dag(valid_dag);
    assert!(result.is_ok());
}