From 182408a6ff1f4d31620f9f76ae964aa1fa99ca8f Mon Sep 17 00:00:00 2001
From: Geoffrey Washburn <geoff@washburght.org>
Date: Sat, 30 Nov 2024 16:02:38 -0500
Subject: [PATCH] Updating crate attributes

---
 src/lib.rs           |  2 +-
 src/main.rs          |  1 +
 tests/knoll_tests.rs | 17 +++++++++++++++--
 3 files changed, 17 insertions(+), 3 deletions(-)

diff --git a/src/lib.rs b/src/lib.rs
index 4265f20..45e185a 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -1,4 +1,4 @@
-#![feature(coverage_attribute)]
+#![cfg_attr(all(coverage_nightly, test), feature(coverage_attribute))]
 ///! Expose the knoll modules as part of a library.  This is only really
 /// necessary so that they will be visible in the `test` directory.
 #[macro_use]
diff --git a/src/main.rs b/src/main.rs
index 54e238e..dfb6879 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,3 +1,4 @@
+#![cfg_attr(all(coverage_nightly, test), feature(coverage_attribute))]
 // Objective-C is needed by the core_graphics module.
 // The core_graphics module also uses static_assertions to validate
 // some data structures at compile time.
diff --git a/tests/knoll_tests.rs b/tests/knoll_tests.rs
index 41fb082..8d46693 100644
--- a/tests/knoll_tests.rs
+++ b/tests/knoll_tests.rs
@@ -1,4 +1,4 @@
-#![feature(coverage_attribute)]
+#![cfg_attr(all(coverage_nightly, test), feature(coverage_attribute))]
 extern crate knoll;
 
 use coverage_helper::test;
@@ -7,7 +7,7 @@ use knoll::real_displays::*;
 use tempfile;
 
 #[test]
-fn tmp_test() {
+fn test_help() {
     let mut vec_out: Vec<u8> = Vec::new();
     let file_err = tempfile::tempfile().expect("Failed to create temporary file.");
     let args = vec!["knoll", "help"];
@@ -18,3 +18,16 @@ fn tmp_test() {
         file_err.try_clone().expect("Clone failed"),
     );
 }
+
+#[test]
+fn test_list() {
+    let mut vec_out: Vec<u8> = Vec::new();
+    let file_err = tempfile::tempfile().expect("Failed to create temporary file.");
+    let args = vec!["knoll", "list"];
+    let _ = run::<RealDisplayState, std::io::Stdin, &mut Vec<u8>, std::fs::File>(
+        &args.into_iter().map(|s| String::from(s)).collect(),
+        std::io::stdin(),
+        &mut vec_out,
+        file_err.try_clone().expect("Clone failed"),
+    );
+}