ormer 0.2.9

A minimalist ORM framework that supports SQLite, PostgreSQL, MySQL, and SqlServer
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
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128

#![cfg(any(feature = "sqlite", feature = "duckdb", feature = "postgresql"))]

use ormer::DbType;
#[cfg(any(feature = "sqlite", feature = "duckdb"))]
use ormer::OrmerError;

#[derive(Debug, ormer::Model)]
#[table = "partial_index_capability_users"]
struct PartialIndexCapabilityUser {
    #[primary]
    id: i32,
    #[index(where = "active = 1")]
    email: String,
    active: bool,
}

#[derive(Debug, ormer::Model)]
#[table = "partial_index_migration_users"]
struct PartialIndexMigrationUserV1 {
    #[primary]
    id: i32,
    active: bool,
}

#[derive(Debug, ormer::Model)]
#[table = "partial_index_migration_users"]
struct PartialIndexMigrationUserV2 {
    #[primary]
    id: i32,
    active: bool,
    #[index(where = "active = 1")]
    email: Option<String>,
}

#[cfg(any(feature = "sqlite", feature = "duckdb"))]
fn assert_partial_index_error(error: OrmerError, db_type: DbType) {
    assert!(matches!(
        error,
        OrmerError::UnsupportedFeature {
            backend,
            feature: "partial index WHERE clauses",
        } if backend == db_type
    ));
}

#[test]
#[cfg(feature = "sqlite")]
fn sqlite_partial_index_create_sql_is_capability_gated() {
    let error = ormer::generate_create_table_sql::<PartialIndexCapabilityUser>(DbType::Sqlite)
        .expect_err("SQLite must reject partial index metadata");
    assert_partial_index_error(error, DbType::Sqlite);
}

#[tokio::test]
#[cfg(feature = "sqlite")]
async fn sqlite_partial_index_create_table_is_capability_gated() -> ormer::Result<()> {
    let db = ormer::Database::connect(DbType::Sqlite, ":memory:").await?;
    let error = db
        .create_table::<PartialIndexCapabilityUser>()
        .execute()
        .await
        .expect_err("SQLite create_table must reject partial index metadata");
    assert_partial_index_error(error, DbType::Sqlite);
    Ok(())
}

#[tokio::test]
#[cfg(feature = "sqlite")]
async fn sqlite_partial_index_migration_plan_is_capability_gated() -> ormer::Result<()> {
    let db = ormer::Database::connect(DbType::Sqlite, ":memory:").await?;
    db.migrate_table::<PartialIndexMigrationUserV1>()
        .execute()
        .await?;

    let error = db
        .migrate_table::<PartialIndexMigrationUserV2>()
        .plan()
        .await
        .expect_err("SQLite migration planning must reject partial index metadata");
    assert_partial_index_error(error, DbType::Sqlite);
    Ok(())
}

#[test]
#[cfg(feature = "duckdb")]
fn duckdb_partial_index_create_sql_is_capability_gated() {
    let error = ormer::generate_create_table_sql::<PartialIndexCapabilityUser>(DbType::DuckDB)
        .expect_err("DuckDB must reject partial index metadata");
    assert_partial_index_error(error, DbType::DuckDB);
}

#[tokio::test]
#[cfg(feature = "duckdb")]
async fn duckdb_partial_index_create_table_is_capability_gated() -> ormer::Result<()> {
    let db = ormer::Database::connect(DbType::DuckDB, ":memory:").await?;
    let error = db
        .create_table::<PartialIndexCapabilityUser>()
        .execute()
        .await
        .expect_err("DuckDB create_table must reject partial index metadata");
    assert_partial_index_error(error, DbType::DuckDB);
    Ok(())
}

#[tokio::test]
#[cfg(feature = "duckdb")]
async fn duckdb_partial_index_migration_plan_is_capability_gated() -> ormer::Result<()> {
    let db = ormer::Database::connect(DbType::DuckDB, ":memory:").await?;
    db.migrate_table::<PartialIndexMigrationUserV1>()
        .execute()
        .await?;

    let error = db
        .migrate_table::<PartialIndexMigrationUserV2>()
        .plan()
        .await
        .expect_err("DuckDB migration planning must reject partial index metadata");
    assert_partial_index_error(error, DbType::DuckDB);
    Ok(())
}

#[test]
#[cfg(feature = "postgresql")]
fn postgresql_partial_index_create_sql_keeps_where_clause() -> ormer::Result<()> {
    let sql = ormer::generate_create_table_sql::<PartialIndexCapabilityUser>(DbType::PostgreSQL)?;
    assert!(sql.contains("WHERE active = 1"));
    Ok(())
}