mirror of
https://source.denx.de/u-boot/u-boot.git
synced 2025-12-19 16:31:27 +01:00
ae3695f691
This function is sometimes useful outside tests. Also it can affect how terminal output is done, e.g. whether ANSI characters should be emitted or not. Move it out of the test_util package and into terminal. Signed-off-by: Simon Glass <sjg@chromium.org>
45 lines
1.7 KiB
Python
45 lines
1.7 KiB
Python
# SPDX-License-Identifier: GPL-2.0+
|
|
# Copyright (c) 2017 Google, Inc
|
|
# Written by Simon Glass <sjg@chromium.org>
|
|
#
|
|
# Test for the image module
|
|
|
|
import unittest
|
|
|
|
from binman.image import Image
|
|
from u_boot_pylib import terminal
|
|
|
|
class TestImage(unittest.TestCase):
|
|
def testInvalidFormat(self):
|
|
image = Image('name', 'node', test=True)
|
|
with self.assertRaises(ValueError) as e:
|
|
image.GetSymbolValue('_binman_something_prop_', False, 'msg', 0)
|
|
self.assertIn(
|
|
"msg: Symbol '_binman_something_prop_' has invalid format",
|
|
str(e.exception))
|
|
|
|
def testMissingSymbol(self):
|
|
image = Image('name', 'node', test=True)
|
|
image._entries = {}
|
|
with self.assertRaises(ValueError) as e:
|
|
image.GetSymbolValue('_binman_type_prop_pname', False, 'msg', 0)
|
|
self.assertIn("msg: Entry 'type' not found in list ()",
|
|
str(e.exception))
|
|
|
|
def testMissingSymbolOptional(self):
|
|
image = Image('name', 'node', test=True)
|
|
image._entries = {}
|
|
with terminal.capture() as (stdout, stderr):
|
|
val = image.GetSymbolValue('_binman_type_prop_pname', True, 'msg', 0)
|
|
self.assertEqual(val, None)
|
|
self.assertEqual("Warning: msg: Entry 'type' not found in list ()\n",
|
|
stderr.getvalue())
|
|
self.assertEqual('', stdout.getvalue())
|
|
|
|
def testBadProperty(self):
|
|
image = Image('name', 'node', test=True)
|
|
image._entries = {'u-boot': 1}
|
|
with self.assertRaises(ValueError) as e:
|
|
image.GetSymbolValue('_binman_u_boot_prop_bad', False, 'msg', 0)
|
|
self.assertIn("msg: No such property 'bad", str(e.exception))
|