From fc27cd6e5f069a21483b234afcbaabec196a1ac5 Mon Sep 17 00:00:00 2001
From: Thomas Churchman <thomas@kepow.org>
Date: Sun, 30 Jun 2024 17:47:41 +0200
Subject: [PATCH] test: add test calling out to bash

---
 src/main.rs  | 49 +++++++++++++++++++++++++++++--------------------
 src/tests.rs | 23 +++++++++++++++++++++++
 2 files changed, 52 insertions(+), 20 deletions(-)
 create mode 100644 src/tests.rs

diff --git a/src/main.rs b/src/main.rs
index 4268c84..d44c1de 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -25,6 +25,9 @@ mod poll;
 mod ringbuffer;
 use ringbuffer::{IoResult, Ringbuffer};
 
+#[cfg(test)]
+mod tests;
+
 const DEFAULT_NUM_LINES: u16 = 24;
 const DEFAULT_NUM_COLUMNS: u16 = 80;
 
@@ -344,8 +347,32 @@ fn from_read(read: &mut impl Read, lines: u16, columns: u16) -> anyhow::Result<S
 }
 
 fn main() -> anyhow::Result<()> {
-    let cli = Cli::parse();
+    let mut cli = Cli::parse();
+    let out = cli.out.take();
+    let screen = run(cli)?;
+
+    let fonts = &[
+        "ui-monospace",
+        "Consolas",
+        "Liberation Mono",
+        "Source Code Pro",
+    ];
+
+    if let Some(out) = out {
+        let mut file = std::fs::OpenOptions::new()
+            .write(true)
+            .truncate(true)
+            .create(true)
+            .open(out)?;
+        write!(file, "{}", screen.to_svg(fonts))?;
+    } else {
+        println!("{}", screen.to_svg(fonts))
+    }
+
+    Ok(())
+}
 
+fn run(cli: Cli) -> anyhow::Result<Screen> {
     if cli.interactive {
         if cli.out.is_none() {
             anyhow::bail!("`--interactive` is set but no SVG output file is specified in `--out`. See `termsnap --help`.");
@@ -433,23 +460,5 @@ fn main() -> anyhow::Result<()> {
         }
     };
 
-    let fonts = &[
-        "ui-monospace",
-        "Consolas",
-        "Liberation Mono",
-        "Source Code Pro",
-    ];
-
-    if let Some(out) = cli.out {
-        let mut file = std::fs::OpenOptions::new()
-            .write(true)
-            .truncate(true)
-            .create(true)
-            .open(out)?;
-        write!(file, "{}", screen.to_svg(fonts))?;
-    } else {
-        println!("{}", screen.to_svg(fonts))
-    }
-
-    Ok(())
+    Ok(screen)
 }
diff --git a/src/tests.rs b/src/tests.rs
new file mode 100644
index 0000000..9408b92
--- /dev/null
+++ b/src/tests.rs
@@ -0,0 +1,23 @@
+use clap::Parser;
+
+use super::{run, Cli};
+
+#[test]
+fn bash_echo() {
+    let cli = Cli::parse_from([
+        "termsnap",
+        "-l",
+        "20",
+        "-c",
+        "80",
+        "--",
+        "bash",
+        "-c",
+        "echo 'hello, world'",
+    ]);
+
+    let screen = run(cli).unwrap();
+    let content: String = screen.cells().map(|c| c.c).collect();
+
+    assert!(content.starts_with("hello, world"));
+}