mirror of
https://github.com/brain-hackers/u-boot-brain
synced 2024-08-19 16:38:53 +09:00
5c890238c4
At present the contents of an entry are set in subclasses simply by assigning to the data and content_size properties. Add some methods to do this, so that we have more control. In particular, add a method to set the contents without changing its size, so we can validate that case. Add a test case for trying to change the size when this is not allowed. Signed-off-by: Simon Glass <sjg@chromium.org>
45 lines
1.4 KiB
Python
45 lines
1.4 KiB
Python
# SPDX-License-Identifier: GPL-2.0+
|
|
# Copyright (c) 2016 Google, Inc
|
|
# Written by Simon Glass <sjg@chromium.org>
|
|
#
|
|
# Entry-type module for testing purposes. Not used in real images.
|
|
#
|
|
|
|
from entry import Entry
|
|
import fdt_util
|
|
import tools
|
|
|
|
|
|
class Entry__testing(Entry):
|
|
"""A fake entry used for testing
|
|
|
|
Properties:
|
|
return_invalid_entry: Return an invalid entry from GetPositions()
|
|
"""
|
|
def __init__(self, section, etype, node):
|
|
Entry.__init__(self, section, etype, node)
|
|
self.return_invalid_entry = fdt_util.GetBool(self._node,
|
|
'return-invalid-entry')
|
|
self.return_unknown_contents = fdt_util.GetBool(self._node,
|
|
'return-unknown-contents')
|
|
self.bad_update_contents = fdt_util.GetBool(self._node,
|
|
'bad-update-contents')
|
|
|
|
def ObtainContents(self):
|
|
if self.return_unknown_contents:
|
|
return False
|
|
self.data = 'a'
|
|
self.contents_size = len(self.data)
|
|
return True
|
|
|
|
def GetPositions(self):
|
|
if self.return_invalid_entry :
|
|
return {'invalid-entry': [1, 2]}
|
|
return {}
|
|
|
|
def ProcessContents(self):
|
|
if self.bad_update_contents:
|
|
# Request to update the conents with something larger, to cause a
|
|
# failure.
|
|
self.ProcessContentsUpdate('aa')
|