| 849 | |
| 850 | #[test] |
| 851 | fn test_extract_class() { |
| 852 | let mut parser = CppParser::new(); |
| 853 | let source = r#" |
| 854 | class User { |
| 855 | public: |
| 856 | std::string name; |
| 857 | int age; |
| 858 | }; |
| 859 | "#; |
| 860 | let entities = parser.extract(source, "user.cpp"); |
| 861 | let user = entities.iter().find(|e| e.name == "User").unwrap(); |
| 862 | assert_eq!(user.kind, EntityKind::Class); |
| 863 | assert!( |
| 864 | user.signature.as_ref().unwrap().contains("class User"), |
| 865 | "Signature: {:?}", |
| 866 | user.signature |
| 867 | ); |
| 868 | } |
| 869 | |
| 870 | #[test] |
| 871 | fn test_extract_struct() { |