fix tests

Author: fabianbees

benches/benchmark_portscan.rs

@@ -64,6 +64,7 @@ fn criterion_benchmark(c: &mut Criterion) {
         true,
         vec![],
         false,
+        false,
     );
 
     c.bench_function("portscan tcp", |b| {
@@ -80,6 +81,7 @@ fn criterion_benchmark(c: &mut Criterion) {
         true,
         vec![],
         true,
+        false,
     );
 
     let mut udp_group = c.benchmark_group("portscan udp");

src/lib.rs

@@ -32,6 +32,7 @@
 //!         true, // accessible, should the output be A11Y compliant?
 //!         vec![9000], // What ports should RustScan exclude?
 //!         false, // is this a UDP scan?
+//!         false, // Output closed ports?
 //!     );
 //!
 //!     let scan_result = block_on(scanner.run());

src/main.rs

@@ -17,6 +17,7 @@ use std::string::ToString;
 use std::time::Duration;
 
 use rustscan::address::parse_addresses;
+use rustscan::scanner::PortStatus;
 
 extern crate colorful;
 extern crate dirs;
@@ -35,8 +36,6 @@ extern crate log;
 /// Faster Nmap scanning with Rust
 /// If you're looking for the actual scanning, check out the module Scanner
 fn main() {
-    use rustscan::scanner::PortStatus;
-
     #[cfg(not(unix))]
     let _ = ansi_term::enable_ansi_support();
 

src/scanner/mod.rs

@@ -353,6 +353,7 @@ mod tests {
             true,
             vec![9000],
             false,
+            false,
         );
         block_on(scanner.run());
         // if the scan fails, it wouldn't be able to assert_eq! as it panicked!
@@ -377,6 +378,7 @@ mod tests {
             true,
             vec![9000],
             false,
+            false,
         );
         block_on(scanner.run());
         // if the scan fails, it wouldn't be able to assert_eq! as it panicked!
@@ -400,6 +402,7 @@ mod tests {
             true,
             vec![9000],
             false,
+            false,
         );
         block_on(scanner.run());
         assert_eq!(1, 1);
@@ -422,6 +425,7 @@ mod tests {
             true,
             vec![9000],
             false,
+            false,
         );
         block_on(scanner.run());
         assert_eq!(1, 1);
@@ -447,6 +451,7 @@ mod tests {
             true,
             vec![9000],
             false,
+            false,
         );
         block_on(scanner.run());
         assert_eq!(1, 1);
@@ -471,6 +476,7 @@ mod tests {
             true,
             vec![9000],
             true,
+            false,
         );
         block_on(scanner.run());
         // if the scan fails, it wouldn't be able to assert_eq! as it panicked!
@@ -495,6 +501,7 @@ mod tests {
             true,
             vec![9000],
             true,
+            false,
         );
         block_on(scanner.run());
         // if the scan fails, it wouldn't be able to assert_eq! as it panicked!
@@ -518,6 +525,7 @@ mod tests {
             true,
             vec![9000],
             true,
+            false,
         );
         block_on(scanner.run());
         assert_eq!(1, 1);
@@ -540,6 +548,7 @@ mod tests {
             true,
             vec![9000],
             true,
+            false,
         );
         block_on(scanner.run());
         assert_eq!(1, 1);