nodedb 0.4.0

Local-first, real-time, edge-to-cloud hybrid database for multi-modal workloads
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

// SPDX-License-Identifier: BUSL-1.1

//! PostgreSQL-compatible RESET behavior for mutable session parameters.

mod common;

use common::pgwire_harness::TestServer;

#[tokio::test]
async fn reset_unknown_parameter_is_rejected() {
    let server = TestServer::start().await;
    assert!(
        server
            .exec("RESET totally_made_up_parameter")
            .await
            .is_err(),
        "RESET must reject parameters outside the mutable allowlist"
    );
}

#[tokio::test]
async fn reset_restores_each_session_default() {
    let server = TestServer::start().await;
    let cases = [
        ("nodedb.consistency", "eventual", "strong"),
        ("default_read_consistency", "eventual", "strong"),
        ("cross_shard_txn", "best_effort_non_atomic", "strict"),
        ("default_transaction_read_only", "on", "off"),
    ];

    for (parameter, changed, expected) in cases {
        server
            .exec(&format!("SET {parameter} = '{changed}'"))
            .await
            .unwrap();
        server.exec(&format!("RESET {parameter}")).await.unwrap();
        assert_eq!(
            server
                .query_text(&format!("SHOW {parameter}"))
                .await
                .unwrap(),
            vec![expected],
            "RESET must restore the default for {parameter}"
        );
    }
}

#[tokio::test]
async fn reset_all_restores_session_defaults() {
    let server = TestServer::start().await;
    server
        .exec("SET nodedb.consistency = eventual")
        .await
        .unwrap();
    server
        .exec("SET default_read_consistency = eventual")
        .await
        .unwrap();

    server.exec("RESET ALL").await.unwrap();

    assert_eq!(
        server.query_text("SHOW nodedb.consistency").await.unwrap(),
        vec!["strong"]
    );
    assert_eq!(
        server
            .query_text("SHOW default_read_consistency")
            .await
            .unwrap(),
        vec!["strong"]
    );
}

#[tokio::test]
async fn reset_all_cannot_change_tenant_inside_transaction() {
    let server = TestServer::start().await;
    server.exec("CREATE TENANT reset_guard ID 2").await.unwrap();
    server.exec("SET nodedb.tenant_id = 2").await.unwrap();
    server.exec("BEGIN").await.unwrap();

    let error = server
        .exec("RESET ALL")
        .await
        .expect_err("RESET ALL must not change tenant inside a transaction");
    assert!(
        error.contains("25001"),
        "expected SQLSTATE 25001, got {error}"
    );

    server.exec("ROLLBACK").await.unwrap();
    let tenant = server.query_rows("SHOW TENANT").await.unwrap();
    assert_eq!(
        tenant[0][0], "2",
        "failed RESET ALL must preserve tenant override"
    );
}