fix(pilota-thrift-parser): add the list end blank test case for annot…

…ation parser
cloudwego · Nov 4, 2024 · 40e2fbb · 40e2fbb
1 parent 6e78a7f
commit 40e2fbb
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 8 deletions.
diff --git a/pilota-thrift-parser/src/parser/annotation.rs b/pilota-thrift-parser/src/parser/annotation.rs
@@ -51,5 +51,14 @@ mod tests {
     #[test]
     fn test_annotations() {
         let _a = Annotations::parse(r#"(go.tag = "json:\"Ids\" split:\"type=tenant\"")"#).unwrap();
+
+        let input = r#"(
+            cpp.type = "DenseFoo",
+            python.type = "DenseFoo",
+            java.final = "",
+            )"#;
+        let (remain, a) = Annotations::parse(input).unwrap();
+        assert!(remain.is_empty());
+        assert_eq!(a.len(), 3);
     }
 }
diff --git a/pilota-thrift-parser/src/parser/struct_.rs b/pilota-thrift-parser/src/parser/struct_.rs
@@ -95,14 +95,8 @@ mod tests {
         let str = r#"struct ImMsgContent {
             1: string user_id (go.tag = 'json:\"user_id,omitempty\"'),
             2: string __files (go.tag = 'json:\"__files,omitempty\"'),
-        }
-        (
-        cpp.type = "DenseFoo",
-        python.type = "DenseFoo",
-        java.final = "",
-        )"#;
-        let (remain, _) = Struct::parse(str).unwrap();
-        assert!(remain.is_empty());
+        }"#;
+        Struct::parse(str).unwrap();
     }
 
     #[test]