sea-orm-sync 2.0.0-rc.38

🐚 The sync version of SeaORM
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115

use crate::common::TestContext;
use crate::local::{bakery, create_tables, worker};
use sea_orm::{
    DbBackend, DerivePartialModel, IntoActiveModel, JoinType, NotSet, QueryOrder, QuerySelect,
    QueryTrait, Set, prelude::*, sea_query::Alias,
};

#[derive(DerivePartialModel)]
#[sea_orm(entity = "worker::Entity")]
struct Worker {
    id: i32,
    name: String,
}

#[derive(DerivePartialModel)]
#[sea_orm(entity = "bakery::Entity")]
struct BakeryWorker {
    id: i32,
    name: String,
    profit_margin: f64,
    #[sea_orm(nested, alias = "manager")]
    manager: Worker,
    #[sea_orm(nested, alias = "cashier")]
    cashier: worker::Model,
}

#[derive(DerivePartialModel)]
#[sea_orm(entity = "worker::Entity")]
struct ManagerOfBakery {
    id: i32,
    name: String,
    #[sea_orm(nested)]
    bakery: bakery::Model,
}

#[sea_orm_macros::test]
fn partial_model_nested_alias() {
    let ctx = TestContext::new("partial_model_nested");
    create_tables(&ctx.db).expect("unable to create tables");

    // TODO: init utils

    worker::Entity::insert(worker::ActiveModel {
        id: Set(1),
        name: Set("Tom".to_owned()),
        ..Default::default()
    })
    .exec(&ctx.db)
    .expect("insert succeeds");

    worker::Entity::insert(worker::ActiveModel {
        id: Set(2),
        name: Set("Jerry".to_owned()),
        ..Default::default()
    })
    .exec(&ctx.db)
    .expect("insert succeeds");

    bakery::Entity::insert(bakery::ActiveModel {
        id: Set(42),
        name: Set("Master Bakery".to_string()),
        profit_margin: Set(4.1),
        manager_id: Set(1),
        cashier_id: Set(2),
    })
    .exec(&ctx.db)
    .expect("insert succeeds");

    let selector = bakery::Entity::find()
        .join_as(
            sea_orm::JoinType::LeftJoin,
            bakery::Relation::Manager.def(),
            "manager",
        )
        .join_as(
            sea_orm::JoinType::LeftJoin,
            bakery::Relation::Cashier.def(),
            "cashier",
        );

    assert_eq!(
        selector.build(DbBackend::MySql).to_string(),
        "SELECT `bakery`.`id`, `bakery`.`name`, `bakery`.`profit_margin`, `bakery`.`manager_id`, `bakery`.`cashier_id` FROM `bakery` LEFT JOIN `worker` AS `manager` ON `bakery`.`manager_id` = `manager`.`id` LEFT JOIN `worker` AS `cashier` ON `bakery`.`cashier_id` = `cashier`.`id`"
    );

    let bakery: BakeryWorker = selector
        .into_partial_model()
        .one(&ctx.db)
        .expect("succeeds to get the result")
        .expect("exactly one model in DB");

    assert_eq!(bakery.manager.name, "Tom");
    assert_eq!(bakery.cashier.name, "Jerry");

    let selector = worker::Entity::find().join(
        sea_orm::JoinType::LeftJoin,
        worker::Relation::BakeryManager.def(),
    );

    assert_eq!(
        selector.build(DbBackend::MySql).to_string(),
        "SELECT `worker`.`id`, `worker`.`name` FROM `worker` LEFT JOIN `bakery` ON `worker`.`id` = `bakery`.`manager_id`"
    );

    let manager: ManagerOfBakery = selector
        .into_partial_model()
        .one(&ctx.db)
        .expect("succeeds to get the result")
        .expect("exactly one model in DB");

    assert_eq!(manager.name, "Tom");
    assert_eq!(manager.bakery.name, "Master Bakery");

    ctx.delete();
}