]>
Commit | Line | Data |
---|---|---|
19790632 SG |
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 image module | |
8 | ||
9 | import unittest | |
10 | ||
11 | from image import Image | |
12 | from elf_test import capture_sys_output | |
13 | ||
14 | class TestImage(unittest.TestCase): | |
15 | def testInvalidFormat(self): | |
16 | image = Image('name', 'node', test=True) | |
17 | with self.assertRaises(ValueError) as e: | |
18 | image.LookupSymbol('_binman_something_prop_', False, 'msg') | |
19 | self.assertIn( | |
20 | "msg: Symbol '_binman_something_prop_' has invalid format", | |
21 | str(e.exception)) | |
22 | ||
23 | def testMissingSymbol(self): | |
24 | image = Image('name', 'node', test=True) | |
25 | image._entries = {} | |
26 | with self.assertRaises(ValueError) as e: | |
27 | image.LookupSymbol('_binman_type_prop_pname', False, 'msg') | |
28 | self.assertIn("msg: Entry 'type' not found in list ()", | |
29 | str(e.exception)) | |
30 | ||
31 | def testMissingSymbolOptional(self): | |
32 | image = Image('name', 'node', test=True) | |
33 | image._entries = {} | |
34 | with capture_sys_output() as (stdout, stderr): | |
35 | val = image.LookupSymbol('_binman_type_prop_pname', True, 'msg') | |
36 | self.assertEqual(val, None) | |
37 | self.assertEqual("Warning: msg: Entry 'type' not found in list ()\n", | |
38 | stderr.getvalue()) | |
39 | self.assertEqual('', stdout.getvalue()) | |
40 | ||
41 | def testBadProperty(self): | |
42 | image = Image('name', 'node', test=True) | |
43 | image._entries = {'u-boot': 1} | |
44 | with self.assertRaises(ValueError) as e: | |
45 | image.LookupSymbol('_binman_u_boot_prop_bad', False, 'msg') | |
46 | self.assertIn("msg: No such property 'bad", str(e.exception)) |