test exit scenarios

src/cmd/fix.rs

@@ -85,9 +85,15 @@ impl Cmd {
                 print!("{}", fixes[selection]);
                 Ok(())
             }
-            Ok(None) => Ok(()),
+            Ok(None) => {
+                eprintln!("Cancelled.");
+                Ok(())
+            }
             // Do not throw an error when Ctrl-C is pressed.
-            Err(dialoguer::Error::IO(e)) if e.kind() == io::ErrorKind::Interrupted => Ok(()),
+            Err(dialoguer::Error::IO(e)) if e.kind() == io::ErrorKind::Interrupted => {
+                eprintln!("Cancelled.");
+                Ok(())
+            }
             Err(e) => Err(Error::Select(e)),
         }
     }

tests/bash.rs

@@ -54,3 +54,16 @@ fn nothing_to_fix(bash: (Session, NamedTempFile)) {
     p.send_line("").unwrap();
     p.expect("nothing to fix").unwrap();
 }
+
+#[rstest]
+fn quit(bash: (Session, NamedTempFile)) {
+    let (mut p, _histfile) = bash;
+
+    p.set_expect_timeout(Some(Duration::from_secs(5)));
+
+    p.send_line("eco 'Hello, world!'").unwrap();
+    p.expect("command not found").unwrap();
+    p.send_line("fix").unwrap();
+    p.send_line("q").unwrap();
+    p.expect("Cancelled.").unwrap();
+}

tests/ui.rs

@@ -0,0 +1,20 @@
+use std::{io::Read, process::Command, thread::sleep, time::Duration};
+
+use expectrl::{Session, Signal};
+
+#[test]
+fn ctrl_c() {
+    let mut cmd = Command::new(format!("{}/target/debug/fixit", env!("CARGO_MANIFEST_DIR")));
+    cmd.env("FIXIT_QUICK_ENABLE", "false");
+    cmd.arg("eco 'Hello, world!'");
+
+    let mut p = Session::spawn(cmd).unwrap();
+
+    sleep(Duration::from_secs(1));
+
+    p.get_process_mut().signal(Signal::SIGTERM).unwrap();
+
+    let mut buf = String::new();
+    p.read_to_string(&mut buf).unwrap();
+    buf.contains("Cacelled.");
+}