blob: 7191919aa51f83a144d7db765d9fea34d29deb66 [file] [log] [blame]
#!/usr/bin/env python3
from __future__ import print_function
import unittest
from vndk_definition_tool import ELFLinkData, PT_SYSTEM, PT_VENDOR
class ELFLinkDataTest(unittest.TestCase):
def setUp(self):
self.x = ELFLinkData(PT_SYSTEM, '/system/lib/libx.so', None, 0)
self.y = ELFLinkData(PT_SYSTEM, '/system/lib/liby.so', None, 0)
self.z = ELFLinkData(PT_SYSTEM, '/system/lib/libz.so', None, 0)
self.w = ELFLinkData(PT_SYSTEM, '/system/lib/libw.so', None, 0)
self.v = ELFLinkData(PT_VENDOR, '/vendor/lib/libv.so', None, 0)
self.x.add_needed_dep(self.y)
self.x.add_dlopen_dep(self.z)
self.z.add_needed_dep(self.w)
self.z.add_dlopen_dep(self.w)
def test_add_dep_and_accessors(self):
self.assertIn(self.y, self.x.deps_needed_all)
self.assertIn(self.x, self.y.users_needed_all)
self.assertNotIn(self.y, self.x.deps_dlopen_all)
self.assertNotIn(self.x, self.y.users_dlopen_all)
self.assertIn(self.z, self.x.deps_dlopen_all)
self.assertIn(self.x, self.z.users_dlopen_all)
self.assertNotIn(self.z, self.x.deps_needed_all)
self.assertNotIn(self.x, self.z.users_needed_all)
def test_remove_dep(self):
self.assertIn(self.y, self.x.deps_needed_all)
self.assertIn(self.x, self.y.users_needed_all)
with self.assertRaises(KeyError):
self.x.hide_dlopen_dep(self.y)
self.assertIn(self.y, self.x.deps_needed_all)
self.assertIn(self.x, self.y.users_needed_all)
self.x.hide_needed_dep(self.y)
self.assertIn(self.y, self.x.deps_needed_hidden)
self.assertIn(self.x, self.y.users_needed_hidden)
self.assertNotIn(self.y, self.x.deps_needed)
self.assertNotIn(self.x, self.y.users_needed)
def test_num_deps(self):
self.assertEqual(2, self.x.num_deps)
self.assertEqual(0, self.y.num_deps)
self.assertEqual(0, self.w.num_deps)
self.assertEqual(0, self.v.num_deps)
# NEEDED and DLOPEN are counted twice.
self.assertEqual(2, self.z.num_deps)
def test_num_users(self):
self.assertEqual(0, self.x.num_users)
self.assertEqual(1, self.y.num_users)
self.assertEqual(1, self.z.num_users)
self.assertEqual(0, self.v.num_users)
# NEEDED and DLOPEN are counted twice.
self.assertEqual(2, self.w.num_users)
def test_has_dep(self):
self.assertTrue(self.x.has_dep(self.y))
self.assertTrue(self.x.has_dep(self.z))
self.assertFalse(self.x.has_dep(self.x))
self.assertFalse(self.x.has_dep(self.w))
def test_has_user(self):
self.assertTrue(self.y.has_user(self.x))
self.assertTrue(self.z.has_user(self.x))
self.assertFalse(self.x.has_user(self.x))
self.assertFalse(self.w.has_user(self.x))
def test_is_system_lib(self):
self.assertTrue(self.x.is_system_lib())
self.assertFalse(self.v.is_system_lib())
def test_get_dep_linked_symbols(self):
self.x.linked_symbols['c'] = self.y
self.x.linked_symbols['b'] = self.y
self.x.linked_symbols['a'] = self.y
self.x.linked_symbols['w'] = self.z
self.x.linked_symbols['z'] = self.z
self.x.linked_symbols['y'] = self.z
self.x.linked_symbols['x'] = self.z
self.assertEqual(['a', 'b', 'c'],
self.x.get_dep_linked_symbols(self.y))
self.assertEqual(['w', 'x', 'y', 'z'],
self.x.get_dep_linked_symbols(self.z))