From f3b7febc2e01d343b4b52f8a26eab2fe44313293 Mon Sep 17 00:00:00 2001
From: Billy Chan <ccw.billy.123@gmail.com>
Date: Sun, 25 Sep 2022 10:43:50 +0800
Subject: [PATCH] Edit test case

---
 sea-orm-codegen/src/entity/base_entity.rs | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/sea-orm-codegen/src/entity/base_entity.rs b/sea-orm-codegen/src/entity/base_entity.rs
index 7e2ef8a9f..0147c6f06 100644
--- a/sea-orm-codegen/src/entity/base_entity.rs
+++ b/sea-orm-codegen/src/entity/base_entity.rs
@@ -166,7 +166,7 @@ impl Entity {
 
 #[cfg(test)]
 mod tests {
-    use quote::format_ident;
+    use quote::{format_ident, quote};
     use sea_query::{ColumnType, ForeignKeyAction};
 
     use crate::{Column, DateTimeCrate, Entity, PrimaryKey, Relation, RelationType};
@@ -438,7 +438,8 @@ mod tests {
     #[test]
     fn test_get_eq_needed() {
         let entity = setup();
+        let expected = quote! {, Eq};
 
-        println!("entity: {:?}", entity.get_eq_needed());
+        assert_eq!(entity.get_eq_needed().to_string(), expected.to_string());
     }
 }