def test_enhances_caller(self):
assembly = """
00000098 <pbl_table_addr>:
8e4: f000 f824 bl 930 <app_log>
00000930 <app_log>:
$t():
"""
c = Collector(None)
self.assertEqual(2, c.parse_assembly_text(assembly))
self.assertTrue(0x00000098 in c.symbols)
self.assertTrue(0x00000930 in c.symbols)
pbl_table_addr = c.symbols[0x00000098]
app_log = c.symbols[0x00000930]
self.assertFalse("callers" in pbl_table_addr)
self.assertFalse("callees" in pbl_table_addr)
self.assertFalse("callers" in app_log)
self.assertFalse("callees" in app_log)
c.enhance_call_tree()
self.assertEqual(pbl_table_addr["callers"], [])
self.assertEqual(pbl_table_addr["callees"], [app_log])
self.assertEqual(app_log["callers"], [pbl_table_addr])
self.assertEqual(app_log["callees"], [])
def test_enhances_caller(self):
assembly = """
00000098 <pbl_table_addr>:
8e4: f000 f824 bl 930 <app_log>
00000930 <app_log>:
$t():
"""
c = Collector()
self.assertEqual(2, c.parse_assembly_text(assembly))
self.assertTrue(c.symbols.has_key(0x00000098))
self.assertTrue(c.symbols.has_key(0x00000930))
pbl_table_addr = c.symbols[0x00000098]
app_log = c.symbols[0x00000930]
self.assertFalse(pbl_table_addr.has_key("callers"))
self.assertFalse(pbl_table_addr.has_key("callees"))
self.assertFalse(app_log.has_key("callers"))
self.assertFalse(app_log.has_key("callees"))
c.enhance_call_tree()
self.assertEqual(pbl_table_addr["callers"], [])
self.assertEqual(pbl_table_addr["callees"], [app_log])
self.assertEqual(app_log["callers"], [pbl_table_addr])
self.assertEqual(app_log["callees"], [])
def test_enhances_caller(self):
assembly = """
00000098 <pbl_table_addr>:
8e4: f000 f824 bl 930 <app_log>
00000930 <app_log>:
$t():
"""
c = Collector(None)
self.assertEqual(2, c.parse_assembly_text(assembly))
self.assertTrue(c.symbols.has_key(0x00000098))
self.assertTrue(c.symbols.has_key(0x00000930))
pbl_table_addr = c.symbols[0x00000098]
app_log = c.symbols[0x00000930]
self.assertFalse(pbl_table_addr.has_key("callers"))
self.assertFalse(pbl_table_addr.has_key("callees"))
self.assertFalse(app_log.has_key("callers"))
self.assertFalse(app_log.has_key("callees"))
c.enhance_call_tree()
self.assertEqual(pbl_table_addr["callers"], [])
self.assertEqual(pbl_table_addr["callees"], [app_log])
self.assertEqual(app_log["callers"], [pbl_table_addr])
self.assertEqual(app_log["callees"], [])
