From 620381cf23b8244f3fde732b2c7cc5df96b06e4d Mon Sep 17 00:00:00 2001
From: ForzenString <964413011@qq.com>
Date: Sun, 3 Mar 2024 16:41:12 +0800
Subject: [PATCH] :white_check_mark: fix test

---
 sea-orm-macros/src/derives/partial_model.rs | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/sea-orm-macros/src/derives/partial_model.rs b/sea-orm-macros/src/derives/partial_model.rs
index 1f95c40f5..0413fc86b 100644
--- a/sea-orm-macros/src/derives/partial_model.rs
+++ b/sea-orm-macros/src/derives/partial_model.rs
@@ -228,7 +228,7 @@ mod util {
 #[cfg(test)]
 mod test {
     use quote::format_ident;
-    use syn::DeriveInput;
+    use syn::{parse_str, DeriveInput, Type};
 
     use crate::derives::partial_model::ColumnAs;
 
@@ -250,11 +250,10 @@ struct PartialModel{
 "#;
     #[test]
     fn test_load_macro_input() -> StdResult<()> {
-        let input = syn::parse_str::<DeriveInput>(CODE_SNIPPET)?;
+        let input = parse_str::<DeriveInput>(CODE_SNIPPET)?;
 
         let middle = DerivePartialModel::new(input).unwrap();
-
-        assert_eq!(middle.entity, Some(format_ident!("Entity")));
+        assert_eq!(middle.entity, Some(parse_str::<Type>("Entity").unwrap()));
         assert_eq!(middle.ident, format_ident!("PartialModel"));
         assert_eq!(middle.fields.len(), 3);
         assert_eq!(