fix tests

Author: fabianbees

benches/benchmark_portscan.rs

@@ -45,6 +45,7 @@ fn criterion_benchmark(c: &mut Criterion) {
         true,
         vec![],
         false,
+        false,
     );
 
     c.bench_function("portscan tcp", |b| {
@@ -61,6 +62,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

@@ -351,6 +351,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!
@@ -375,6 +376,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!
@@ -398,6 +400,7 @@ mod tests {
             true,
             vec![9000],
             false,
+            false,
         );
         block_on(scanner.run());
         assert_eq!(1, 1);
@@ -420,6 +423,7 @@ mod tests {
             true,
             vec![9000],
             false,
+            false,
         );
         block_on(scanner.run());
         assert_eq!(1, 1);
@@ -445,6 +449,7 @@ mod tests {
             true,
             vec![9000],
             false,
+            false,
         );
         block_on(scanner.run());
         assert_eq!(1, 1);
@@ -469,6 +474,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!
@@ -493,6 +499,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!
@@ -516,6 +523,7 @@ mod tests {
             true,
             vec![9000],
             true,
+            false,
         );
         block_on(scanner.run());
         assert_eq!(1, 1);
@@ -538,6 +546,7 @@ mod tests {
             true,
             vec![9000],
             true,
+            false,
         );
         block_on(scanner.run());
         assert_eq!(1, 1);