mongreldb-kit 0.58.2

Schema-aware, storage-backed application persistence layer for MongrelDB with migrations, relational constraints, and a query builder.
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

use mongreldb_kit::{Column, ColumnType, Database, Schema, Table};

fn database() -> (tempfile::TempDir, Database) {
    let directory = tempfile::tempdir().unwrap();
    let schema = Schema::new(vec![Table {
        id: 1,
        name: "items".into(),
        columns: vec![Column::new(1, "id", ColumnType::Int64)],
        primary_key: vec!["id".into()],
        indexes: vec![],
        foreign_keys: vec![],
        unique_constraints: vec![],
        check_constraints: vec![],
    }])
    .unwrap();
    let db = Database::create(directory.path(), schema).unwrap();
    (directory, db)
}

fn ids(db: &Database) -> Vec<i64> {
    db.sql_rows("SELECT id FROM items ORDER BY id")
        .unwrap()
        .into_iter()
        .map(|row| row["id"].as_i64().unwrap())
        .collect()
}

#[test]
fn sql_savepoint_rolls_back_only_later_writes() {
    let (_directory, db) = database();

    db.sql("BEGIN").unwrap();
    db.sql("INSERT INTO items VALUES (1)").unwrap();
    db.sql("SAVEPOINT sp1").unwrap();
    db.sql("INSERT INTO items VALUES (2)").unwrap();
    db.sql("ROLLBACK TO sp1").unwrap();
    db.sql("COMMIT").unwrap();

    assert_eq!(ids(&db), vec![1]);
}

#[test]
fn sql_nested_savepoint_retains_target_and_removes_later_savepoints() {
    let (_directory, db) = database();

    db.sql("BEGIN").unwrap();
    db.sql("INSERT INTO items VALUES (1)").unwrap();
    db.sql("SAVEPOINT sp1").unwrap();
    db.sql("INSERT INTO items VALUES (2)").unwrap();
    db.sql("SAVEPOINT sp2").unwrap();
    db.sql("INSERT INTO items VALUES (3)").unwrap();
    db.sql("ROLLBACK TO sp1").unwrap();
    db.sql("INSERT INTO items VALUES (4)").unwrap();
    db.sql("ROLLBACK TO sp1").unwrap();
    db.sql("INSERT INTO items VALUES (4)").unwrap();
    db.sql("COMMIT").unwrap();

    assert_eq!(ids(&db), vec![1, 4]);

    db.sql("BEGIN").unwrap();
    db.sql("SAVEPOINT outer").unwrap();
    db.sql("SAVEPOINT inner").unwrap();
    db.sql("ROLLBACK TO outer").unwrap();
    let error = db.sql("ROLLBACK TO inner").unwrap_err();
    assert!(error.to_string().contains("no savepoint named 'inner'"));
    db.sql("ROLLBACK").unwrap();
}

#[test]
fn sql_release_removes_target_and_nested_savepoints() {
    let (_directory, db) = database();

    for name in ["outer", "inner"] {
        db.sql("BEGIN").unwrap();
        db.sql("SAVEPOINT outer").unwrap();
        db.sql("SAVEPOINT inner").unwrap();
        db.sql("RELEASE SAVEPOINT outer").unwrap();
        let error = db.sql(&format!("ROLLBACK TO {name}")).unwrap_err();
        assert!(
            error
                .to_string()
                .contains(&format!("no savepoint named '{name}'")),
            "{error}"
        );
        db.sql("ROLLBACK").unwrap();
    }
}

#[test]
fn sql_rollback_to_recovers_an_aborted_transaction() {
    let (_directory, db) = database();

    db.sql("BEGIN").unwrap();
    db.sql("INSERT INTO items VALUES (1)").unwrap();
    db.sql("SAVEPOINT stable").unwrap();
    assert!(db
        .sql("INSERT INTO items VALUES (2); SELECT * FROM missing_table")
        .is_err());
    assert!(db
        .sql("COMMIT")
        .unwrap_err()
        .to_string()
        .contains("aborted"));

    db.sql("ROLLBACK TO stable").unwrap();
    db.sql("INSERT INTO items VALUES (3)").unwrap();
    db.sql("COMMIT").unwrap();

    assert_eq!(ids(&db), vec![1, 3]);
}