Massive refactor/renaming. No more upper-case filenames.
This commit is contained in:
35
tests/test_Target.py
Normal file
35
tests/test_Target.py
Normal file
@@ -0,0 +1,35 @@
|
||||
#!/usr/bin/python2.7
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
from py.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()
|
||||
Reference in New Issue
Block a user