zippa-db 0.1.1

A fast, lightweight, cross-platform database client for PostgreSQL, MySQL, and SQLite.
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

//! Transactions in a query tab: its own connection, the status bar's word
//! for an open one, Commit and Roll Back, and asking before one is thrown
//! away.

use super::*;

#[gpui_kit::test]
fn a_transaction_lasts_between_runs_and_commit_ends_it(cx: &mut TestAppContext) {
    let (database, handle, session) = workspace_session(cx);

    run_in_tab(cx, &handle, &session, "BEGIN");
    assert_eq!(transaction(cx, &session), TxnState::Open);
    run_in_tab(
        cx,
        &handle,
        &session,
        "insert into items values (3, 'gamma', 3.0, NULL)",
    );
    assert_eq!(
        transaction(cx, &session),
        TxnState::Open,
        "the next run is on the same connection, inside the same transaction"
    );
    assert_eq!(
        runtime::block_on(other_count(&database)),
        2,
        "nothing is visible elsewhere before the commit"
    );

    click_workspace(cx, &handle, "transaction-commit");
    assert_eq!(transaction(cx, &session), TxnState::Idle);
    assert_eq!(runtime::block_on(other_count(&database)), 3);
}

#[gpui_kit::test]
fn roll_back_in_the_status_bar_undoes_the_transaction(cx: &mut TestAppContext) {
    let (database, handle, session) = workspace_session(cx);

    run_in_tab(cx, &handle, &session, "BEGIN");
    run_in_tab(cx, &handle, &session, "delete from items");
    click_workspace(cx, &handle, "transaction-rollback");

    assert_eq!(transaction(cx, &session), TxnState::Idle);
    assert_eq!(runtime::block_on(other_count(&database)), 2);
}

#[gpui_kit::test]
fn closing_a_tab_with_an_open_transaction_asks_and_rolls_it_back(cx: &mut TestAppContext) {
    let (database, handle, session) = workspace_session(cx);

    run_in_tab(cx, &handle, &session, "BEGIN");
    run_in_tab(
        cx,
        &handle,
        &session,
        "insert into items values (3, 'gamma', 3.0, NULL)",
    );
    assert!(!session.read_with(cx, |session, cx| session.tab_is_dirty_for_test(0, cx)));

    close_workspace_tab(cx, &handle, &session, 0);
    assert!(
        workspace_dialog_open(cx, &handle),
        "a tab with an open transaction should be asked about before it closes"
    );

    // Keeping it open keeps the transaction.
    click_workspace(cx, &handle, "cancel");
    assert_eq!(transaction(cx, &session), TxnState::Open);

    close_workspace_tab(cx, &handle, &session, 0);
    click_workspace(cx, &handle, "ok");
    cx.run_until_parked();
    assert_eq!(
        session.read_with(cx, |session, cx| session.tab_titles(cx)),
        ["Query 2"],
        "the tab closed on the answer"
    );
    assert_eq!(
        runtime::block_on(other_count(&database)),
        2,
        "closing the tab rolled its transaction back"
    );
}

#[gpui_kit::test]
fn a_tab_without_a_transaction_closes_without_asking(cx: &mut TestAppContext) {
    let (_database, handle, session) = workspace_session(cx);

    run_in_tab(cx, &handle, &session, "select 1");
    assert_eq!(transaction(cx, &session), TxnState::Idle);

    close_workspace_tab(cx, &handle, &session, 0);
    assert!(!workspace_dialog_open(cx, &handle));
}

#[gpui_kit::test]
fn each_query_tab_has_its_own_transaction(cx: &mut TestAppContext) {
    let (database, handle, session) = workspace_session(cx);

    run_in_tab(cx, &handle, &session, "BEGIN");
    run_in_tab(
        cx,
        &handle,
        &session,
        "insert into items values (3, 'gamma', 3.0, NULL)",
    );

    // A second tab is on a connection of its own, outside the transaction.
    cx.update_window(handle.window.into(), |_, window, cx| {
        session.update(cx, |session, cx| session.open_tab_for_test(window, cx));
    })
    .unwrap();
    cx.run_until_parked();
    assert_eq!(transaction(cx, &session), TxnState::Idle);
    run_in_tab(cx, &handle, &session, "select count(*) from items");
    let status = session.read_with(cx, |session, cx| session.active_status_for_test(cx));
    assert!(status.contains("1 row"), "{status}");
    assert_eq!(runtime::block_on(other_count(&database)), 2);
}