1*4f44304bSSimon Glass# 2*4f44304bSSimon Glass# Copyright (c) 2016 Google, Inc 3*4f44304bSSimon Glass# Written by Simon Glass <sjg@chromium.org> 4*4f44304bSSimon Glass# 5*4f44304bSSimon Glass# SPDX-License-Identifier: GPL-2.0+ 6*4f44304bSSimon Glass# 7*4f44304bSSimon Glass# Test for the Entry class 8*4f44304bSSimon Glass 9*4f44304bSSimon Glassimport collections 10*4f44304bSSimon Glassimport unittest 11*4f44304bSSimon Glass 12*4f44304bSSimon Glassimport entry 13*4f44304bSSimon Glass 14*4f44304bSSimon Glassclass TestEntry(unittest.TestCase): 15*4f44304bSSimon Glass def testEntryContents(self): 16*4f44304bSSimon Glass """Test the Entry bass class""" 17*4f44304bSSimon Glass base_entry = entry.Entry(None, None, None, read_node=False) 18*4f44304bSSimon Glass self.assertEqual(True, base_entry.ObtainContents()) 19*4f44304bSSimon Glass 20*4f44304bSSimon Glass def testUnknownEntry(self): 21*4f44304bSSimon Glass """Test that unknown entry types are detected""" 22*4f44304bSSimon Glass Node = collections.namedtuple('Node', ['name', 'path']) 23*4f44304bSSimon Glass node = Node('invalid-name', 'invalid-path') 24*4f44304bSSimon Glass with self.assertRaises(ValueError) as e: 25*4f44304bSSimon Glass entry.Entry.Create(None, node, node.name) 26*4f44304bSSimon Glass self.assertIn("Unknown entry type 'invalid-name' in node " 27*4f44304bSSimon Glass "'invalid-path'", str(e.exception)) 28