From b302a115866f9c322713a9d344485de91fcb0e27 Mon Sep 17 00:00:00 2001
From: Sam Samai <sam@studio2pi.com.au>
Date: Tue, 13 Jul 2021 21:11:11 +1000
Subject: [PATCH] Test left_join

---
 tests/relational_tests.rs | 75 ++++++++++++++++++++++++++++++---------
 1 file changed, 58 insertions(+), 17 deletions(-)

diff --git a/tests/relational_tests.rs b/tests/relational_tests.rs
index 26b374c5..2a07db3a 100644
--- a/tests/relational_tests.rs
+++ b/tests/relational_tests.rs
@@ -1,10 +1,7 @@
-use sea_orm::{entity::*, InsertResult};
-
-// pub mod bakery_chain;
-// pub use bakery_chain::*;
+use sea_orm::{entity::*, query::*, FromQueryResult};
 
 pub mod common;
-pub use common::{setup::*, TestContext};
+pub use common::{bakery_chain::*, setup::*, TestContext};
 
 #[async_std::test]
 // cargo test --test realtional_tests -- --nocapture
@@ -14,25 +11,69 @@ async fn main() {
 
 pub async fn test_left_join() {
     let ctx = TestContext::new("mysql://root:@localhost", "test_left_join").await;
-    let seaside_bakery = bakery::ActiveModel {
+
+    let bakery = bakery::ActiveModel {
         name: Set("SeaSide Bakery".to_owned()),
         profit_margin: Set(10.4),
         ..Default::default()
-    };
-    let res: InsertResult = Bakery::insert(seaside_bakery)
-        .exec(&ctx.db)
-        .await
-        .expect("could not insert bakery");
+    }
+    .save(&ctx.db)
+    .await
+    .expect("could not insert bakery");
 
-    let bakery: Option<bakery::Model> = Bakery::find_by_id(res.last_insert_id)
+    let _baker_1 = baker::ActiveModel {
+        name: Set("Baker 1".to_owned()),
+        bakery_id: Set(Some(bakery.id.clone().unwrap())),
+        ..Default::default()
+    }
+    .save(&ctx.db)
+    .await
+    .expect("could not insert baker");
+
+    let _baker_2 = baker::ActiveModel {
+        name: Set("Baker 2".to_owned()),
+        bakery_id: Set(None),
+        ..Default::default()
+    }
+    .save(&ctx.db)
+    .await
+    .expect("could not insert baker");
+
+    #[derive(Debug, FromQueryResult)]
+    struct SelectResult {
+        name: String,
+        bakery_name: Option<String>,
+    }
+
+    let select = baker::Entity::find()
+        .left_join(bakery::Entity)
+        .select_only()
+        .column(baker::Column::Name)
+        .column_as(bakery::Column::Name, "bakery_name")
+        .filter(baker::Column::Name.contains("Baker 1"));
+
+    let result = select
+        .into_model::<SelectResult>()
         .one(&ctx.db)
         .await
-        .expect("could not find bakery");
+        .unwrap()
+        .unwrap();
+    assert_eq!(result.bakery_name, Some("SeaSide Bakery".to_string()));
 
-    assert!(bakery.is_some());
-    let bakery_model = bakery.unwrap();
-    assert_eq!(bakery_model.name, "SeaSide Bakery");
-    assert_eq!(bakery_model.profit_margin, 10.4);
+    let select = baker::Entity::find()
+        .left_join(bakery::Entity)
+        .select_only()
+        .column(baker::Column::Name)
+        .column_as(bakery::Column::Name, "bakery_name")
+        .filter(baker::Column::Name.contains("Baker 2"));
+
+    let result = select
+        .into_model::<SelectResult>()
+        .one(&ctx.db)
+        .await
+        .unwrap()
+        .unwrap();
+    assert_eq!(result.bakery_name, None);
 
     ctx.delete().await;
 }