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

Test nested transaction

Browse Source
This commit is contained in:
Billy Chan 2021-10-11 23:00:24 +08:00
parent ecaa0dca18
commit 12800468b1
No known key found for this signature in database
GPG Key ID: A2D690CAC7DF3CC7
1 changed files with 245 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

245
tests/transaction_tests.rs
View File

@ -101,3 +101,248 @@ fn _transaction_with_reference<'a>(
Ok(())
})
}
#[sea_orm_macros::test]
#[cfg(any(
feature = "sqlx-mysql",
feature = "sqlx-sqlite",
feature = "sqlx-postgres"
))]
pub async fn transaction_nested() {
let ctx = TestContext::new("transaction_nested_test").await;
ctx.db
.transaction::<_, _, DbErr>(|txn| {
Box::pin(async move {
let _ = bakery::ActiveModel {
name: Set("SeaSide Bakery".to_owned()),
profit_margin: Set(10.4),
..Default::default()
}
.save(txn)
.await?;
let _ = bakery::ActiveModel {
name: Set("Top Bakery".to_owned()),
profit_margin: Set(15.0),
..Default::default()
}
.save(txn)
.await?;
// Try nested transaction committed
txn.transaction::<_, _, DbErr>(|txn| {
Box::pin(async move {
let _ = bakery::ActiveModel {
name: Set("Nested Bakery".to_owned()),
profit_margin: Set(88.88),
..Default::default()
}
.save(txn)
.await?;
let bakeries = Bakery::find()
.filter(bakery::Column::Name.contains("Bakery"))
.all(txn)
.await?;
assert_eq!(bakeries.len(), 3);
// Try nested-nested transaction rollbacked
let is_err = txn
.transaction::<_, _, DbErr>(|txn| {
Box::pin(async move {
let _ = bakery::ActiveModel {
name: Set("Rock n Roll Bakery".to_owned()),
profit_margin: Set(28.8),
..Default::default()
}
.save(txn)
.await?;
let bakeries = Bakery::find()
.filter(bakery::Column::Name.contains("Bakery"))
.all(txn)
.await?;
assert_eq!(bakeries.len(), 4);
if true {
Err(DbErr::Query("Force Rollback!".to_owned()))
} else {
Ok(())
}
})
})
.await
.is_err();
assert!(is_err);
let bakeries = Bakery::find()
.filter(bakery::Column::Name.contains("Bakery"))
.all(txn)
.await?;
assert_eq!(bakeries.len(), 3);
// Try nested-nested transaction committed
txn.transaction::<_, _, DbErr>(|txn| {
Box::pin(async move {
let _ = bakery::ActiveModel {
name: Set("Rock n Roll Bakery".to_owned()),
profit_margin: Set(28.8),
..Default::default()
}
.save(txn)
.await?;
let bakeries = Bakery::find()
.filter(bakery::Column::Name.contains("Bakery"))
.all(txn)
.await?;
assert_eq!(bakeries.len(), 4);
Ok(())
})
})
.await
.unwrap();
let bakeries = Bakery::find()
.filter(bakery::Column::Name.contains("Bakery"))
.all(txn)
.await?;
assert_eq!(bakeries.len(), 4);
Ok(())
})
})
.await
.unwrap();
// Try nested transaction rollbacked
let is_err = txn
.transaction::<_, _, DbErr>(|txn| {
Box::pin(async move {
let _ = bakery::ActiveModel {
name: Set("Rock n Roll Bakery".to_owned()),
profit_margin: Set(28.8),
..Default::default()
}
.save(txn)
.await?;
let bakeries = Bakery::find()
.filter(bakery::Column::Name.contains("Bakery"))
.all(txn)
.await?;
assert_eq!(bakeries.len(), 5);
// Try nested-nested transaction committed
txn.transaction::<_, _, DbErr>(|txn| {
Box::pin(async move {
let _ = bakery::ActiveModel {
name: Set("Rock n Roll Bakery".to_owned()),
profit_margin: Set(28.8),
..Default::default()
}
.save(txn)
.await?;
let bakeries = Bakery::find()
.filter(bakery::Column::Name.contains("Bakery"))
.all(txn)
.await?;
assert_eq!(bakeries.len(), 6);
Ok(())
})
})
.await
.unwrap();
let bakeries = Bakery::find()
.filter(bakery::Column::Name.contains("Bakery"))
.all(txn)
.await?;
assert_eq!(bakeries.len(), 6);
// Try nested-nested transaction rollbacked
let is_err = txn
.transaction::<_, _, DbErr>(|txn| {
Box::pin(async move {
let _ = bakery::ActiveModel {
name: Set("Rock n Roll Bakery".to_owned()),
profit_margin: Set(28.8),
..Default::default()
}
.save(txn)
.await?;
let bakeries = Bakery::find()
.filter(bakery::Column::Name.contains("Bakery"))
.all(txn)
.await?;
assert_eq!(bakeries.len(), 7);
if true {
Err(DbErr::Query("Force Rollback!".to_owned()))
} else {
Ok(())
}
})
})
.await
.is_err();
assert!(is_err);
let bakeries = Bakery::find()
.filter(bakery::Column::Name.contains("Bakery"))
.all(txn)
.await?;
assert_eq!(bakeries.len(), 6);
if true {
Err(DbErr::Query("Force Rollback!".to_owned()))
} else {
Ok(())
}
})
})
.await
.is_err();
assert!(is_err);
let bakeries = Bakery::find()
.filter(bakery::Column::Name.contains("Bakery"))
.all(txn)
.await?;
assert_eq!(bakeries.len(), 4);
Ok(())
})
})
.await
.unwrap();
let bakeries = Bakery::find()
.filter(bakery::Column::Name.contains("Bakery"))
.all(&ctx.db)
.await
.unwrap();
assert_eq!(bakeries.len(), 4);
ctx.delete().await;
}