50 lines
1.3 KiB
Rust
50 lines
1.3 KiB
Rust
pub mod common;
|
|
|
|
pub use common::{bakery_chain::*, setup::*, TestContext};
|
|
use sea_orm::{entity::prelude::*, *};
|
|
use sea_query::Query;
|
|
|
|
#[sea_orm_macros::test]
|
|
#[cfg(any(
|
|
feature = "sqlx-mysql",
|
|
feature = "sqlx-sqlite",
|
|
feature = "sqlx-postgres"
|
|
))]
|
|
async fn main() -> Result<(), DbErr> {
|
|
use bakery::*;
|
|
|
|
let ctx = TestContext::new("returning_tests").await;
|
|
let db = &ctx.db;
|
|
let builder = db.get_database_backend();
|
|
|
|
let mut insert = Query::insert();
|
|
insert
|
|
.into_table(Entity)
|
|
.columns(vec![Column::Name, Column::ProfitMargin])
|
|
.values_panic(vec!["Bakery Shop".into(), 0.5.into()]);
|
|
|
|
let mut update = Query::update();
|
|
update
|
|
.table(Entity)
|
|
.values(vec![
|
|
(Column::Name, "Bakery Shop".into()),
|
|
(Column::ProfitMargin, 0.5.into()),
|
|
])
|
|
.and_where(Column::Id.eq(1));
|
|
|
|
if db.support_returning() {
|
|
let mut returning = Query::select();
|
|
returning.columns(vec![Column::Id, Column::Name, Column::ProfitMargin]);
|
|
insert.returning(returning.clone());
|
|
update.returning(returning);
|
|
}
|
|
|
|
create_tables(db).await?;
|
|
db.query_one(builder.build(&insert)).await?;
|
|
db.query_one(builder.build(&update)).await?;
|
|
assert!(false);
|
|
ctx.delete().await;
|
|
|
|
Ok(())
|
|
}
|