mockgres 0.0.29

An in-memory database that replicates a reasonable subset of Postgres functionality to make unit tests that rely on a database to run.
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

mod common;

#[tokio::test(flavor = "multi_thread")]
async fn rollback_discards_changes() {
    let ctx = common::start().await;

    ctx.client
        .execute("create table tx_items(id int primary key)", &[])
        .await
        .expect("create table");

    ctx.client.execute("begin", &[]).await.expect("begin");
    ctx.client
        .execute("insert into tx_items values (1)", &[])
        .await
        .expect("insert in tx");

    // Visible inside the transaction.
    let inside = ctx
        .client
        .query("select id from tx_items", &[])
        .await
        .expect("query inside tx");
    assert_eq!(inside.len(), 1);

    ctx.client.execute("rollback", &[]).await.expect("rollback");

    let rows = ctx
        .client
        .query("select id from tx_items", &[])
        .await
        .expect("query after rollback");
    assert!(rows.is_empty(), "rollback should discard inserted rows");

    let _ = ctx.shutdown.send(());
}

#[tokio::test(flavor = "multi_thread")]
async fn commit_persists_and_errors_on_misuse() {
    let ctx = common::start().await;

    ctx.client
        .execute("create table tx_books(id int primary key, title text)", &[])
        .await
        .expect("create table");

    ctx.client.execute("begin", &[]).await.expect("begin");
    ctx.client
        .execute(
            "insert into tx_books values (1, 'draft'), (2, 'final')",
            &[],
        )
        .await
        .expect("insert data");
    ctx.client.execute("commit", &[]).await.expect("commit");

    let rows = ctx
        .client
        .query("select id from tx_books order by id", &[])
        .await
        .expect("load rows");
    assert_eq!(rows.len(), 2);
    assert_eq!(rows[0].get::<_, i32>(0), 1);
    assert_eq!(rows[1].get::<_, i32>(0), 2);

    // COMMIT without BEGIN should error.
    let err = ctx.client.execute("commit", &[]).await.expect_err("no txn");
    let msg = err
        .as_db_error()
        .expect("expected db error")
        .message()
        .to_lowercase();
    assert!(
        msg.contains("no transaction"),
        "unexpected commit error: {msg:?}"
    );

    ctx.client.execute("begin", &[]).await.expect("begin 2");
    let err = ctx
        .client
        .execute("begin", &[])
        .await
        .expect_err("double begin");
    let msg = err
        .as_db_error()
        .expect("expected db error")
        .message()
        .to_lowercase();
    assert!(
        msg.contains("already in progress"),
        "unexpected begin error: {msg:?}"
    );
    ctx.client
        .execute("rollback", &[])
        .await
        .expect("rollback 2");

    let _ = ctx.shutdown.send(());
}