aft/
lib.rs

1pub mod ast_grep_lang;
2pub mod backup;
3pub mod callgraph;
4pub mod calls;
5pub mod checkpoint;
6pub mod commands;
7pub mod config;
8pub mod context;
9pub mod edit;
10pub mod error;
11pub mod extract;
12pub mod format;
13pub mod fuzzy_match;
14pub mod imports;
15pub mod indent;
16pub mod language;
17pub mod lsp;
18pub mod lsp_hints;
19pub mod parser;
20pub mod protocol;
21pub mod symbols;
22
23#[cfg(test)]
24mod tests {
25    use super::*;
26    use config::Config;
27    use error::AftError;
28    use protocol::{RawRequest, Response};
29
30    // --- Protocol serialization ---
31
32    #[test]
33    fn raw_request_deserializes_ping() {
34        let json = r#"{"id":"1","command":"ping"}"#;
35        let req: RawRequest = serde_json::from_str(json).unwrap();
36        assert_eq!(req.id, "1");
37        assert_eq!(req.command, "ping");
38        assert!(req.lsp_hints.is_none());
39    }
40
41    #[test]
42    fn raw_request_deserializes_echo_with_params() {
43        let json = r#"{"id":"2","command":"echo","message":"hello"}"#;
44        let req: RawRequest = serde_json::from_str(json).unwrap();
45        assert_eq!(req.id, "2");
46        assert_eq!(req.command, "echo");
47        // "message" is captured in the flattened params
48        assert_eq!(req.params["message"], "hello");
49    }
50
51    #[test]
52    fn raw_request_preserves_unknown_fields() {
53        let json = r#"{"id":"3","command":"ping","future_field":"abc","nested":{"x":1}}"#;
54        let req: RawRequest = serde_json::from_str(json).unwrap();
55        assert_eq!(req.params["future_field"], "abc");
56        assert_eq!(req.params["nested"]["x"], 1);
57    }
58
59    #[test]
60    fn raw_request_with_lsp_hints() {
61        let json = r#"{"id":"4","command":"ping","lsp_hints":{"completions":["foo","bar"]}}"#;
62        let req: RawRequest = serde_json::from_str(json).unwrap();
63        assert!(req.lsp_hints.is_some());
64        let hints = req.lsp_hints.unwrap();
65        assert_eq!(hints["completions"][0], "foo");
66    }
67
68    #[test]
69    fn response_success_round_trip() {
70        let resp = Response::success("42", serde_json::json!({"command": "pong"}));
71        let json_str = serde_json::to_string(&resp).unwrap();
72        let v: serde_json::Value = serde_json::from_str(&json_str).unwrap();
73        assert_eq!(v["id"], "42");
74        assert_eq!(v["success"], true);
75        assert_eq!(v["command"], "pong");
76    }
77
78    #[test]
79    fn response_error_round_trip() {
80        let resp = Response::error("99", "unknown_command", "unknown command: foo");
81        let json_str = serde_json::to_string(&resp).unwrap();
82        let v: serde_json::Value = serde_json::from_str(&json_str).unwrap();
83        assert_eq!(v["id"], "99");
84        assert_eq!(v["success"], false);
85        assert_eq!(v["code"], "unknown_command");
86        assert_eq!(v["message"], "unknown command: foo");
87    }
88
89    // --- Error formatting ---
90
91    #[test]
92    fn error_display_symbol_not_found() {
93        let err = AftError::SymbolNotFound {
94            name: "foo".into(),
95            file: "bar.rs".into(),
96        };
97        assert_eq!(err.to_string(), "symbol 'foo' not found in bar.rs");
98        assert_eq!(err.code(), "symbol_not_found");
99    }
100
101    #[test]
102    fn error_display_ambiguous_symbol() {
103        let err = AftError::AmbiguousSymbol {
104            name: "Foo".into(),
105            candidates: vec!["a.rs:10".into(), "b.rs:20".into()],
106        };
107        let s = err.to_string();
108        assert!(s.contains("Foo"));
109        assert!(s.contains("a.rs:10, b.rs:20"));
110    }
111
112    #[test]
113    fn error_display_parse_error() {
114        let err = AftError::ParseError {
115            message: "unexpected token".into(),
116        };
117        assert_eq!(err.to_string(), "parse error: unexpected token");
118    }
119
120    #[test]
121    fn error_display_file_not_found() {
122        let err = AftError::FileNotFound {
123            path: "/tmp/missing.rs".into(),
124        };
125        assert_eq!(err.to_string(), "file not found: /tmp/missing.rs");
126    }
127
128    #[test]
129    fn error_display_invalid_request() {
130        let err = AftError::InvalidRequest {
131            message: "missing field".into(),
132        };
133        assert_eq!(err.to_string(), "invalid request: missing field");
134    }
135
136    #[test]
137    fn error_display_checkpoint_not_found() {
138        let err = AftError::CheckpointNotFound {
139            name: "pre-refactor".into(),
140        };
141        assert_eq!(err.to_string(), "checkpoint not found: pre-refactor");
142        assert_eq!(err.code(), "checkpoint_not_found");
143    }
144
145    #[test]
146    fn error_display_no_undo_history() {
147        let err = AftError::NoUndoHistory {
148            path: "src/main.rs".into(),
149        };
150        assert_eq!(err.to_string(), "no undo history for: src/main.rs");
151        assert_eq!(err.code(), "no_undo_history");
152    }
153
154    #[test]
155    fn error_display_ambiguous_match() {
156        let err = AftError::AmbiguousMatch {
157            pattern: "TODO".into(),
158            count: 5,
159        };
160        assert_eq!(
161            err.to_string(),
162            "pattern 'TODO' matches 5 occurrences, expected exactly 1"
163        );
164        assert_eq!(err.code(), "ambiguous_match");
165    }
166
167    #[test]
168    fn error_to_json_has_code_and_message() {
169        let err = AftError::FileNotFound { path: "/x".into() };
170        let j = err.to_error_json();
171        assert_eq!(j["code"], "file_not_found");
172        assert!(j["message"].as_str().unwrap().contains("/x"));
173    }
174
175    // --- Config defaults ---
176
177    #[test]
178    fn config_default_values() {
179        let cfg = Config::default();
180        assert!(cfg.project_root.is_none());
181        assert_eq!(cfg.validation_depth, 1);
182        assert_eq!(cfg.checkpoint_ttl_hours, 24);
183        assert_eq!(cfg.max_symbol_depth, 10);
184        assert_eq!(cfg.formatter_timeout_secs, 10);
185        assert_eq!(cfg.type_checker_timeout_secs, 30);
186    }
187}
aft/lib.rs

aft/
lib.rs
