def test_scanResultSummary_argument_by_invalid_value_should_raise_ValueError(self):
"""
Test scanResultSummary(self, instanceId, by="type")
"""
sfdb = SpiderFootDb(self.default_options, False)
instance_id = "example instance id"
with self.assertRaises(ValueError):
sfdb.scanResultSummary(instance_id, "invalid filter type")
def test_scanResultSummary_argument_instanceId_of_invalid_type_should_raise_TypeError(self):
"""
Test scanResultSummary(self, instanceId, by="type")
"""
sfdb = SpiderFootDb(self.default_options, False)
invalid_types = [None, list(), dict(), int()]
for invalid_type in invalid_types:
with self.subTest(invalid_type=invalid_type):
with self.assertRaises(TypeError):
sfdb.scanResultSummary(invalid_type)
def scansummary(self, id, by):
"""Summary of scan results
Args:
id (str): scan ID
by: TBD
"""
cherrypy.response.headers['Content-Type'] = "application/json; charset=utf-8"
retdata = []
dbh = SpiderFootDb(self.config)
try:
scandata = dbh.scanResultSummary(id, by)
except Exception:
return json.dumps(retdata).encode('utf-8')
try:
statusdata = dbh.scanInstanceGet(id)
except Exception:
return json.dumps(retdata).encode('utf-8')
for row in scandata:
if row[0] == "ROOT":
continue
lastseen = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(row[2]))
retdata.append([row[0], row[1], lastseen, row[3], row[4], statusdata[5]])
return json.dumps(retdata).encode('utf-8')
def scansummary(self, id, by):
"""Summary of scan results.
Args:
id (str): scan ID
by: TBD
Returns:
str: scan summary as JSON
"""
retdata = []
dbh = SpiderFootDb(self.config)
try:
scandata = dbh.scanResultSummary(id, by)
except Exception:
return retdata
try:
statusdata = dbh.scanInstanceGet(id)
except Exception:
return retdata
for row in scandata:
if row[0] == "ROOT":
continue
lastseen = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(row[2]))
retdata.append([row[0], row[1], lastseen, row[3], row[4], statusdata[5]])
return retdata
def test_scanResultSummary_should_return_a_list(self):
"""
Test scanResultSummary(self, instanceId, by="type")
"""
sfdb = SpiderFootDb(self.default_options, False)
instance_id = "example instance id"
scan_results_summary = sfdb.scanResultSummary(instance_id, "type")
self.assertIsInstance(scan_results_summary, list)
