From 32f82a0c9b8722048a0439f250b20728526b3fb2 Mon Sep 17 00:00:00 2001
From: Billy Chan <ccw.billy.123@gmail.com>
Date: Mon, 18 Oct 2021 17:39:30 +0800
Subject: [PATCH] Test self referencing relation

---
 tests/common/features/mod.rs       |  2 +
 tests/common/features/schema.rs    | 27 ++++++++-
 tests/common/features/self_join.rs | 77 ++++++++++++++++++++++++++
 tests/self_join_tests.rs           | 89 ++++++++++++++++++++++++++++++
 4 files changed, 194 insertions(+), 1 deletion(-)
 create mode 100644 tests/common/features/self_join.rs
 create mode 100644 tests/self_join_tests.rs

diff --git a/tests/common/features/mod.rs b/tests/common/features/mod.rs
index ff716f01..1e709edf 100644
--- a/tests/common/features/mod.rs
+++ b/tests/common/features/mod.rs
@@ -2,8 +2,10 @@ pub mod applog;
 pub mod metadata;
 pub mod repository;
 pub mod schema;
+pub mod self_join;
 
 pub use applog::Entity as Applog;
 pub use metadata::Entity as Metadata;
 pub use repository::Entity as Repository;
 pub use schema::*;
+pub use self_join::Entity as SelfJoin;
diff --git a/tests/common/features/schema.rs b/tests/common/features/schema.rs
index 44b011c5..7e6f6424 100644
--- a/tests/common/features/schema.rs
+++ b/tests/common/features/schema.rs
@@ -3,12 +3,13 @@ pub use super::super::bakery_chain::*;
 use super::*;
 use crate::common::setup::create_table;
 use sea_orm::{error::*, sea_query, DatabaseConnection, DbConn, ExecResult};
-use sea_query::ColumnDef;
+use sea_query::{ColumnDef, ForeignKeyCreateStatement};
 
 pub async fn create_tables(db: &DatabaseConnection) -> Result<(), DbErr> {
     create_log_table(db).await?;
     create_metadata_table(db).await?;
     create_repository_table(db).await?;
+    create_self_join_table(db).await?;
 
     Ok(())
 }
@@ -75,3 +76,27 @@ pub async fn create_repository_table(db: &DbConn) -> Result<ExecResult, DbErr> {
 
     create_table(db, &stmt, Repository).await
 }
+
+pub async fn create_self_join_table(db: &DbConn) -> Result<ExecResult, DbErr> {
+    let stmt = sea_query::Table::create()
+        .table(self_join::Entity)
+        .col(
+            ColumnDef::new(self_join::Column::Uuid)
+                .uuid()
+                .not_null()
+                .primary_key(),
+        )
+        .col(ColumnDef::new(self_join::Column::UuidRef).uuid())
+        .col(ColumnDef::new(self_join::Column::Time).time())
+        .foreign_key(
+            ForeignKeyCreateStatement::new()
+                .name("fk-self_join-self_join")
+                .from_tbl(SelfJoin)
+                .from_col(self_join::Column::UuidRef)
+                .to_tbl(SelfJoin)
+                .to_col(self_join::Column::Uuid),
+        )
+        .to_owned();
+
+    create_table(db, &stmt, SelfJoin).await
+}
diff --git a/tests/common/features/self_join.rs b/tests/common/features/self_join.rs
new file mode 100644
index 00000000..de491a09
--- /dev/null
+++ b/tests/common/features/self_join.rs
@@ -0,0 +1,77 @@
+use sea_orm::entity::prelude::*;
+
+#[derive(Clone, Debug, PartialEq, DeriveEntityModel)]
+#[sea_orm(table_name = "self_join")]
+pub struct Model {
+    #[sea_orm(primary_key, auto_increment = false)]
+    pub uuid: Uuid,
+    pub uuid_ref: Option<Uuid>,
+    pub time: Option<Time>,
+}
+
+#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
+pub enum Relation {
+    #[sea_orm(belongs_to = "Entity", from = "Column::UuidRef", to = "Column::Uuid")]
+    SelfReferencing,
+}
+
+pub struct SelfReferencingLink;
+
+impl Linked for SelfReferencingLink {
+    type FromEntity = Entity;
+
+    type ToEntity = Entity;
+
+    fn link(&self) -> Vec<RelationDef> {
+        vec![Relation::SelfReferencing.def()]
+    }
+}
+
+impl ActiveModelBehavior for ActiveModel {}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+    use pretty_assertions::assert_eq;
+    use sea_orm::*;
+
+    #[test]
+    fn find_linked_001() {
+        let self_join_model = Model {
+            uuid: Uuid::default(),
+            uuid_ref: None,
+            time: None,
+        };
+
+        assert_eq!(
+            self_join_model
+                .find_linked(SelfReferencingLink)
+                .build(DbBackend::MySql)
+                .to_string(),
+            [
+                r#"SELECT `self_join`.`uuid`, `self_join`.`uuid_ref`, `self_join`.`time`"#,
+                r#"FROM `self_join`"#,
+                r#"INNER JOIN `self_join` AS `r0` ON `r0`.`uuid_ref` = `self_join`.`uuid`"#,
+                r#"WHERE `r0`.`uuid` = '00000000-0000-0000-0000-000000000000'"#,
+            ]
+            .join(" ")
+        );
+    }
+
+    #[test]
+    fn find_also_linked_001() {
+        assert_eq!(
+            Entity::find()
+                .find_also_linked(SelfReferencingLink)
+                .build(DbBackend::MySql)
+                .to_string(),
+            [
+                r#"SELECT `self_join`.`uuid` AS `A_uuid`, `self_join`.`uuid_ref` AS `A_uuid_ref`, `self_join`.`time` AS `A_time`,"#,
+                r#"`r0`.`uuid` AS `B_uuid`, `r0`.`uuid_ref` AS `B_uuid_ref`, `r0`.`time` AS `B_time`"#,
+                r#"FROM `self_join`"#,
+                r#"LEFT JOIN `self_join` AS `r0` ON `self_join`.`uuid_ref` = `r0`.`uuid`"#,
+            ]
+            .join(" ")
+        );
+    }
+}
diff --git a/tests/self_join_tests.rs b/tests/self_join_tests.rs
new file mode 100644
index 00000000..0c85a070
--- /dev/null
+++ b/tests/self_join_tests.rs
@@ -0,0 +1,89 @@
+pub mod common;
+
+pub use common::{features::*, setup::*, TestContext};
+use pretty_assertions::assert_eq;
+use sea_orm::{entity::prelude::*, *};
+
+#[sea_orm_macros::test]
+#[cfg(any(
+    feature = "sqlx-mysql",
+    feature = "sqlx-sqlite",
+    feature = "sqlx-postgres"
+))]
+async fn main() -> Result<(), DbErr> {
+    let ctx = TestContext::new("self_join_tests").await;
+    create_tables(&ctx.db).await?;
+    create_metadata(&ctx.db).await?;
+    ctx.delete().await;
+
+    Ok(())
+}
+
+pub async fn create_metadata(db: &DatabaseConnection) -> Result<(), DbErr> {
+    let model = self_join::Model {
+        uuid: Uuid::new_v4(),
+        uuid_ref: None,
+        time: Some(Time::from_hms(1, 00, 00)),
+    };
+
+    model.clone().into_active_model().insert(db).await?;
+
+    let linked_model = self_join::Model {
+        uuid: Uuid::new_v4(),
+        uuid_ref: Some(model.clone().uuid),
+        time: Some(Time::from_hms(2, 00, 00)),
+    };
+
+    linked_model.clone().into_active_model().insert(db).await?;
+
+    let not_linked_model = self_join::Model {
+        uuid: Uuid::new_v4(),
+        uuid_ref: None,
+        time: Some(Time::from_hms(3, 00, 00)),
+    };
+
+    not_linked_model
+        .clone()
+        .into_active_model()
+        .insert(db)
+        .await?;
+
+    assert_eq!(
+        model
+            .find_linked(self_join::SelfReferencingLink)
+            .all(db)
+            .await?,
+        vec![]
+    );
+
+    assert_eq!(
+        linked_model
+            .find_linked(self_join::SelfReferencingLink)
+            .all(db)
+            .await?,
+        vec![model.clone()]
+    );
+
+    assert_eq!(
+        not_linked_model
+            .find_linked(self_join::SelfReferencingLink)
+            .all(db)
+            .await?,
+        vec![]
+    );
+
+    assert_eq!(
+        self_join::Entity::find()
+            .find_also_linked(self_join::SelfReferencingLink)
+            .order_by_asc(self_join::Column::Time)
+            .all(db)
+            .await?,
+        vec![
+            (model.clone(), None),
+            (linked_model, Some(model)),
+            (not_linked_model, None),
+        ]
+    );
+
+    Ok(())
+}