freewvs.git

1) import unittest
2) import glob
3) import json
4) import sys
5) import difflib

6) import re

7) 
8) 

9) def versioncompare(safe_version, find_version):
10)     safe_version_tup = [int(x) for x in safe_version.split(".")]
11)     find_version_tup = [int(x) for x in find_version.split(".")]
12)     return find_version_tup < safe_version_tup
13) 
14) 

15) class TestJsonLint(unittest.TestCase):

16) 
17)     @unittest.skipIf(sys.version_info < (3, 6, 0),
18)                      "json.dumps force-sorts on python 3.5")

19)     def test_json_lint(self):
20)         valid = True
21)         for f in glob.glob("freewvsdb/*.json"):

22)             fp = open(f, encoding="ascii")

23)             orig = fp.read()
24)             fp.close()
25)             tmp = json.loads(orig)
26)             new = json.dumps(tmp, indent=2)

27)             if orig != new:

28)                 print(f"json {f} not valid")

29)                 sys.stdout.writelines(difflib.unified_diff(orig, new))
30)                 valid = False

31)         self.assertTrue(valid)

32) 

33)     def test_json_values(self):
34)         jconfig = []
35)         for cfile in glob.glob('freewvsdb/*.json'):

36)             with open(cfile, encoding="ascii") as json_file:

37)                 jconfig += json.load(json_file)
38) 

39)         mkeys = {'name', 'url', 'safe', 'vuln', 'detection'}

40)         for item in jconfig:
41) 
42)             # check for all mandatory keys
43)             self.assertEqual(mkeys.intersection(item.keys()), mkeys,

44)                              msg=f"Missing key in {item['name']}")

45) 
46)             # check we have at least one detection
47)             self.assertTrue(len(item['detection']) >= 1,

48)                             msg=f"No detection in {item['name']}")

49) 
50)             # vuln needs to be CVE or HTTPS URL
51)             self.assertTrue(re.match("^CVE-[0-9]*-[0-9]*$", item['vuln'])
52)                             or item['vuln'].startswith("https://"),

53)                             msg=f"{item['name']}: Invalid vuln {item['vuln']}")

54) 
55)             # make sure old_safe is properly sorted
56)             if 'old_safe' in item:
57)                 old_safe = item['old_safe'].split(',')
58)                 for i in range(1, len(old_safe)):
59)                     self.assertTrue(versioncompare(old_safe[i - 1],
60)                                                    old_safe[i]),

61)                                     msg=f"{item['name']}: Invalid old_safe"
62)                                         " ordering {item['old_safe']}")

63) 

64)             # make sure latest is not outdated
65)             if 'latest' in item and item['safe'] != "":
66)                 self.assertTrue(not versioncompare(item['safe'],
67)                                                    item['latest']),

68)                                 msg=f"{item['name']}: Safe version "
69)                                     "{item['safe']} newer than latest"
70)                                     " {item['latest']}")

71) 

72)             # subdir needs to be integer

73)             for det in item['detection']:

74)                 self.assertTrue(isinstance(det['subdir'], int),

75)                                 msg=f"{item['name']}: subdir not int")

76)