flowcode-core 0.4.3-alpha

Core execution engine for FlowCode data scripting language
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
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

// duckdb select verb integration tests
#![cfg(feature = "duckdb")]

use std::io::Write;
use tempfile::{Builder, TempPath};

use flowcode_core::duckdb_session::DuckSession;
use flowcode_core::types::TypedValue;
use flowcode_core::ast::ArgValue;
use flowcode_core::duckdb_verbs::map_verb_to_sql;

fn create_test_csv(content: &str) -> (TempPath, String) {
    let mut temp_file = Builder::new()
        .prefix("test_")
        .suffix(".csv")
        .tempfile()
        .unwrap();
    writeln!(temp_file, "{}", content).unwrap();
    let temp_path = temp_file.into_temp_path(); // close handle, keep file
    let path_str = temp_path.to_str().unwrap().to_string();
    (temp_path, path_str)
}

#[test]
fn test_select_all_columns() {
    let csv_content = "region,sales\nEast,1200\nWest,800";
    let (_guard, file_path) = create_test_csv(csv_content);
    let session = DuckSession::new().unwrap();
    let table_name = "test_sales_select_all";

    let load_result = session.load_csv_auto(table_name, &file_path);
    assert!(load_result.is_ok(), "Failed to load CSV: {:?}", load_result.err());

    let args = vec![TypedValue::new(
        flowcode_core::types::ValueKind::String,
        ArgValue::String(table_name.to_string())
    )];
    
    let result = map_verb_to_sql("select", &args, &session);
    assert!(result.is_ok());
    let sql_query = result.unwrap();
    assert_eq!(sql_query, format!("SELECT * FROM {}", table_name));

    let query_result = session.execute_sql(&sql_query);
    assert!(query_result.is_ok(), "Select query failed: {:?}", query_result.err());

    if let Ok(TypedValue { value: ArgValue::Table(data), .. }) = query_result {
        assert_eq!(data.len(), 2, "Expected 2 rows from select");
        assert_eq!(data[0].len(), 2, "Expected 2 columns from select");
    } else {
        panic!("Expected table data in select result");
    }
}

#[test]
fn test_select_specific_columns() {
    let csv_content = "region,sales,category\nEast,1200,A\nWest,800,B";
    let (_guard, file_path) = create_test_csv(csv_content);
    let session = DuckSession::new().unwrap();
    let table_name = "test_sales_select_columns";

    let load_result = session.load_csv_auto(table_name, &file_path);
    assert!(load_result.is_ok(), "Failed to load CSV: {:?}", load_result.err());

    let args = vec![
        TypedValue::new(
            flowcode_core::types::ValueKind::String,
            ArgValue::String(table_name.to_string())
        ),
        TypedValue::new(
            flowcode_core::types::ValueKind::String,
            ArgValue::String("region".to_string())
        ),
        TypedValue::new(
            flowcode_core::types::ValueKind::String,
            ArgValue::String("sales".to_string())
        )
    ];
    
    let result = map_verb_to_sql("select", &args, &session);
    assert!(result.is_ok());
    let sql_query = result.unwrap();
    assert_eq!(sql_query, format!("SELECT region, sales FROM {}", table_name));

    let query_result = session.execute_sql(&sql_query);
    assert!(query_result.is_ok(), "Select query failed: {:?}", query_result.err());

    if let Ok(TypedValue { value: ArgValue::Table(data), .. }) = query_result {
        assert_eq!(data.len(), 2, "Expected 2 rows from select");
        assert_eq!(data[0].len(), 2, "Expected 2 columns from select");
    } else {
        panic!("Expected table data in select result");
    }
}

#[test]
fn test_select_no_arguments() {
    let session = DuckSession::new().unwrap();
    let args = vec![];
    
    let result = map_verb_to_sql("select", &args, &session);
    assert!(result.is_ok());
    let sql_query = result.unwrap();
    assert_eq!(sql_query, "SELECT * FROM employees");
}