1use std::env;
2use std::env::VarError;
3
4pub fn get() -> Result<String, VarError> {
6 match env::var("VISUAL") {
7 Ok(result) => return Ok(result),
8 Err(VarError::NotPresent) => {},
9 Err(error) => return Err(error),
10 }
11
12 match env::var("EDITOR") {
13 Ok(result) => return Ok(result),
14 Err(VarError::NotPresent) => {},
15 Err(error) => return Err(error),
16 }
17
18 Ok("vi".to_string())
19}
20
21#[cfg(test)]
22mod tests {
23 use std::env;
24
25 fn it_falls_back_to_vi() {
26 env::remove_var("VISUAL");
27 env::remove_var("EDITOR");
28
29 assert_eq!(super::get(), Ok("vi".to_string()));
30 }
31
32 fn it_returns_visual() {
33 env::set_var("VISUAL", "test1");
34 env::remove_var("EDITOR");
35
36 assert_eq!(super::get(), Ok("test1".to_string()));
37 }
38
39 fn it_returns_editor() {
40 env::remove_var("VISUAL");
41 env::set_var("EDITOR", "test2");
42
43 assert_eq!(super::get(), Ok("test2".to_string()));
44 }
45
46 fn it_returns_visual_before_editor() {
47 env::set_var("VISUAL", "test3");
48 env::set_var("EDITOR", "test4");
49
50 assert_eq!(super::get(), Ok("test3".to_string()));
51 }
52
53 #[test]
54 fn all_tests() {
55 it_falls_back_to_vi();
58 it_returns_visual();
59 it_returns_editor();
60 it_returns_visual_before_editor();
61 }
62}