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

Add load_many unit test

Browse Source
This commit is contained in:
Panagiotis Karatakis 2022-11-22 10:38:54 +02:00
parent 94650bf5c6
commit f7f90cd7c5
2 changed files with 78 additions and 48 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

121
src/entity/loader.rs
View File

@ -12,11 +12,9 @@ pub trait LoaderTrait {
/// Source model
type Model: ModelTrait;
///
/// Used to eager load has_one relations
///
///
///
///
async fn load_one<R, C>(&self, stmt: Select<R>, db: &C) -> Result<Vec<Option<R::Model>>, DbErr>
where
C: ConnectionTrait,
@ -26,35 +24,9 @@ pub trait LoaderTrait {
<<Self as LoaderTrait>::Model as ModelTrait>::Entity: Related<R>,
<<<<Self as LoaderTrait>::Model as ModelTrait>::Entity as EntityTrait>::Column as FromStr>::Err: Debug;
///
/// Used to eager load has_many relations
///
/// # Example
///
/// ```
/// use sea_orm::{tests_cfg::*, entity::loader::*};
///
/// let db = MockDatabase::new(DbBackend::Postgres)
/// .append_query_results(vec![
/// vec![cake::Model {
/// id: 1,
/// name: "New York Cheese".to_owned(),
/// }
/// .into_mock_row()],
/// vec![fruit::Model {
/// id: 1,
/// name: "Apple".to_owned(),
/// cake_id: Some(1),
/// }
/// .into_mock_row()],
/// ])
/// .into_connection();
///
/// let cakes = vec![cake::Model { id: 1, name: "New York Cheese".to_owned(), }];
///
/// let fruits = cakes.load_many(fruit::Entity::find(), &db);
///
/// assert_eq!(fruits, vec![[fruit::Model { id: 1, name: "Apple".to_owned(), cake_id: Some(1), }]]);
/// ```
async fn load_many<R, C>(&self, stmt: Select<R>, db: &C) -> Result<Vec<Vec<R::Model>>, DbErr>
where
C: ConnectionTrait,
@ -97,7 +69,7 @@ where
.map(|model: &M| extract_key(&rel_def.from_col, model))
.collect();
let condition = prepare_condition::<M>(&rel_def.to_col, &keys);
let condition = prepare_condition::<<R as EntityTrait>::Model>(&rel_def.to_col, &keys);
let stmt = <Select<R> as QueryFilter>::filter(stmt, condition);
@ -153,7 +125,7 @@ where
.map(|model: &M| extract_key(&rel_def.from_col, model))
.collect();
let condition = prepare_condition::<M>(&rel_def.to_col, &keys);
let condition = prepare_condition::<<R as EntityTrait>::Model>(&rel_def.to_col, &keys);
let stmt = <Select<R> as QueryFilter>::filter(stmt, condition);
@ -171,14 +143,21 @@ where
.for_each(|value: <R as EntityTrait>::Model| {
let key = extract_key(&rel_def.to_col, &value);
let vec = hashmap.get_mut(&format!("{:?}", key)).unwrap();
let vec = hashmap
.get_mut(&format!("{:?}", key))
.expect("Failed at finding key on hashmap");
vec.push(value);
});
let result: Vec<Vec<R::Model>> = keys
.iter()
.map(|key: &Vec<Value>| hashmap.remove(&format!("{:?}", key)).to_owned().unwrap())
.map(|key: &Vec<Value>| {
hashmap
.remove(&format!("{:?}", key))
.to_owned()
.expect("Failed to convert key to owned")
})
.collect();
Ok(result)
@ -196,7 +175,7 @@ where
<<<Model as ModelTrait>::Entity as EntityTrait>::Column as FromStr>::from_str(
&a.to_string(),
)
.unwrap();
.expect("Failed at mapping string to column A:1");
vec![model.get(column_a)]
}
Identity::Binary(a, b) => {
@ -204,12 +183,12 @@ where
<<<Model as ModelTrait>::Entity as EntityTrait>::Column as FromStr>::from_str(
&a.to_string(),
)
.unwrap();
.expect("Failed at mapping string to column A:2");
let column_b =
<<<Model as ModelTrait>::Entity as EntityTrait>::Column as FromStr>::from_str(
&b.to_string(),
)
.unwrap();
.expect("Failed at mapping string to column B:2");
vec![model.get(column_a), model.get(column_b)]
}
Identity::Ternary(a, b, c) => {
@ -217,17 +196,17 @@ where
<<<Model as ModelTrait>::Entity as EntityTrait>::Column as FromStr>::from_str(
&a.to_string(),
)
.unwrap();
.expect("Failed at mapping string to column A:3");
let column_b =
<<<Model as ModelTrait>::Entity as EntityTrait>::Column as FromStr>::from_str(
&b.to_string(),
)
.unwrap();
.expect("Failed at mapping string to column B:3");
let column_c =
<<<Model as ModelTrait>::Entity as EntityTrait>::Column as FromStr>::from_str(
&c.to_string(),
)
.unwrap();
.expect("Failed at mapping string to column C:3");
vec![
model.get(column_a),
model.get(column_b),
@ -246,7 +225,7 @@ where
Identity::Unary(column_a) => {
let column_a: <M::Entity as EntityTrait>::Column =
<<M::Entity as EntityTrait>::Column as FromStr>::from_str(&column_a.to_string())
.unwrap();
.expect("Failed at mapping string to column *A:1");
Condition::all().add(ColumnTrait::is_in(
&column_a,
keys.iter()
@ -257,10 +236,10 @@ where
Identity::Binary(column_a, column_b) => {
let column_a: <M::Entity as EntityTrait>::Column =
<<M::Entity as EntityTrait>::Column as FromStr>::from_str(&column_a.to_string())
.unwrap();
.expect("Failed at mapping string to column *A:2");
let column_b: <M::Entity as EntityTrait>::Column =
<<M::Entity as EntityTrait>::Column as FromStr>::from_str(&column_b.to_string())
.unwrap();
.expect("Failed at mapping string to column *B:2");
Condition::all().add(
Expr::tuple([
SimpleExpr::Column(column_a.into_column_ref()),
@ -276,13 +255,13 @@ where
Identity::Ternary(column_a, column_b, column_c) => {
let column_a: <M::Entity as EntityTrait>::Column =
<<M::Entity as EntityTrait>::Column as FromStr>::from_str(&column_a.to_string())
.unwrap();
.expect("Failed at mapping string to column *A:3");
let column_b: <M::Entity as EntityTrait>::Column =
<<M::Entity as EntityTrait>::Column as FromStr>::from_str(&column_b.to_string())
.unwrap();
.expect("Failed at mapping string to column *B:3");
let column_c: <M::Entity as EntityTrait>::Column =
<<M::Entity as EntityTrait>::Column as FromStr>::from_str(&column_c.to_string())
.unwrap();
.expect("Failed at mapping string to column *C:3");
Condition::all().add(
Expr::tuple([
SimpleExpr::Column(column_a.into_column_ref()),
@ -300,3 +279,53 @@ where
}
}
}
#[cfg(test)]
mod tests {
#[tokio::test]
async fn test_load_many() {
use crate::{
entity::prelude::*, tests_cfg::*, DbBackend, IntoMockRow, LoaderTrait, MockDatabase,
};
let db = MockDatabase::new(DbBackend::Postgres)
.append_query_results(vec![
vec![fruit::Model {
id: 1,
name: "Apple".to_owned(),
cake_id: Some(1),
}
.into_mock_row()],
])
.into_connection();
let cakes = vec![
cake::Model {
id: 1,
name: "New York Cheese".to_owned(),
},
cake::Model {
id: 2,
name: "London Cheese".to_owned(),
},
];
let fruits = cakes
.load_many(fruit::Entity::find(), &db)
.await
.expect("Should return something");
assert_eq!(
fruits,
vec![
vec![fruit::Model {
id: 1,
name: "Apple".to_owned(),
cake_id: Some(1),
}],
vec![]
]
);
}
}

5
src/entity/prelude.rs
View File

@ -1,8 +1,9 @@
pub use crate::{
error::*, ActiveEnum, ActiveModelBehavior, ActiveModelTrait, ColumnDef, ColumnTrait,
ColumnType, CursorTrait, DatabaseConnection, DbConn, EntityName, EntityTrait, EnumIter,
ForeignKeyAction, Iden, IdenStatic, Linked, ModelTrait, PaginatorTrait, PrimaryKeyToColumn,
PrimaryKeyTrait, QueryFilter, QueryResult, Related, RelationDef, RelationTrait, Select, Value,
ForeignKeyAction, Iden, IdenStatic, Linked, LoaderTrait, ModelTrait, PaginatorTrait,
PrimaryKeyToColumn, PrimaryKeyTrait, QueryFilter, QueryResult, Related, RelationDef,
RelationTrait, Select, Value,
};
#[cfg(feature = "macros")]