switchy_database 0.5.0

Switchy database package
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

#![cfg(all(feature = "sqlite-sqlx", feature = "schema"))]

use std::sync::Arc;
use switchy_database::{
    Database,
    schema::{Column, DataType},
    sqlx::sqlite::SqliteSqlxDatabase,
};

#[tokio::test]
async fn typed_foreign_keys_enforce_constraints_and_preserve_metadata() {
    let pool = sqlx::sqlite::SqlitePoolOptions::new()
        .max_connections(1)
        .connect("sqlite::memory:")
        .await
        .unwrap();
    let db = SqliteSqlxDatabase::new(Arc::new(switchy_async::sync::Mutex::new(pool)));
    db.sqlite_foreign_keys(false).await.unwrap();
    db.sqlite_foreign_keys(true).await.unwrap();
    let column = |name: &str| Column {
        name: name.into(),
        data_type: DataType::BigInt,
        nullable: false,
        auto_increment: false,
        default: None,
    };
    db.create_table("parent")
        .column(column("id"))
        .primary_key("id")
        .execute(&db)
        .await
        .unwrap();
    db.create_table("child")
        .column(column("parent_id"))
        .foreign_key_column("parent_id", "parent", "id")
        .execute(&db)
        .await
        .unwrap();
    assert!(db.column_exists("parent", "id").await.unwrap());
    assert!(!db.column_exists("parent", "missing").await.unwrap());
    assert!(
        db.insert("child")
            .value("parent_id", 1_i64)
            .execute(&db)
            .await
            .is_err()
    );
    db.insert("parent")
        .value("id", 1_i64)
        .execute(&db)
        .await
        .unwrap();
    db.insert("child")
        .value("parent_id", 1_i64)
        .execute(&db)
        .await
        .unwrap();
    assert_eq!(db.sqlite_foreign_key_check().await.unwrap(), []);
    db.close().await.unwrap();
}