Testing joins with MockDatabase

[billy1624]

Is it possible to somehow have a function that generates such a map given two Entity structs? Like iterating all columns adding A_ to the first and B_ to the second

Originally posted by @cemoktra in #443 (reply in thread)

---

Just as a proposal:

fn entity_to_prefixed_btreemap<T>(model: T::Model, prefix: &str) -> BTreeMap<String, sea_orm::Value>
    where
        T: EntityTrait,
{
    let mut mapped = BTreeMap::new();
    for column in T::Column::iter() {
        mapped.insert(format!("{}{}", prefix, column.as_str()), model.get(column));
    }
    mapped
}

and then:

fn mock_join<T1, T2>(model1: T1::Model, model2: T2::Model) -> BTreeMap<String, sea_orm::Value>
    where
        T1: EntityTrait,
        T2: EntityTrait,        
{
    let mut map = entity_to_prefixed_btreemap::<T1>(model1, "A_");
    map.extend(entity_to_prefixed_btreemap::<T2>(model2, "B_"));
    map
}

Originally posted by @cemoktra in #443 (reply in thread)

[cemoktra]

Note that this BTreeMap has owned String as key, so it might be also useful to implement IntoMockRow for BTreeMap<String, Value>.

And the above can be optimized to be generic only over ModelTrait

But maybe an alternative would be to have IntoMockRow implemented for (Model1, Model2) as external interface:

fn into_mock_row(self) -> sea_orm::MockRow {
    mock_join::<M1::Entity, M2::Entity>(self.0, self.1)
}

[cemoktra]

Putting it all together:

pub struct MockJoin<M1, M2>(M1, M2);

impl<M1, M2> IntoMockRow for MockJoin<M1, M2>
where
    M1: ModelTrait,
    M2: ModelTrait,
{
    fn into_mock_row(self) -> sea_orm::MockRow {
        let mut mapped_join = BTreeMap::new();

        for column in <<M1 as sea_orm::ModelTrait>::Entity as EntityTrait>::Column::iter() {
            mapped_join.insert(format!("A_{}", column.as_str()), self.0.get(column));
        }
        for column in <<M2 as sea_orm::ModelTrait>::Entity as EntityTrait>::Column::iter() {
            mapped_join.insert(format!("B_{}", column.as_str()), self.1.get(column));
        }

        mapped_join
            .iter()
            .map(|(key, value)| (key.as_str(), value.to_owned()))
            .collect::<BTreeMap<&str, sea_orm::Value>>()
            .into_mock_row()
    }
}

[billy1624]

And perhaps...

impl<M, N> IntoMockRow for (M, N)
where
    M: ModelTrait,
    N: ModelTrait,
{
    fn into_mock_row(self) -> MockRow { ... }
}

impl IntoMockRow for (BTreeMap<&str, Value>, BTreeMap<&str, Value>) {
    fn into_mock_row(self) -> MockRow { ... }
}

[cemoktra]

Yeah exactly