#[derive(Debug, ormer::Model)]
#[table = "test_roles"]
struct TestRole {
#[primary]
id: i32,
name: String,
}
#[tokio::test]
async fn test_insert_or_update() -> Result<(), Box<dyn std::error::Error>> {
let db = ormer::Database::connect(ormer::DbType::Turso, ":memory:").await?;
db.create_table::<TestRole>().await?;
db.insert_or_update(&TestRole {
id: 1,
name: "admin".to_string(),
})
.await?;
println!("第一次插入成功");
let roles = db.select::<TestRole>().collect::<Vec<_>>().await?;
println!("第一次查询: {:?}", roles);
assert_eq!(roles.len(), 1);
assert_eq!(roles[0].name, "admin");
db.insert_or_update(&TestRole {
id: 1,
name: "super_admin".to_string(),
})
.await?;
println!("第二次 insert_or_update 成功(更新操作)");
let roles = db.select::<TestRole>().collect::<Vec<_>>().await?;
println!("第二次查询: {:?}", roles);
assert_eq!(roles.len(), 1);
assert_eq!(roles[0].name, "super_admin");
println!("\n测试通过!insert_or_update 方法在遇到重复记录时成功执行了更新操作");
Ok(())
}