def test_dataParentChildToTree_invalid_data_type_should_return_TypeError( self): """ Test dataParentChildToTree(data) """ invalid_types = [None, "", list(), int()] for invalid_type in invalid_types: with self.subTest(invalid_type=invalid_type): with self.assertRaises(TypeError): SpiderFootHelpers.dataParentChildToTree(invalid_type)
def scanelementtypediscovery(self, id, eventType): """Scan element type discovery. Args: id (str): scan ID eventType (str): filter by event type Returns: str: JSON """ dbh = SpiderFootDb(self.config) pc = dict() datamap = dict() retdata = dict() # Get the events we will be tracing back from try: leafSet = dbh.scanResultEvent(id, eventType) [datamap, pc] = dbh.scanElementSourcesAll(id, leafSet) except Exception: return retdata # Delete the ROOT key as it adds no value from a viz perspective del pc['ROOT'] retdata['tree'] = SpiderFootHelpers.dataParentChildToTree(pc) retdata['data'] = datamap return retdata
def test_dataParentChildToTree_should_return_dict(self): """ Test dataParentChildToTree(data) """ tree = SpiderFootHelpers.dataParentChildToTree( {"test": { "123": "456" }}) self.assertIsInstance(tree, dict)
def test_dataParentChildToTree_should_return_dict(self): """ Test dataParentChildToTree(data) """ invalid_types = [None, "", list(), int()] for invalid_type in invalid_types: with self.subTest(invalid_type=invalid_type): tree = SpiderFootHelpers.dataParentChildToTree(invalid_type) self.assertIsInstance(tree, dict) tree = SpiderFootHelpers.dataParentChildToTree(dict()) self.assertIsInstance(tree, dict) tree = SpiderFootHelpers.dataParentChildToTree( {"test": { "123": "456" }}) self.assertIsInstance(tree, dict)
def test_dataParentChildToTree_empty_data_should_return_ValueError(self): """ Test dataParentChildToTree(data) """ with self.assertRaises(ValueError): SpiderFootHelpers.dataParentChildToTree(dict())