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)