mirror of https://github.com/Qortal/Brooklyn
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
48 lines
1.8 KiB
48 lines
1.8 KiB
# SPDX-License-Identifier: GPL-2.0+ |
|
# Copyright (c) 2017 Google, Inc |
|
# Written by Simon Glass <[email protected]> |
|
# |
|
# Test for the image module |
|
|
|
import unittest |
|
|
|
from image import Image |
|
from test_util import capture_sys_output |
|
|
|
class TestImage(unittest.TestCase): |
|
def testInvalidFormat(self): |
|
image = Image('name', 'node', test=True) |
|
section = image._section |
|
with self.assertRaises(ValueError) as e: |
|
section.LookupSymbol('_binman_something_prop_', False, 'msg') |
|
self.assertIn( |
|
"msg: Symbol '_binman_something_prop_' has invalid format", |
|
str(e.exception)) |
|
|
|
def testMissingSymbol(self): |
|
image = Image('name', 'node', test=True) |
|
section = image._section |
|
section._entries = {} |
|
with self.assertRaises(ValueError) as e: |
|
section.LookupSymbol('_binman_type_prop_pname', False, 'msg') |
|
self.assertIn("msg: Entry 'type' not found in list ()", |
|
str(e.exception)) |
|
|
|
def testMissingSymbolOptional(self): |
|
image = Image('name', 'node', test=True) |
|
section = image._section |
|
section._entries = {} |
|
with capture_sys_output() as (stdout, stderr): |
|
val = section.LookupSymbol('_binman_type_prop_pname', True, 'msg') |
|
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) |
|
section = image._section |
|
section._entries = {'u-boot': 1} |
|
with self.assertRaises(ValueError) as e: |
|
section.LookupSymbol('_binman_u_boot_prop_bad', False, 'msg') |
|
self.assertIn("msg: No such property 'bad", str(e.exception))
|
|
|