54 lines
1.6 KiB
Python
54 lines
1.6 KiB
Python
#!/usr/bin/python2.7
|
|
# -*- coding: utf-8 -*-
|
|
|
|
import sys
|
|
sys.path.insert(0, '..')
|
|
|
|
from wifite.model.handshake import Handshake
|
|
|
|
import unittest
|
|
|
|
class TestHandshake(unittest.TestCase):
|
|
''' Test suite for Target parsing an generation '''
|
|
|
|
def getFile(self, filename):
|
|
''' Helper method to parse targets from filename '''
|
|
import os, inspect
|
|
this_file = os.path.abspath(inspect.getsourcefile(self.getFile))
|
|
this_dir = os.path.dirname(this_file)
|
|
return os.path.join(this_dir, 'files', filename)
|
|
|
|
def testAnalyze(self):
|
|
hs_file = self.getFile('handshake_exists.cap')
|
|
hs = Handshake(hs_file, bssid='A4:2B:8C:16:6B:3A')
|
|
try:
|
|
hs.analyze()
|
|
except Exception, e:
|
|
fail()
|
|
|
|
def testHandshakeTshark(self):
|
|
hs_file = self.getFile('handshake_exists.cap')
|
|
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='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='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='A4:2B:8C:16:6B:3A')
|
|
assert(len(hs.aircrack_handshakes()) > 0)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|
|
|