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