pgtemp 0.7.1

Start local postgres servers for testing without Docker
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

//! Tests for the dump and restore functionality

use std::io::Write;

use pgtemp::PgTempDB;
use sqlx::postgres::PgConnection;
use sqlx::prelude::*;

// TODO: test dump and restore with migrations

#[tokio::test]
/// create a table and insert into it, then dump it and restore it to a new database
async fn dump_and_restore() {
    let temp = tempfile::tempdir().unwrap();
    let dump_path = temp.path().join("dump.sql");

    let db = PgTempDB::builder()
        .dump_database(&dump_path)
        .start_async()
        .await;

    let mut conn = PgConnection::connect(&db.connection_uri())
        .await
        .expect("failed to connect to db");

    sqlx::query(
        "
        CREATE TABLE person (
            id      SERIAL PRIMARY KEY,
            name    TEXT NOT NULL
        )
    ",
    )
    .execute(&mut conn)
    .await
    .expect("failed to create table");

    for i in 0..10 {
        let name = format!("example name {}", i);
        sqlx::query("INSERT INTO person (name) VALUES ($1)")
            .bind(name)
            .execute(&mut conn)
            .await
            .expect("failed to insert name into values");
    }

    drop(conn);
    drop(db); // database is dumped here

    // start a new database and load the data
    let db = PgTempDB::builder()
        .load_database(&dump_path)
        .start_async()
        .await;

    let mut conn = PgConnection::connect(&db.connection_uri())
        .await
        .expect("failed to connect to db");
    // check the data is there
    let rows = sqlx::query("SELECT id, name FROM person ORDER BY name ASC")
        .fetch_all(&mut conn)
        .await
        .expect("failed to select names from person");
    assert_eq!(rows.len(), 10);

    let row = &rows[0];
    let id: i32 = row.get(0);
    let name: &str = row.get(1);

    assert_eq!(id, 1);
    assert_eq!(name, "example name 0");

    let row = &rows[9];
    let id: i32 = row.get(0);
    let name: &str = row.get(1);

    assert_eq!(id, 10);
    assert_eq!(name, "example name 9");
}

#[tokio::test]
/// make sure that we correctly error on bad database dumps.
#[should_panic(expected = "syntax error at or near \\\"INVALID\\")]
async fn panic_on_load_error() {
    let temp = tempfile::tempdir().unwrap();
    let db_dump_path = temp.path().join("dump.sql");

    // Create some bad database dumps
    let mut f = std::fs::File::create(&db_dump_path).unwrap();
    f.write_all(b"INVALID SQL").unwrap();
    f.flush().expect("Failed to flush file");

    // Try to load it (it should fail)
    PgTempDB::builder()
        .load_database(&db_dump_path)
        .start_async()
        .await;
}