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

Add more find_also_related test cases

Browse Source 
From darkmmon/sea-orm/commit/ce5f6991391fdbb0b8deb72ddd404944bb05cf98
This commit is contained in:
Chris Tsang 2023-08-18 13:10:31 +01:00
parent d50312c081
commit 08c3441a9c
1 changed files with 134 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

139
src/executor/select.rs
View File

@ -1134,6 +1134,33 @@ mod tests {
sea_orm::tests_cfg::cake::Model { id, name } sea_orm::tests_cfg::cake::Model { id, name }
} }
fn filling_model(id: i32) -> sea_orm::tests_cfg::filling::Model {
let name = match id {
1 => "apple juice",
2 => "orange jam",
3 => "fruit",
4 => "chocolate crust",
_ => "",
}
.to_string();
sea_orm::tests_cfg::filling::Model {
id,
name,
vendor_id: Some(1),
ignored_attr: 0,
}
}
fn cake_filling_models(
cake_id: i32,
filling_id: i32,
) -> (
sea_orm::tests_cfg::cake::Model,
sea_orm::tests_cfg::filling::Model,
) {
(cake_model(cake_id), filling_model(filling_id))
}
fn fruit_model(id: i32, cake_id: Option<i32>) -> sea_orm::tests_cfg::fruit::Model { fn fruit_model(id: i32, cake_id: Option<i32>) -> sea_orm::tests_cfg::fruit::Model {
let name = match id { let name = match id {
1 => "apple", 1 => "apple",
@ -1230,7 +1257,7 @@ mod tests {
.append_query_results([[ .append_query_results([[
cake_fruit_model(1, 1), cake_fruit_model(1, 1),
cake_fruit_model(1, 2), cake_fruit_model(1, 2),
cake_fruit_model(2, 2), cake_fruit_model(2, 3),
]]) ]])
.into_connection(); .into_connection();
@ -1239,7 +1266,7 @@ mod tests {
[ [
(cake_model(1), Some(fruit_model(1, Some(1)))), (cake_model(1), Some(fruit_model(1, Some(1)))),
(cake_model(1), Some(fruit_model(2, Some(1)))), (cake_model(1), Some(fruit_model(2, Some(1)))),
(cake_model(2), Some(fruit_model(2, Some(2)))) (cake_model(2), Some(fruit_model(3, Some(2))))
] ]
); );
@ -1255,7 +1282,7 @@ mod tests {
.append_query_results([[ .append_query_results([[
cake_fruit_model(1, 1).into_mock_row(), cake_fruit_model(1, 1).into_mock_row(),
cake_fruit_model(1, 2).into_mock_row(), cake_fruit_model(1, 2).into_mock_row(),
cake_fruit_model(2, 2).into_mock_row(), cake_fruit_model(2, 3).into_mock_row(),
(cake_model(3), None::<fruit::Model>).into_mock_row(), (cake_model(3), None::<fruit::Model>).into_mock_row(),
]]) ]])
.into_connection(); .into_connection();
@ -1265,7 +1292,59 @@ mod tests {
[ [
(cake_model(1), Some(fruit_model(1, Some(1)))), (cake_model(1), Some(fruit_model(1, Some(1)))),
(cake_model(1), Some(fruit_model(2, Some(1)))), (cake_model(1), Some(fruit_model(2, Some(1)))),
(cake_model(2), Some(fruit_model(2, Some(2)))), (cake_model(2), Some(fruit_model(3, Some(2)))),
(cake_model(3), None)
]
);
Ok(())
}
#[smol_potat::test]
pub async fn also_related_many_to_many() -> Result<(), sea_orm::DbErr> {
use sea_orm::tests_cfg::*;
use sea_orm::{DbBackend, EntityTrait, IntoMockRow, MockDatabase};
let db = MockDatabase::new(DbBackend::Postgres)
.append_query_results([[
cake_filling_models(1, 1).into_mock_row(),
cake_filling_models(1, 2).into_mock_row(),
cake_filling_models(2, 2).into_mock_row(),
]])
.into_connection();
assert_eq!(
Cake::find().find_also_related(Filling).all(&db).await?,
[
(cake_model(1), Some(filling_model(1))),
(cake_model(1), Some(filling_model(2))),
(cake_model(2), Some(filling_model(2))),
]
);
Ok(())
}
#[smol_potat::test]
pub async fn also_related_many_to_many_2() -> Result<(), sea_orm::DbErr> {
use sea_orm::tests_cfg::*;
use sea_orm::{DbBackend, EntityTrait, IntoMockRow, MockDatabase};
let db = MockDatabase::new(DbBackend::Postgres)
.append_query_results([[
cake_filling_models(1, 1).into_mock_row(),
cake_filling_models(1, 2).into_mock_row(),
cake_filling_models(2, 2).into_mock_row(),
(cake_model(3), None::<filling::Model>).into_mock_row(),
]])
.into_connection();
assert_eq!(
Cake::find().find_also_related(Filling).all(&db).await?,
[
(cake_model(1), Some(filling_model(1))),
(cake_model(1), Some(filling_model(2))),
(cake_model(2), Some(filling_model(2))),
(cake_model(3), None) (cake_model(3), None)
] ]
); );
@ -1383,6 +1462,56 @@ mod tests {
Ok(()) Ok(())
} }
#[smol_potat::test]
pub async fn with_related_many_to_many() -> Result<(), sea_orm::DbErr> {
use sea_orm::tests_cfg::*;
use sea_orm::{DbBackend, EntityTrait, IntoMockRow, MockDatabase};
let db = MockDatabase::new(DbBackend::Postgres)
.append_query_results([[
cake_filling_models(1, 1).into_mock_row(),
cake_filling_models(1, 2).into_mock_row(),
cake_filling_models(2, 2).into_mock_row(),
]])
.into_connection();
assert_eq!(
Cake::find().find_with_related(Filling).all(&db).await?,
[
(cake_model(1), vec![filling_model(1), filling_model(2)]),
(cake_model(2), vec![filling_model(2)]),
]
);
Ok(())
}
#[smol_potat::test]
pub async fn with_related_many_to_many_2() -> Result<(), sea_orm::DbErr> {
use sea_orm::tests_cfg::*;
use sea_orm::{DbBackend, EntityTrait, IntoMockRow, MockDatabase};
let db = MockDatabase::new(DbBackend::Postgres)
.append_query_results([[
cake_filling_models(1, 1).into_mock_row(),
cake_filling_models(1, 2).into_mock_row(),
cake_filling_models(2, 2).into_mock_row(),
(cake_model(3), None::<filling::Model>).into_mock_row(),
]])
.into_connection();
assert_eq!(
Cake::find().find_with_related(Filling).all(&db).await?,
[
(cake_model(1), vec![filling_model(1), filling_model(2)]),
(cake_model(2), vec![filling_model(2)]),
(cake_model(3), vec![])
]
);
Ok(())
}
#[smol_potat::test] #[smol_potat::test]
pub async fn also_linked_base() -> Result<(), sea_orm::DbErr> { pub async fn also_linked_base() -> Result<(), sea_orm::DbErr> {
use sea_orm::tests_cfg::*; use sea_orm::tests_cfg::*;
@ -1668,4 +1797,4 @@ mod tests {
Ok(()) Ok(())
} }
} }