git.sur5r.net Git - u-boot/blob - tools/binman/elf_test.py

   1 #
   2 # Copyright (c) 2017 Google, Inc
   3 # Written by Simon Glass <sjg@chromium.org>
   4 #
   5 # SPDX-License-Identifier:      GPL-2.0+
   6 #
   7 # Test for the elf module
   8
   9 import os
  10 import sys
  11 import unittest
  12
  13 import elf
  14
  15 binman_dir = os.path.dirname(os.path.realpath(sys.argv[0]))
  16 fname = os.path.join(binman_dir, 'test', 'u_boot_ucode_ptr')
  17
  18 class TestElf(unittest.TestCase):
  19     def testAllSymbols(self):
  20         syms = elf.GetSymbols(fname, [])
  21         self.assertIn('.ucode', syms)
  22
  23     def testRegexSymbols(self):
  24         syms = elf.GetSymbols(fname, ['ucode'])
  25         self.assertIn('.ucode', syms)
  26         syms = elf.GetSymbols(fname, ['missing'])
  27         self.assertNotIn('.ucode', syms)
  28         syms = elf.GetSymbols(fname, ['missing', 'ucode'])
  29         self.assertIn('.ucode', syms)
  30
  31 if __name__ == '__main__':
  32     unittest.main()