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