caisin/sea-orm
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix: update many cast enum values (#1178)

Browse Source
This commit is contained in:
Billy Chan 2022-12-19 17:09:18 +08:00 committed by GitHub
parent d205338f3f
commit 17ed7156c4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 32 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
src/query/update.rs
View File

@ -189,7 +189,8 @@ where
for col in E::Column::iter() {
let av = model.get(col);
if av.is_set() {
self.query.value(col, av.unwrap());
let expr = cast_text_as_enum(Expr::val(av.into_value().unwrap()), &col);
self.query.value(col, expr);
}
}
self
@ -207,7 +208,7 @@ where
#[cfg(test)]
mod tests {
use crate::tests_cfg::{cake, fruit};
use crate::tests_cfg::{cake, fruit, lunch_set, sea_orm_active_enums::Tea};
use crate::{entity::*, query::*, DbBackend};
use sea_query::{Expr, Value};
@ -294,4 +295,33 @@ mod tests {
r#"UPDATE "fruit" SET "id" = 3 WHERE "fruit"."id" = 2"#,
);
}
#[test]
fn update_7() {
assert_eq!(
Update::many(lunch_set::Entity)
.set(lunch_set::ActiveModel {
tea: Set(Tea::EverydayTea),
..Default::default()
})
.filter(lunch_set::Column::Tea.eq(Tea::BreakfastTea))
.build(DbBackend::Postgres)
.to_string(),
r#"UPDATE "lunch_set" SET "tea" = CAST('EverydayTea' AS tea) WHERE "lunch_set"."tea" = CAST('BreakfastTea' AS tea)"#,
);
}
#[test]
fn update_8() {
assert_eq!(
Update::one(lunch_set::ActiveModel {
id: Unchanged(1),
tea: Set(Tea::EverydayTea),
..Default::default()
})
.build(DbBackend::Postgres)
.to_string(),
r#"UPDATE "lunch_set" SET "tea" = CAST('EverydayTea' AS tea) WHERE "lunch_set"."id" = 1"#,
);
}
}