alexandre/wifite2
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fixing unit tests

Browse Source
This commit is contained in:
derv82
2015-06-03 20:46:56 -07:00
parent 3faa01c9e8
commit 9b7cc7fb8d
2 changed files with 9 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
py/tests/test_Handshake.py
View File

@@ -1,5 +1,8 @@
#!/usr/bin/python #!/usr/bin/python
import sys
sys.path.insert(0, '..')
from Handshake import Handshake from Handshake import Handshake
import unittest import unittest
@@ -16,7 +19,8 @@ class TestHandshake(unittest.TestCase):
def testAnalyze(self): def testAnalyze(self):
hs_file = self.getFile('handshake_exists.cap') 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: try:
hs.analyze() hs.analyze()
except Exception, e: except Exception, e:
@@ -24,23 +28,23 @@ class TestHandshake(unittest.TestCase):
def testHandshakeTshark(self): def testHandshakeTshark(self):
hs_file = self.getFile('handshake_exists.cap') 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) assert(len(hs.tshark_handshakes()) > 0)
def testHandshakePyrit(self): def testHandshakePyrit(self):
hs_file = self.getFile('handshake_exists.cap') 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) assert(len(hs.pyrit_handshakes()) > 0)
def testHandshakeCowpatty(self): def testHandshakeCowpatty(self):
hs_file = self.getFile('handshake_exists.cap') 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() hs.divine_bssid_and_essid()
assert(len(hs.cowpatty_handshakes()) > 0) assert(len(hs.cowpatty_handshakes()) > 0)
def testHandshakeAircrack(self): def testHandshakeAircrack(self):
hs_file = self.getFile('handshake_exists.cap') 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) assert(len(hs.aircrack_handshakes()) > 0)

19
py/tests/test_Target.py
View File

@@ -30,24 +30,5 @@ class TestTarget(unittest.TestCase):
if t.bssid == '00:1D:D5:9B:11:00': if t.bssid == '00:1D:D5:9B:11:00':
assert(len(t.clients) > 0) 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__': if __name__ == '__main__':
unittest.main() unittest.main()