sql-cli 1.73.1

SQL query tool for CSV/JSON with both interactive TUI and non-interactive CLI modes - perfect for exploration and automation
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
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146

/// Test that `ViewportManager` is correctly synced with the current buffer
/// when loading multiple files
use sql_cli::buffer::{Buffer, BufferAPI, BufferManager};
use sql_cli::data::data_view::DataView;
use sql_cli::data::datatable::{DataColumn, DataRow, DataTable, DataValue};
use std::sync::Arc;

#[test]
fn test_viewport_sync_after_multi_file_load() {
    // Create mock DataTables with distinct data
    let mut datatable1 = DataTable::new("table1");
    datatable1.add_column(DataColumn::new("id"));
    datatable1.add_column(DataColumn::new("name"));
    datatable1
        .add_row(DataRow::new(vec![
            DataValue::String("1".to_string()),
            DataValue::String("first_buffer".to_string()),
        ]))
        .unwrap();
    datatable1
        .add_row(DataRow::new(vec![
            DataValue::String("2".to_string()),
            DataValue::String("first_buffer".to_string()),
        ]))
        .unwrap();
    let dataview1 = DataView::new(Arc::new(datatable1));

    let mut datatable2 = DataTable::new("table2");
    datatable2.add_column(DataColumn::new("code"));
    datatable2.add_column(DataColumn::new("desc"));
    datatable2
        .add_row(DataRow::new(vec![
            DataValue::String("A".to_string()),
            DataValue::String("second_buffer".to_string()),
        ]))
        .unwrap();
    datatable2
        .add_row(DataRow::new(vec![
            DataValue::String("B".to_string()),
            DataValue::String("second_buffer".to_string()),
        ]))
        .unwrap();
    let dataview2 = DataView::new(Arc::new(datatable2));

    // Create a BufferManager and add buffers
    let mut buffer_manager = BufferManager::new();

    // Create first buffer with dataview1
    let mut buffer1 = Buffer::new(1);
    buffer1.set_dataview(Some(dataview1.clone()));
    buffer1.set_name("file1.csv".to_string());
    buffer_manager.add_buffer(buffer1);

    // Create second buffer with dataview2
    let mut buffer2 = Buffer::new(2);
    buffer2.set_dataview(Some(dataview2.clone()));
    buffer2.set_name("file2.csv".to_string());
    buffer_manager.add_buffer(buffer2);

    // Switch to buffer 2 (simulating what happens when adding files)
    buffer_manager.switch_to(1);
    assert_eq!(buffer_manager.current_index(), 1, "Should be on buffer 2");

    // Switch back to buffer 1 (simulating what run_enhanced_tui_multi does)
    buffer_manager.switch_to(0);
    assert_eq!(
        buffer_manager.current_index(),
        0,
        "Should be on buffer 1 after switch"
    );

    // Verify the current buffer has the correct dataview
    let current_buffer = buffer_manager.current().unwrap();
    let buffer_dataview = current_buffer.get_dataview().unwrap();

    assert_eq!(
        buffer_dataview.column_names(),
        vec!["id", "name"],
        "Buffer 1 should have first dataview headers"
    );

    assert_eq!(
        buffer_dataview.row_count(),
        2,
        "Buffer 1 should have 2 rows"
    );
}

#[test]
fn test_buffer_switching_preserves_dataview() {
    // Create two DataTables with different data
    let mut datatable1 = DataTable::new("table1");
    datatable1.add_column(DataColumn::new("col1"));
    datatable1
        .add_row(DataRow::new(vec![DataValue::String(
            "buffer1_data".to_string(),
        )]))
        .unwrap();
    let dataview1 = DataView::new(Arc::new(datatable1));

    let mut datatable2 = DataTable::new("table2");
    datatable2.add_column(DataColumn::new("col2"));
    datatable2
        .add_row(DataRow::new(vec![DataValue::String(
            "buffer2_data".to_string(),
        )]))
        .unwrap();
    let dataview2 = DataView::new(Arc::new(datatable2));

    // Create BufferManager
    let mut buffer_manager = BufferManager::new();

    // Add both buffers
    let mut buffer1 = Buffer::new(1);
    buffer1.set_dataview(Some(dataview1.clone()));
    buffer1.set_name("test1.csv".to_string());
    buffer_manager.add_buffer(buffer1);

    let mut buffer2 = Buffer::new(2);
    buffer2.set_dataview(Some(dataview2.clone()));
    buffer2.set_name("test2.csv".to_string());
    buffer_manager.add_buffer(buffer2);

    // Switch between buffers multiple times and verify data is preserved
    for _ in 0..3 {
        // Switch to buffer 1
        buffer_manager.switch_to(0);
        let current = buffer_manager.current().unwrap();
        let dataview = current.get_dataview().unwrap();
        assert_eq!(
            dataview.column_names(),
            vec!["col1"],
            "Buffer 1 should have col1 header"
        );

        // Switch to buffer 2
        buffer_manager.switch_to(1);
        let current = buffer_manager.current().unwrap();
        let dataview = current.get_dataview().unwrap();
        assert_eq!(
            dataview.column_names(),
            vec!["col2"],
            "Buffer 2 should have col2 header"
        );
    }
}