From 60ec9e0719216ad916d8d8dd2f776c5ea604ebd7 Mon Sep 17 00:00:00 2001
From: Benjamin Wheeler <benjaminwheeler0510@gmail.com>
Date: Tue, 18 Feb 2025 22:38:42 -0500
Subject: [PATCH] Add unit tests for `--winter`, `--debug`, `--terrain` flags.

---
 src/args.rs | 30 ++++++++++++++++++++++++++++++
 1 file changed, 30 insertions(+)

diff --git a/src/args.rs b/src/args.rs
index 22c245c..9412954 100644
--- a/src/args.rs
+++ b/src/args.rs
@@ -105,3 +105,33 @@ fn parse_duration(arg: &str) -> Result<std::time::Duration, std::num::ParseIntEr
     let seconds = arg.parse()?;
     Ok(std::time::Duration::from_secs(seconds))
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn test_flags() {
+        // Test that winter/terrain/debug are SetTrue
+        let cmd = vec![
+            "arnis",
+            "--path",
+            "",
+            "--bbox",
+            "",
+            "--winter",
+            "--terrain",
+            "--debug",
+        ];
+        let args = Args::parse_from(cmd.iter());
+        assert_eq!(args.winter, true);
+        assert_eq!(args.debug, true);
+        assert_eq!(args.terrain, true);
+
+        let cmd = vec!["arnis", "--path", "", "--bbox", ""];
+        let args = Args::parse_from(cmd.iter());
+        assert_eq!(args.winter, false);
+        assert_eq!(args.debug, false);
+        assert_eq!(args.terrain, false);
+    }
+}