other: add test for battery flag if battery feature is off

1 files changed, 13 insertions, 0 deletions

diff --git a/tests/arg_tests.rs b/tests/arg_tests.rs
index 649b1cfb..95afb635 100644
--- a/tests/arg_tests.rs
+++ b/tests/arg_tests.rs
@@ -168,3 +168,16 @@ fn test_missing_default_widget_type() {
             "The following required arguments were not provided",
         ));
 }
+
+#[test]
+fn test_battery_flag() {
+    if !cfg!(feature = "battery") {
+        Command::new(get_binary_location())
+            .arg("--battery")
+            .assert()
+            .failure()
+            .stderr(predicate::str::contains(
+                "'--battery' which wasn't expected",
+            ));
+    }
+}