36 lines
1.1 KiB
Python
36 lines
1.1 KiB
Python
#!/usr/bin/python2.7
|
|
# -*- coding: utf-8 -*-
|
|
|
|
from wifite.tools.airodump import Airodump
|
|
|
|
import unittest
|
|
|
|
class TestTarget(unittest.TestCase):
|
|
''' Test suite for Target parsing an generation '''
|
|
|
|
airodump_csv = 'airodump.csv'
|
|
|
|
def getTargets(self, filename):
|
|
''' Helper method to parse targets from filename '''
|
|
import os, inspect
|
|
this_file = os.path.abspath(inspect.getsourcefile(TestTarget.getTargets))
|
|
this_dir = os.path.dirname(this_file)
|
|
csv_file = os.path.join(this_dir, 'files', filename)
|
|
# Load targets from CSV file
|
|
return Airodump.get_targets_from_csv(csv_file)
|
|
|
|
def testTargetParsing(self):
|
|
''' Asserts target parsing finds targets '''
|
|
targets = self.getTargets(TestTarget.airodump_csv)
|
|
assert(len(targets) > 0)
|
|
|
|
def testTargetClients(self):
|
|
''' Asserts target parsing captures clients properly '''
|
|
targets = self.getTargets(TestTarget.airodump_csv)
|
|
for t in targets:
|
|
if t.bssid == '00:1D:D5:9B:11:00':
|
|
assert(len(t.clients) > 0)
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|