From 9b7cc7fb8d7873e0ca33eee00db54aaa2737000f Mon Sep 17 00:00:00 2001
From: derv82 <derv82@gmail.com>
Date: Wed, 3 Jun 2015 20:46:56 -0700
Subject: [PATCH] Fixing unit tests

---
 py/tests/test_Handshake.py | 14 +++++++++-----
 py/tests/test_Target.py    | 19 -------------------
 2 files changed, 9 insertions(+), 24 deletions(-)

diff --git a/py/tests/test_Handshake.py b/py/tests/test_Handshake.py
index 0b75b7d..28cef79 100644
--- a/py/tests/test_Handshake.py
+++ b/py/tests/test_Handshake.py
@@ -1,5 +1,8 @@
 #!/usr/bin/python
 
+import sys
+sys.path.insert(0, '..')
+
 from Handshake import Handshake
 
 import unittest
@@ -16,7 +19,8 @@ class TestHandshake(unittest.TestCase):
 
     def testAnalyze(self):
         hs_file = self.getFile('handshake_exists.cap')
-        hs = Handshake(hs_file, bssid='30:85:a9:39:d2:18')
+        print hs_file
+        hs = Handshake(hs_file, bssid='A4:2B:8C:16:6B:3A')
         try:
             hs.analyze()
         except Exception, e:
@@ -24,23 +28,23 @@ class TestHandshake(unittest.TestCase):
 
     def testHandshakeTshark(self):
         hs_file = self.getFile('handshake_exists.cap')
-        hs = Handshake(hs_file, bssid='30:85:a9:39:d2:18')
+        hs = Handshake(hs_file, bssid='A4:2B:8C:16:6B:3A')
         assert(len(hs.tshark_handshakes()) > 0)
 
     def testHandshakePyrit(self):
         hs_file = self.getFile('handshake_exists.cap')
-        hs = Handshake(hs_file, bssid='30:85:a9:39:d2:18')
+        hs = Handshake(hs_file, bssid='A4:2B:8C:16:6B:3A')
         assert(len(hs.pyrit_handshakes()) > 0)
 
     def testHandshakeCowpatty(self):
         hs_file = self.getFile('handshake_exists.cap')
-        hs = Handshake(hs_file, bssid='30:85:a9:39:d2:18')
+        hs = Handshake(hs_file, bssid='A4:2B:8C:16:6B:3A')
         hs.divine_bssid_and_essid()
         assert(len(hs.cowpatty_handshakes()) > 0)
 
     def testHandshakeAircrack(self):
         hs_file = self.getFile('handshake_exists.cap')
-        hs = Handshake(hs_file, bssid='30:85:a9:39:d2:18')
+        hs = Handshake(hs_file, bssid='A4:2B:8C:16:6B:3A')
         assert(len(hs.aircrack_handshakes()) > 0)
 
 
diff --git a/py/tests/test_Target.py b/py/tests/test_Target.py
index 6936ceb..18281cc 100644
--- a/py/tests/test_Target.py
+++ b/py/tests/test_Target.py
@@ -30,24 +30,5 @@ class TestTarget(unittest.TestCase):
             if t.bssid == '00:1D:D5:9B:11:00':
                 assert(len(t.clients) > 0)
 
-    def testTargetFilter(self):
-        ''' Asserts target filtering works '''
-        targets = self.getTargets(TestTarget.airodump_csv)
-
-        opnTargets = Airodump.filter_targets(targets, opn=True, wpa=False, wep=False)
-        for t in opnTargets:
-            if 'OPN' not in t.encryption:
-                fail()
-
-        wpaTargets = Airodump.filter_targets(targets, opn=False, wpa=True, wep=False)
-        for t in wpaTargets:
-            if 'WPA' not in t.encryption:
-                fail()
-
-        wepTargets = Airodump.filter_targets(targets, opn=False, wpa=False, wep=True)
-        for t in wepTargets:
-            if 'WEP' not in t.encryption:
-                fail()
-
 if __name__ == '__main__':
     unittest.main()