Edit test case

SeaQL · Sep 25, 2022 · f3b7feb · f3b7feb
1 parent e29a11e
commit f3b7feb
Showing 1 changed file with 3 additions and 2 deletions.
diff --git 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());
     }
 }