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())
