Updated tests #16

Signed-off-by: Kartik Sibal

Author: kartiksibal

tests/test_scrapers.py

@@ -66,31 +66,36 @@ def test_debian_extract_data():
 
     test_data = json.loads(test_input)
 
-    expected = {
-        'git-repair': {
-            'TEMP-0807341-84E914': {
-                'status': 'open',
-                'urgency': 'unimportant',
-            }
-        },
-        'mimetex': {
-            'CVE-2009-1382': {
-                'fixed_version': '1.50-1.1',
-                'status': 'resolved',
-                'urgency': 'medium',
-            },
-            'CVE-2009-2459': {
-                'fixed_version': '1.50-1.1',
-                'status': 'not-resolved',
-                'urgency': 'medium'
-            },
-        },
-        'sysvinit': {
-            'TEMP-0517018-A83CE6': {
-                'status': 'open',
-                'urgency': 'unimportant',
-            },
-        },
-    }
+    expected = [{
+                    'fixed_Version': '1.50-1.1',
+                    'package_name': 'mimetex',
+                    'status': 'resolved',
+                    'urgency': 'medium',
+                    'vulnerability_id': 'CVE-2009-1382'
+                },
+
+                {
+                    'fixed_Version': '1.50-1.1',
+                    'package_name': 'mimetex',
+                    'status': 'not-resolved',
+                    'urgency': 'medium',
+                    'vulnerability_id': 'CVE-2009-2459'
+                },
+
+                {
+                    'fixed_Version': None,
+                    'package_name': 'git-repair',
+                    'status': 'open',
+                    'urgency': 'unimportant',
+                    'vulnerability_id': 'TEMP-0807341-84E914'
+                },
+
+                {
+                    'fixed_Version': None,
+                    'package_name': 'sysvinit',
+                    'status': 'open',
+                    'urgency': 'unimportant',
+                    'vulnerability_id': 'TEMP-0517018-A83CE6'
+                }]
 
     assert expected == debian.extract_data(test_data)