pylibfdt: Support device-tree creation/expansion

Add support for fdt_open_into() and fdt_create_empty_tree() from the
Python library. The former is named resize() since it better fits with
what the Python binding actually does.

Signed-off-by: Simon Glass <sjg@chromium.org>
Signed-off-by: David Gibson <david@gibson.dropbear.id.au>
This commit is contained in:
Simon Glass 2018-06-06 15:37:04 -06:00 committed by David Gibson
parent 483e170625
commit 5a598671fd
2 changed files with 50 additions and 1 deletions

View file

@ -55,7 +55,7 @@ import unittest
sys.path.insert(0, '../pylibfdt')
import libfdt
from libfdt import FdtException, QUIET_NOTFOUND, QUIET_ALL
from libfdt import Fdt, FdtException, QUIET_NOTFOUND, QUIET_ALL
def get_err(err_code):
"""Convert an error code into an error message
@ -364,6 +364,22 @@ class PyLibfdtTests(unittest.TestCase):
self.fdt.get_mem_rsv(0))
self.assertEquals([123456789, 010000], self.fdt.get_mem_rsv(1))
def testEmpty(self):
"""Test that we can create an empty tree"""
self.assertEquals(-libfdt.NOSPACE,
Fdt.create_empty_tree(1, (libfdt.NOSPACE,)))
fdt = Fdt.create_empty_tree(128)
self.assertEquals(128, fdt.totalsize())
def testOpenInto(self):
"""Test that we can resize a tree"""
fdt = Fdt.create_empty_tree(128)
self.assertEquals(128, fdt.totalsize())
fdt.resize(256)
self.assertEquals(256, fdt.totalsize())
fdt.pack()
self.assertTrue(fdt.totalsize() < 128)
if __name__ == "__main__":
unittest.main()