python-iwlist/test.py at master · src-dev/python-iwlist · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
importjson
importos
importiwlist
importunittest

deffileContent(filename):
f=open(filename)
content=f.read()
f.close()
returncontent

classTestParse(unittest.TestCase):

defsetUp(self):
dirs=os.listdir("test")
self.cases= []
fordindirs:
scanFile=os.path.join("test", d, "scan.txt")
vectorsFile=os.path.join("test", d, "vectors.json")
case= {
"name": d,
"parsed": iwlist.parse(fileContent(scanFile)),
"expected": json.loads(fileContent(vectorsFile)),
 }
self.cases.append(case)

deftearDown(self):
self.cases= []

deftest_parse_length(self):
forcaseinself.cases:
self.assertEqual(len(case["expected"]), len(case["parsed"]))

deftest_cells_have_all_expected_keys(self):
forcaseinself.cases:
foriinrange(len(case["expected"])):
e=case["expected"][i]
p=case["parsed"][i]
forkeyine:
msg="\nkey is in expected but missing in parsed:\nkey: %s\ntestdir: %s\ncellnumber: %s"% (key, case["name"], e["cellnumber"])
self.assertTrue(keyinp, msg)

deftest_cells_dont_have_extra_keys(self):
forcaseinself.cases:
foriinrange(len(case["expected"])):
e=case["expected"][i]
p=case["parsed"][i]
forkeyinp:
msg="\nkey was parsed but missing in expected:\nkey: %s\ntestdir: %s\ncellnumber: %s"% (key, case["name"], e["cellnumber"])
self.assertTrue(keyine, msg)

deftest_cells_have_expected_values(self):
forcaseinself.cases:
foriinrange(len(case["expected"])):
e=case["expected"][i]
forkine:
ev=e[k]
pv=case["parsed"][i][k]
msg="\nwrong value for field:\ntestdir: %s\nfield: %s\ncellnumber: %s\nexpected: %s\nactual: %s"% (case["name"], k, e["cellnumber"], ev, pv)
self.assertTrue(ev==pv, msg)

if__name__=='__main__':
unittest.main()