bug: fix broken check from last commit, add tests

3 files changed, 27 insertions, 0 deletions

diff --git a/tests/arg_tests.rs b/tests/arg_tests.rs
index f6dac340..6f19eb8c 100644
--- a/tests/arg_tests.rs
+++ b/tests/arg_tests.rs
@@ -156,3 +156,17 @@ fn test_invalid_default_widget_2() -> Result<(), Box<dyn std::error::Error>> {
 
     Ok(())
 }
+
+#[test]
+fn test_missing_default_widget_type() -> Result<(), Box<dyn std::error::Error>> {
+    Command::new(get_binary_location())
+        .arg("--default_widget_count")
+        .arg("3")
+        .assert()
+        .failure()
+        .stderr(predicate::str::contains(
+            "The following required arguments were not provided",
+        ));
+
+    Ok(())
+}
diff --git a/tests/invalid_config_tests.rs b/tests/invalid_config_tests.rs
index cda6beb5..49db85da 100644
--- a/tests/invalid_config_tests.rs
+++ b/tests/invalid_config_tests.rs
@@ -147,3 +147,14 @@ fn test_empty_battery() -> Result<(), Box<dyn std::error::Error>> {
         ));
     Ok(())
 }
+
+#[test]
+fn test_invalid_default_widget_count() -> Result<(), Box<dyn std::error::Error>> {
+    Command::new(get_binary_location())
+        .arg("-C")
+        .arg("./tests/invalid_configs/invalid_default_widget_count.toml")
+        .assert()
+        .failure()
+        .stderr(predicate::str::contains("it must be used with"));
+    Ok(())
+}
diff --git a/tests/invalid_configs/invalid_default_widget_count.toml b/tests/invalid_configs/invalid_default_widget_count.toml
new file mode 100644
index 00000000..0e299d12
--- /dev/null
+++ b/tests/invalid_configs/invalid_default_widget_count.toml
@@ -0,0 +1,2 @@
+[flags]
+default_widget_count = 3
\ No newline at end of file