From 1d1ab506989e3d31d49869900417c3b45ce28a7d Mon Sep 17 00:00:00 2001
From: Dominik Maier <domenukk@gmail.com>
Date: Mon, 26 Oct 2020 05:51:52 +0100
Subject: [PATCH] fixed unittest

---
 src/corpus/mod.rs   | 6 ++++--
 src/inputs/bytes.rs | 2 --
 src/inputs/mod.rs   | 1 -
 src/mutators/mod.rs | 1 -
 4 files changed, 4 insertions(+), 6 deletions(-)

diff --git a/src/corpus/mod.rs b/src/corpus/mod.rs
index ed8d1a5168..a332b0b2fc 100644
--- a/src/corpus/mod.rs
+++ b/src/corpus/mod.rs
@@ -63,7 +63,6 @@ impl Corpus for DefaultQueue<'_> {
     }
 
     fn add(&mut self, entry: Box<dyn Testcase>) {
-        println!("printing {?entry}")
         self.entries.push(entry);
     }
 
@@ -92,8 +91,11 @@ impl Corpus for DefaultQueue<'_> {
 
     /// Gets the next entry
     fn get(&mut self) -> Result<&Box<dyn Testcase>, AflError> {
+        if self.entries.len() == 0 {
+            return Err(AflError::Unknown)
+        }
         self.pos = self.pos + 1;
-        if self.pos > self.entries.len() {
+        if self.pos >= self.entries.len() {
             self.cycles = self.cycles + 1;
             self.pos = 0;
         }
diff --git a/src/inputs/bytes.rs b/src/inputs/bytes.rs
index 3ced8e4015..ac3b80f020 100644
--- a/src/inputs/bytes.rs
+++ b/src/inputs/bytes.rs
@@ -1,5 +1,3 @@
-use std::io::Error;
-
 use crate::AflError;
 use crate::inputs::Input;
 
diff --git a/src/inputs/mod.rs b/src/inputs/mod.rs
index 3583907086..4d1be063e4 100644
--- a/src/inputs/mod.rs
+++ b/src/inputs/mod.rs
@@ -1,6 +1,5 @@
 pub mod bytes;
 
-use std::io::Error;
 use std::fs::File;
 use std::io::Write;
 use std::io::Read;
diff --git a/src/mutators/mod.rs b/src/mutators/mod.rs
index 8650b9904f..9abff03e84 100644
--- a/src/mutators/mod.rs
+++ b/src/mutators/mod.rs
@@ -1,6 +1,5 @@
 use std::io::Error;
 use crate::inputs::Input;
-use crate::utils::Rand;
 
 pub mod scheduled;