add test and remove once_cell dependency

Author: Dauaaa

Cargo.lock

@@ -38,7 +38,6 @@ itertools = "0.13.0"
 hickory-resolver = { version = "0.24.0", features = ["dns-over-rustls"] }
 anyhow = "1.0.40"
 text_placeholder = { version = "0.5", features = ["struct_context"] }
-once_cell = "1.20.2"
 
 [dev-dependencies]
 parameterized = "2.0.0"

Cargo.toml

@@ -62,8 +62,7 @@ fn generate_code(port_payload_map: BTreeMap<Vec<u16>, Vec<u8>>) {
     let dest_path = PathBuf::from("src/generated.rs");
 
     let mut generated_code = String::new();
-    generated_code.push_str("use std::collections::BTreeMap;\n");
-    generated_code.push_str("use once_cell::sync::Lazy;\n\n");
+    generated_code.push_str("use std::{collections::BTreeMap, sync::LazyLock};\n\n");
 
     generated_code.push_str("fn generated_data() -> BTreeMap<Vec<u16>, Vec<u8>> {\n");
     generated_code.push_str("    let mut map = BTreeMap::new();\n");
@@ -92,7 +91,7 @@ fn generate_code(port_payload_map: BTreeMap<Vec<u16>, Vec<u8>>) {
     generated_code.push_str("}\n\n");
 
     generated_code.push_str(
-        "static PARSED_DATA: Lazy<BTreeMap<Vec<u16>, Vec<u8>>> = Lazy::new(generated_data);\n",
+        "static PARSED_DATA: LazyLock<BTreeMap<Vec<u16>, Vec<u8>>> = LazyLock::new(generated_data);\n",
     );
     generated_code.push_str("pub fn get_parsed_data() -> &'static BTreeMap<Vec<u16>, Vec<u8>> {\n");
     generated_code.push_str("    &PARSED_DATA\n");

build.rs

@@ -1,5 +1,4 @@
-use once_cell::sync::Lazy;
-use std::collections::BTreeMap;
+use std::{collections::BTreeMap, sync::LazyLock};
 
 fn generated_data() -> BTreeMap<Vec<u16>, Vec<u8>> {
     let mut map = BTreeMap::new();
@@ -2990,7 +2989,7 @@ fn generated_data() -> BTreeMap<Vec<u16>, Vec<u8>> {
     map
 }
 
-static PARSED_DATA: Lazy<BTreeMap<Vec<u16>, Vec<u8>>> = Lazy::new(generated_data);
+static PARSED_DATA: LazyLock<BTreeMap<Vec<u16>, Vec<u8>>> = LazyLock::new(generated_data);
 pub fn get_parsed_data() -> &'static BTreeMap<Vec<u16>, Vec<u8>> {
     &PARSED_DATA
 }

src/generated.rs

@@ -41,6 +41,8 @@
 //! ```
 #![allow(clippy::needless_doctest_main)]
 
+use std::sync::OnceLock;
+
 pub mod tui;
 
 pub mod input;
@@ -60,7 +62,7 @@ pub mod generated;
 /// Static variable defining the current state of execution. The cli binary should
 /// set it to true by calling set_cli_mode.
 #[doc(hidden)]
-pub static IS_CLI_MODE: once_cell::sync::OnceCell<bool> = once_cell::sync::OnceCell::new();
+pub static IS_CLI_MODE: OnceLock<bool> = OnceLock::new();
 
 /// Set IS_CLI_MODE to true.
 #[doc(hidden)]

src/lib.rs

@@ -0,0 +1,43 @@
+/*
+ * Test rustscan logging utilities, ensuring library doesn't log to stdout.
+ */
+#![feature(internal_output_capture)]
+
+use std::{net::IpAddr, str::FromStr, time::Duration};
+
+use futures::executor::block_on;
+use rustscan::{input::ScanOrder, port_strategy::PortStrategy, scanner::Scanner};
+use std::io;
+use std::sync::{Arc, Mutex};
+
+#[test]
+fn no_logging_scanner() {
+    // "open" tcp connection on random port
+    let listener = std::net::TcpListener::bind("127.0.0.1:0").unwrap();
+    // get the port from above connection
+    let port = listener.local_addr().unwrap().port();
+
+    let stdout_buf = Arc::new(Mutex::new(vec![]));
+
+    // capture stdout
+    io::set_output_capture(Some(stdout_buf.clone()));
+    block_on(
+        Scanner::new(
+            &[IpAddr::from_str("127.0.0.1").unwrap()],
+            100,
+            Duration::from_secs(5),
+            3,
+            false,
+            PortStrategy::pick(&None, Some(vec![port]), ScanOrder::Random),
+            true,
+            vec![],
+            false,
+        )
+        .run(),
+    );
+    // stop capturing stdout
+    io::set_output_capture(None);
+
+    let captured_stdout = stdout_buf.lock().unwrap();
+    assert!(captured_stdout.is_empty());
+}