sabiql 1.12.4

A fast, driver-less TUI for browsing and editing PostgreSQL databases
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

use super::*;
use harness::{explorer_selected_state, table_detail_loaded_state, with_current_result};
use sabiql_app::model::shared::ui_state::FocusMode;
use sabiql_domain::{QueryResult, QuerySource};

fn adhoc_result(query: &str) -> QueryResult {
    QueryResult::success(
        query.to_string(),
        vec!["count".to_string()],
        vec![vec!["42".to_string()]],
        5,
        QuerySource::Adhoc,
    )
}

#[test]
fn preview_with_history_hint() {
    let mut state = table_detail_loaded_state();
    let mut terminal = create_test_terminal();

    // Current result is Preview, but history has adhoc entries
    with_current_result(&mut state);
    state.query.push_history(Arc::new(adhoc_result("SELECT 1")));
    state.ui.focused_pane = FocusedPane::Result;

    let output = render_to_string(&mut terminal, &mut state);

    insta::assert_snapshot!(output);
}

#[test]
fn result_pane_history_mode() {
    let mut state = explorer_selected_state();
    let mut terminal = create_test_terminal();

    // Push 3 adhoc results
    for i in 1..=3 {
        state
            .query
            .push_history(Arc::new(adhoc_result(&format!("SELECT {i}"))));
    }
    state
        .query
        .set_current_result(Arc::new(adhoc_result("SELECT 3")));
    state.query.enter_history(1); // viewing 2/3
    state.ui.focused_pane = FocusedPane::Result;

    let output = render_to_string(&mut terminal, &mut state);

    insta::assert_snapshot!(output);
}

fn wide_adhoc_result(query: &str) -> QueryResult {
    QueryResult::success(
        query.to_string(),
        (1..=10).map(|i| format!("column_{i}")).collect(),
        vec![(1..=10).map(|i| format!("value_{i}")).collect()],
        12,
        QuerySource::Adhoc,
    )
}

#[test]
fn history_mode_with_horizontal_scroll() {
    let mut state = explorer_selected_state();
    let mut terminal = create_test_terminal();

    let long_query = "SELECT column_1, column_2, column_3, column_4, column_5 FROM very_long_table_name WHERE id > 100";
    for i in 1..=3 {
        state
            .query
            .push_history(Arc::new(wide_adhoc_result(&format!("SELECT {i}"))));
    }
    state
        .query
        .set_current_result(Arc::new(wide_adhoc_result(long_query)));
    state.query.enter_history(2); // viewing 3/3
    state.ui.focus_mode = FocusMode::focused(state.ui.focused_pane);

    let output = render_to_string(&mut terminal, &mut state);

    insta::assert_snapshot!(output);
}

#[test]
fn result_query_with_history_hint() {
    let mut state = explorer_selected_state();
    let mut terminal = create_test_terminal();

    // Push history but do NOT enter history mode (history_index = None)
    for i in 1..=2 {
        state
            .query
            .push_history(Arc::new(adhoc_result(&format!("SELECT {i}"))));
    }
    state
        .query
        .set_current_result(Arc::new(adhoc_result("SELECT 2")));
    state.ui.focused_pane = FocusedPane::Result;

    let output = render_to_string(&mut terminal, &mut state);

    insta::assert_snapshot!(output);
}

#[test]
fn focus_mode_history_mode() {
    let mut state = explorer_selected_state();
    let mut terminal = create_test_terminal();

    for i in 1..=3 {
        state
            .query
            .push_history(Arc::new(adhoc_result(&format!("SELECT {i}"))));
    }
    state
        .query
        .set_current_result(Arc::new(adhoc_result("SELECT 3")));
    state.query.enter_history(0); // viewing 1/3
    state.ui.focus_mode = FocusMode::focused(state.ui.focused_pane);

    let output = render_to_string(&mut terminal, &mut state);

    insta::assert_snapshot!(output);
}