add cobalt importer

dr3dd589 · dr3dd589 · commit 9128bac94a39 · 2019-06-05T15:52:08.000+05:30
diff --git a/dojo/fixtures/test_type.json b/dojo/fixtures/test_type.json
@@ -340,5 +340,13 @@
 		},
 		"model": "dojo.test_type",
 		"pk":56
+	},
+	{
+		"fields": {
+      "name": "Cobalt.io Scan"
+    },
+    "model": "dojo.test_type",
+    "pk": 57
+
 	}
 ]
diff --git a/dojo/forms.py b/dojo/forms.py
@@ -297,7 +297,8 @@ class ImportScanForm(forms.Form):
                          ("Openscap Vulnerability Scan", "Openscap Vulnerability Scan"),
                          ("Wapiti Scan", "Wapiti Scan"),
                          ("Immuniweb Scan", "Immuniweb Scan"),
-                         ("Sonatype Application Scan", "Sonatype Application Scan"))
+                         ("Sonatype Application Scan", "Sonatype Application Scan"),
+                         ("Cobalt.io Scan", "Cobalt.io Scan"))
 
     SORTED_SCAN_TYPE_CHOICES = sorted(SCAN_TYPE_CHOICES, key=lambda x: x[1])
     scan_date = forms.DateTimeField(
diff --git a/dojo/templates/dojo/import_scan_results.html b/dojo/templates/dojo/import_scan_results.html
@@ -42,6 +42,7 @@ <h3> Add Tests</h3>
 	<li><b>Clair Klar Scan</b> -  Import JSON reports of Docker image vulnerabilities from clair klar client.</li>
 	<li><b>Contrast Scanner</b> - CSV Report</b></li>
 	<li><b>Checkmarx Detailed XML Report</b></li>
+	<li><b>Cobalt.io Scan</b> - CSV Report</b></li>
 	<li><b>Crashtest Security JUnit XML Report</b></li>
 	<li><b>DawnScanner</b> - Dawnscanner (-j) output file can be imported in JSON format.</li>
 	<li><b>Dependency Check</b> - OWASP Dependency Check output can be imported in Xml format.</li>
diff --git a/dojo/tools/factory.py b/dojo/tools/factory.py
@@ -50,6 +50,7 @@
 from dojo.tools.openscap.parser import OpenscapXMLParser
 from dojo.tools.immuniweb.parser import ImmuniwebXMLParser
 from dojo.tools.wapiti.parser import WapitiXMLParser
+from dojo.tools.cobalt.parser import CobaltCSVParser
 
 __author__ = 'Jay Paz'
 
@@ -165,6 +166,8 @@ def import_parser_factory(file, test, active, verified, scan_type=None):
         parser = ImmuniwebXMLParser(file, test)
     elif scan_type == 'Wapiti Scan':
         parser = WapitiXMLParser(file, test)
+    elif scan_type == 'Cobalt.io Scan':
+        parser = CobaltCSVParser(file, test)
     else:
         raise ValueError('Unknown Test Type')
 

Original file line number	Diff line number	Diff line change
`@@ -340,5 +340,13 @@`
`340`	`340`	`},`
`341`	`341`	`"model": "dojo.test_type",`
`342`	`342`	`"pk":56`
	`343`	`+ },`
	`344`	`+ {`
	`345`	`+ "fields": {`
	`346`	`+ "name": "Cobalt.io Scan"`
	`347`	`+ },`
	`348`	`+ "model": "dojo.test_type",`
	`349`	`+ "pk": 57`
	`350`	`+`
`343`	`351`	`}`
`344`	`352`	`]`