From 2a5258ceefcb5da939edd30e89563ef83c4aa698 Mon Sep 17 00:00:00 2001
From: derv82 <derv82@gmail.com>
Date: Sun, 14 May 2017 17:42:52 -0400
Subject: [PATCH] Fixing unit tests

---
 py/Configuration.py        | 1 +
 py/tests/test_Handshake.py | 3 +--
 py/tests/test_Target.py    | 2 +-
 runtests.sh                | 2 +-
 4 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/py/Configuration.py b/py/Configuration.py
index 93ff2c2..a26fa21 100644
--- a/py/Configuration.py
+++ b/py/Configuration.py
@@ -7,6 +7,7 @@ import os
 
 class Configuration(object):
     ''' Stores configuration variables and functions for Wifite. '''
+    verbose = 0
 
     initialized = False # Flag indicating config has been initialized
     temp_dir = None     # Temporary directory
diff --git a/py/tests/test_Handshake.py b/py/tests/test_Handshake.py
index 21fefc7..b0ac1c9 100644
--- a/py/tests/test_Handshake.py
+++ b/py/tests/test_Handshake.py
@@ -4,7 +4,7 @@
 import sys
 sys.path.insert(0, '..')
 
-from Handshake import Handshake
+from py.Handshake import Handshake
 
 import unittest
 
@@ -20,7 +20,6 @@ class TestHandshake(unittest.TestCase):
 
     def testAnalyze(self):
         hs_file = self.getFile('handshake_exists.cap')
-        print hs_file
         hs = Handshake(hs_file, bssid='A4:2B:8C:16:6B:3A')
         try:
             hs.analyze()
diff --git a/py/tests/test_Target.py b/py/tests/test_Target.py
index c092d1c..d16f1d9 100644
--- a/py/tests/test_Target.py
+++ b/py/tests/test_Target.py
@@ -1,7 +1,7 @@
 #!/usr/bin/python2.7
 # -*- coding: utf-8 -*-
 
-from Airodump import Airodump
+from py.Airodump import Airodump
 
 import unittest
 
diff --git a/runtests.sh b/runtests.sh
index 3e6f1e3..da9b3ed 100755
--- a/runtests.sh
+++ b/runtests.sh
@@ -1,2 +1,2 @@
 #!/bin/sh
-python2.7 -m unittest discover py
+python2.7 -m unittest discover py/tests -v