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

Improve testcase

Browse Source
This commit is contained in:
Chris Tsang 2022-11-23 23:47:14 +08:00
parent dcfba6f685
commit 45b391f434
1 changed files with 18 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
tests/loader_tests.rs
View File

@ -115,20 +115,20 @@ async fn loader_load_many() -> Result<(), DbErr> {
.await .await
.expect("could not insert baker"); .expect("could not insert baker");
let _baker_3 = baker::ActiveModel { let baker_3 = baker::ActiveModel {
name: Set("John".to_owned()), name: Set("John".to_owned()),
contact_details: Set(serde_json::json!({})), contact_details: Set(serde_json::json!({})),
bakery_id: Set(Some(bakery_1.id)), bakery_id: Set(Some(bakery_2.id)),
..Default::default() ..Default::default()
} }
.insert(&ctx.db) .insert(&ctx.db)
.await .await
.expect("could not insert baker"); .expect("could not insert baker");
let _baker_4 = baker::ActiveModel { let baker_4 = baker::ActiveModel {
name: Set("Baker 4".to_owned()), name: Set("Baker 4".to_owned()),
contact_details: Set(serde_json::json!({})), contact_details: Set(serde_json::json!({})),
bakery_id: Set(None), bakery_id: Set(Some(bakery_2.id)),
..Default::default() ..Default::default()
} }
.insert(&ctx.db) .insert(&ctx.db)
@ -153,7 +153,20 @@ async fn loader_load_many() -> Result<(), DbErr> {
assert_eq!(bakeries, vec![bakery_1, bakery_2]); assert_eq!(bakeries, vec![bakery_1, bakery_2]);
assert_eq!(bakers, vec![vec![baker_1, baker_2], vec![]]); assert_eq!(
bakers,
vec![
vec![baker_1.clone(), baker_2.clone()],
vec![baker_4.clone()]
]
);
let bakers = bakeries
.load_many(baker::Entity::find(), &ctx.db)
.await
.expect("Should load bakers");
assert_eq!(bakers, vec![vec![baker_1, baker_2], vec![baker_3, baker_4]]);
Ok(()) Ok(())
} }