1 # SPDX-License-Identifier: GPL-2.0+
2 # Copyright (c) 2017 Google, Inc
3 # Written by Simon Glass <sjg@chromium.org>
5 # Test for the image module
9 from image import Image
10 from test_util import capture_sys_output
12 class TestImage(unittest.TestCase):
13 def testInvalidFormat(self):
14 image = Image('name', 'node', test=True)
15 section = image._section
16 with self.assertRaises(ValueError) as e:
17 section.LookupSymbol('_binman_something_prop_', False, 'msg')
19 "msg: Symbol '_binman_something_prop_' has invalid format",
22 def testMissingSymbol(self):
23 image = Image('name', 'node', test=True)
24 section = image._section
26 with self.assertRaises(ValueError) as e:
27 section.LookupSymbol('_binman_type_prop_pname', False, 'msg')
28 self.assertIn("msg: Entry 'type' not found in list ()",
31 def testMissingSymbolOptional(self):
32 image = Image('name', 'node', test=True)
33 section = image._section
35 with capture_sys_output() as (stdout, stderr):
36 val = section.LookupSymbol('_binman_type_prop_pname', True, 'msg')
37 self.assertEqual(val, None)
38 self.assertEqual("Warning: msg: Entry 'type' not found in list ()\n",
40 self.assertEqual('', stdout.getvalue())
42 def testBadProperty(self):
43 image = Image('name', 'node', test=True)
44 section = image._section
45 section._entries = {'u-boot': 1}
46 with self.assertRaises(ValueError) as e:
47 section.LookupSymbol('_binman_u_boot_prop_bad', False, 'msg')
48 self.assertIn("msg: No such property 'bad", str(e.exception))