def test_scan_result_summary_invalid_type_should_raise(self):
"""
Test scanResultSummary(self, instanceId, by="type")
"""
sfdb = SpiderFootDb(self.default_options, False)
with self.assertRaises(ValueError) as cm:
scan_results_summary = sfdb.scanResultSummary(None, None)
with self.assertRaises(ValueError) as cm:
scan_results_summary = sfdb.scanResultSummary(
"", "invalid filter type")
def test_scanResultSummary_argument_by_of_invalid_type_should_raise_TypeError(self):
"""
Test scanResultSummary(self, instanceId, by="type")
"""
sfdb = SpiderFootDb(self.default_options, False)
instance_id = "example instance id"
invalid_types = [None, list(), dict(), int()]
for invalid_type in invalid_types:
with self.subTest(invalid_type=invalid_type):
with self.assertRaises(TypeError) as cm:
scan_results_summary = sfdb.scanResultSummary(instance_id, invalid_type)
with self.assertRaises(ValueError) as cm:
scan_results_summary = sfdb.scanResultSummary(instance_id, "invalid filter type")
def scansummary(self, id, by):
"""Summary of scan results
Args:
id (str): scan ID
by: TBD
"""
retdata = []
dbh = SpiderFootDb(self.config)
try:
scandata = dbh.scanResultSummary(id, by)
except Exception:
return json.dumps(retdata)
try:
statusdata = dbh.scanInstanceGet(id)
except Exception:
return json.dumps(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 json.dumps(retdata)
def scansummary(self, id):
dbh = SpiderFootDb(self.config)
data = dbh.scanResultSummary(id)
retdata = []
for row in data:
lastseen = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(row[2]))
retdata.append([row[0], row[1], lastseen, row[3], row[4]])
return json.dumps(retdata)
def test_scan_result_summary_should_return_a_list(self):
"""
Test scanResultSummary(self, instanceId, by="type")
"""
sfdb = SpiderFootDb(self.default_options, False)
scan_results_summary = sfdb.scanResultSummary("", "type")
self.assertIsInstance(scan_results_summary, list)
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) as cm:
scan_results_summary = sfdb.scanResultSummary(instance_id, "invalid filter type")
def scansummary(self, id, by):
retdata = []
dbh = SpiderFootDb(self.config)
try:
data = dbh.scanResultSummary(id, by)
except:
return json.dumps(retdata)
for row in data:
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]])
return json.dumps(retdata)
