Easy joins with MockDatabase #447

src/database/mock.rs

@@ -1,7 +1,7 @@
 use crate::{
     error::*, DatabaseConnection, DbBackend, EntityTrait, ExecResult, ExecResultHolder, Iden,
-    Iterable, MockDatabaseConnection, MockDatabaseTrait, ModelTrait, QueryResult, QueryResultRow,
-    Statement,
+    IdenStatic, Iterable, MockDatabaseConnection, MockDatabaseTrait, ModelTrait, QueryResult,
+    QueryResultRow, Statement, SelectA, SelectB
 };
 use sea_query::{Value, ValueType, Values};
 use std::{collections::BTreeMap, sync::Arc};
@@ -204,6 +204,33 @@ where
     }
 }
 
+impl<M, N> IntoMockRow for (M, N)
+where
+    M: ModelTrait,
+    N: ModelTrait,
+{
+    fn into_mock_row(self) -> MockRow {
+        let mut mapped_join = BTreeMap::new();
+
+        for column in <<M as ModelTrait>::Entity as EntityTrait>::Column::iter() {
+            mapped_join.insert(format!("{}{}", SelectA.as_str(), column.as_str()), self.0.get(column));
+        }
+        for column in <<N as ModelTrait>::Entity as EntityTrait>::Column::iter() {
+            mapped_join.insert(format!("{}{}", SelectB.as_str(), column.as_str()), self.1.get(column));
+        }
+
+        mapped_join.into_mock_row()
+    }
+}
+
+impl IntoMockRow for BTreeMap<String, Value> {
+    fn into_mock_row(self) -> MockRow {
+        MockRow {
+            values: self.into_iter().map(|(k, v)| (k, v)).collect(),
+        }
+    }
+}
+
 impl IntoMockRow for BTreeMap<&str, Value> {
     fn into_mock_row(self) -> MockRow {
         MockRow {
@@ -312,7 +339,7 @@ impl OpenTransaction {
 mod tests {
     use crate::{
         entity::*, tests_cfg::*, ConnectionTrait, DbBackend, DbErr, MockDatabase, Statement,
-        Transaction, TransactionError,
+        Transaction, TransactionError, IntoMockRow,
     };
     use pretty_assertions::assert_eq;
 
@@ -605,4 +632,26 @@ mod tests {
 
         Ok(())
     }
+
+    #[smol_potat::test]
+    async fn test_mocked_join() {
+        let mocked_row = (
+            cake::Model {
+                id: 1,
+                name: "Apple Cake".to_owned(),
+            },
+            fruit::Model {
+                id: 2,
+                name: "Apple".to_owned(),
+                cake_id: Some(1),
+            }
+        ).into_mock_row();
+
+        let a_id = mocked_row.try_get::<i32>("A_id");
+        assert!(a_id.is_ok());
+        assert_eq!(1, a_id.unwrap());
+        let b_id = mocked_row.try_get::<i32>("B_id");
+        assert!(b_id.is_ok());
+        assert_eq!(2, b_id.unwrap());
+    }
 }