ormer 0.1.2

The simplest ORM for Rust.
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
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149

/// drop_table 功能测试
use ormer::{Database, DbType, Model};

#[derive(Debug, Model)]
#[table = "test_drop_users"]
struct TestDropUser {
    #[primary(auto)]
    id: i32,
    name: String,
    age: i32,
}

#[derive(Debug, Model)]
#[table = "test_drop_roles"]
struct TestDropRole {
    #[primary]
    id: i32,
    role_name: String,
}

#[tokio::test]
async fn test_drop_table_basic() {
    let db = Database::connect(DbType::Turso, ":memory:").await.unwrap();

    // 创建表
    db.create_table::<TestDropUser>().await.unwrap();

    // 验证表存在 - 尝试插入数据
    db.insert(&TestDropUser {
        id: 1,
        name: "Alice".to_string(),
        age: 20,
    })
    .await
    .unwrap();

    // 删除表
    db.drop_table::<TestDropUser>().await.unwrap();

    // 验证表已删除 - 重新创建表应该成功（如果表还存在会验证schema）
    db.create_table::<TestDropUser>().await.unwrap();

    // 清理
    db.drop_table::<TestDropUser>().await.unwrap();
}

#[tokio::test]
async fn test_drop_table_if_exists() {
    let db = Database::connect(DbType::Turso, ":memory:").await.unwrap();

    // 删除一个不存在的表（应该不报错）
    db.drop_table::<TestDropRole>().await.unwrap();

    // 创建表
    db.create_table::<TestDropRole>().await.unwrap();

    // 删除表
    db.drop_table::<TestDropRole>().await.unwrap();

    // 再次删除（应该不报错，因为使用了 IF EXISTS）
    db.drop_table::<TestDropRole>().await.unwrap();
}

#[tokio::test]
async fn test_drop_multiple_tables() {
    let db = Database::connect(DbType::Turso, ":memory:").await.unwrap();

    // 创建多个表
    db.create_table::<TestDropUser>().await.unwrap();
    db.create_table::<TestDropRole>().await.unwrap();

    // 插入一些数据
    db.insert(&TestDropUser {
        id: 1,
        name: "Bob".to_string(),
        age: 25,
    })
    .await
    .unwrap();

    db.insert(&TestDropRole {
        id: 1,
        role_name: "admin".to_string(),
    })
    .await
    .unwrap();

    // 删除所有表
    db.drop_table::<TestDropUser>().await.unwrap();
    db.drop_table::<TestDropRole>().await.unwrap();

    // 验证表已删除 - 重新创建应该成功
    db.create_table::<TestDropUser>().await.unwrap();
    db.create_table::<TestDropRole>().await.unwrap();

    // 清理
    db.drop_table::<TestDropUser>().await.unwrap();
    db.drop_table::<TestDropRole>().await.unwrap();
}

#[tokio::test]
async fn test_drop_table_and_recreate() {
    let db = Database::connect(DbType::Turso, ":memory:").await.unwrap();

    // 创建表
    db.create_table::<TestDropUser>().await.unwrap();

    // 插入数据
    db.insert(&TestDropUser {
        id: 1,
        name: "Charlie".to_string(),
        age: 30,
    })
    .await
    .unwrap();

    db.insert(&TestDropUser {
        id: 2,
        name: "Diana".to_string(),
        age: 28,
    })
    .await
    .unwrap();

    // 查询验证数据存在
    let users = db
        .select::<TestDropUser>()
        .collect::<Vec<_>>()
        .await
        .unwrap();
    assert_eq!(users.len(), 2);

    // 删除表
    db.drop_table::<TestDropUser>().await.unwrap();

    // 重新创建表
    db.create_table::<TestDropUser>().await.unwrap();

    // 验证表是空的
    let users = db
        .select::<TestDropUser>()
        .collect::<Vec<_>>()
        .await
        .unwrap();
    assert_eq!(users.len(), 0);

    // 清理
    db.drop_table::<TestDropUser>().await.unwrap();
}