Simon Glass | f7ba5f0 | 2019-10-31 07:42:54 -0600 | [diff] [blame] | 1 | #!/usr/bin/env python3 |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 2 | # SPDX-License-Identifier: GPL-2.0+ |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 3 | |
| 4 | """ |
| 5 | Tests for the Fdt module |
| 6 | Copyright (c) 2018 Google, Inc |
| 7 | Written by Simon Glass <sjg@chromium.org> |
| 8 | """ |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 9 | |
Simon Glass | 7640b16 | 2022-07-30 20:57:09 -0600 | [diff] [blame] | 10 | from argparse import ArgumentParser |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 11 | import os |
Simon Glass | a004f29 | 2019-07-20 12:23:49 -0600 | [diff] [blame] | 12 | import shutil |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 13 | import sys |
Simon Glass | a004f29 | 2019-07-20 12:23:49 -0600 | [diff] [blame] | 14 | import tempfile |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 15 | import unittest |
| 16 | |
| 17 | # Bring in the patman libraries |
| 18 | our_path = os.path.dirname(os.path.realpath(__file__)) |
Simon Glass | b4fa949 | 2020-04-17 18:09:05 -0600 | [diff] [blame] | 19 | sys.path.insert(1, os.path.join(our_path, '..')) |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 20 | |
Simon Glass | ff139b6 | 2021-11-23 11:03:38 -0700 | [diff] [blame] | 21 | # Bring in the libfdt module |
| 22 | sys.path.insert(2, 'scripts/dtc/pylibfdt') |
| 23 | sys.path.insert(2, os.path.join(our_path, '../../scripts/dtc/pylibfdt')) |
| 24 | sys.path.insert(2, os.path.join(our_path, |
| 25 | '../../build-sandbox_spl/scripts/dtc/pylibfdt')) |
| 26 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 27 | #pylint: disable=wrong-import-position |
Simon Glass | bf77667 | 2020-04-17 18:09:04 -0600 | [diff] [blame] | 28 | from dtoc import fdt |
| 29 | from dtoc import fdt_util |
Simon Glass | d866e62 | 2021-11-23 11:03:39 -0700 | [diff] [blame] | 30 | from dtoc.fdt_util import fdt32_to_cpu, fdt64_to_cpu |
Simon Glass | 8a455fc | 2022-02-11 13:23:20 -0700 | [diff] [blame] | 31 | from dtoc.fdt import Type, BytesToValue |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 32 | import libfdt |
Simon Glass | 4583c00 | 2023-02-23 18:18:04 -0700 | [diff] [blame] | 33 | from u_boot_pylib import test_util |
| 34 | from u_boot_pylib import tools |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 35 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 36 | #pylint: disable=protected-access |
| 37 | |
| 38 | def _get_property_value(dtb, node, prop_name): |
Simon Glass | f9b88b3 | 2018-07-06 10:27:29 -0600 | [diff] [blame] | 39 | """Low-level function to get the property value based on its offset |
| 40 | |
| 41 | This looks directly in the device tree at the property's offset to find |
| 42 | its value. It is useful as a check that the property is in the correct |
| 43 | place. |
| 44 | |
| 45 | Args: |
| 46 | node: Node to look in |
| 47 | prop_name: Property name to find |
| 48 | |
| 49 | Returns: |
| 50 | Tuple: |
| 51 | Prop object found |
| 52 | Value of property as a string (found using property offset) |
| 53 | """ |
| 54 | prop = node.props[prop_name] |
| 55 | |
| 56 | # Add 12, which is sizeof(struct fdt_property), to get to start of data |
| 57 | offset = prop.GetOffset() + 12 |
| 58 | data = dtb.GetContents()[offset:offset + len(prop.value)] |
Simon Glass | 479dd30 | 2020-11-08 20:36:20 -0700 | [diff] [blame] | 59 | return prop, [chr(x) for x in data] |
Simon Glass | f9b88b3 | 2018-07-06 10:27:29 -0600 | [diff] [blame] | 60 | |
Simon Glass | dff51a5 | 2021-02-03 06:00:56 -0700 | [diff] [blame] | 61 | def find_dtb_file(dts_fname): |
| 62 | """Locate a test file in the test/ directory |
| 63 | |
| 64 | Args: |
| 65 | dts_fname (str): Filename to find, e.g. 'dtoc_test_simple.dts] |
| 66 | |
| 67 | Returns: |
| 68 | str: Path to the test filename |
| 69 | """ |
| 70 | return os.path.join('tools/dtoc/test', dts_fname) |
| 71 | |
Simon Glass | f9b88b3 | 2018-07-06 10:27:29 -0600 | [diff] [blame] | 72 | |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 73 | class TestFdt(unittest.TestCase): |
| 74 | """Tests for the Fdt module |
| 75 | |
| 76 | This includes unit tests for some functions and functional tests for the fdt |
| 77 | module. |
| 78 | """ |
| 79 | @classmethod |
| 80 | def setUpClass(cls): |
Simon Glass | c1aa66e | 2022-01-29 14:14:04 -0700 | [diff] [blame] | 81 | tools.prepare_output_dir(None) |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 82 | |
| 83 | @classmethod |
| 84 | def tearDownClass(cls): |
Simon Glass | c1aa66e | 2022-01-29 14:14:04 -0700 | [diff] [blame] | 85 | tools.finalise_output_dir() |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 86 | |
| 87 | def setUp(self): |
Simon Glass | dff51a5 | 2021-02-03 06:00:56 -0700 | [diff] [blame] | 88 | self.dtb = fdt.FdtScan(find_dtb_file('dtoc_test_simple.dts')) |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 89 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 90 | def test_fdt(self): |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 91 | """Test that we can open an Fdt""" |
| 92 | self.dtb.Scan() |
| 93 | root = self.dtb.GetRoot() |
| 94 | self.assertTrue(isinstance(root, fdt.Node)) |
| 95 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 96 | def test_get_node(self): |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 97 | """Test the GetNode() method""" |
| 98 | node = self.dtb.GetNode('/spl-test') |
| 99 | self.assertTrue(isinstance(node, fdt.Node)) |
Simon Glass | e44bc83 | 2019-07-20 12:23:39 -0600 | [diff] [blame] | 100 | |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 101 | node = self.dtb.GetNode('/i2c@0/pmic@9') |
| 102 | self.assertTrue(isinstance(node, fdt.Node)) |
| 103 | self.assertEqual('pmic@9', node.name) |
Simon Glass | 2a2d91d | 2018-07-06 10:27:28 -0600 | [diff] [blame] | 104 | self.assertIsNone(self.dtb.GetNode('/i2c@0/pmic@9/missing')) |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 105 | |
Simon Glass | e44bc83 | 2019-07-20 12:23:39 -0600 | [diff] [blame] | 106 | node = self.dtb.GetNode('/') |
| 107 | self.assertTrue(isinstance(node, fdt.Node)) |
| 108 | self.assertEqual(0, node.Offset()) |
| 109 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 110 | def test_flush(self): |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 111 | """Check that we can flush the device tree out to its file""" |
| 112 | fname = self.dtb._fname |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 113 | with open(fname, 'rb') as inf: |
| 114 | inf.read() |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 115 | os.remove(fname) |
| 116 | with self.assertRaises(IOError): |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 117 | with open(fname, 'rb'): |
| 118 | pass |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 119 | self.dtb.Flush() |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 120 | with open(fname, 'rb') as inf: |
| 121 | inf.read() |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 122 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 123 | def test_pack(self): |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 124 | """Test that packing a device tree works""" |
| 125 | self.dtb.Pack() |
| 126 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 127 | def test_get_fdt_raw(self): |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 128 | """Tetst that we can access the raw device-tree data""" |
Simon Glass | 8a455fc | 2022-02-11 13:23:20 -0700 | [diff] [blame] | 129 | self.assertTrue(isinstance(self.dtb.GetContents(), bytes)) |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 130 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 131 | def test_get_props(self): |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 132 | """Tests obtaining a list of properties""" |
| 133 | node = self.dtb.GetNode('/spl-test') |
| 134 | props = self.dtb.GetProps(node) |
Simon Glass | e316fba | 2023-02-13 08:56:34 -0700 | [diff] [blame] | 135 | self.assertEqual(['boolval', 'bootph-all', 'bytearray', 'byteval', |
| 136 | 'compatible', 'int64val', 'intarray', 'intval', |
| 137 | 'longbytearray', 'maybe-empty-int', 'notstring', |
| 138 | 'stringarray', 'stringval', ], |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 139 | sorted(props.keys())) |
| 140 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 141 | def test_check_error(self): |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 142 | """Tests the ChecKError() function""" |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 143 | with self.assertRaises(ValueError) as exc: |
Simon Glass | 2a2d91d | 2018-07-06 10:27:28 -0600 | [diff] [blame] | 144 | fdt.CheckErr(-libfdt.NOTFOUND, 'hello') |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 145 | self.assertIn('FDT_ERR_NOTFOUND: hello', str(exc.exception)) |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 146 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 147 | def test_get_fdt(self): |
| 148 | """Test getting an Fdt object from a node""" |
Simon Glass | 94a7c60 | 2018-07-17 13:25:46 -0600 | [diff] [blame] | 149 | node = self.dtb.GetNode('/spl-test') |
| 150 | self.assertEqual(self.dtb, node.GetFdt()) |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 151 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 152 | def test_bytes_to_value(self): |
| 153 | """Test converting a string list into Python""" |
Simon Glass | b5f0daf | 2019-05-17 22:00:41 -0600 | [diff] [blame] | 154 | self.assertEqual(BytesToValue(b'this\0is\0'), |
Simon Glass | 5ea9dcc | 2020-11-08 20:36:17 -0700 | [diff] [blame] | 155 | (Type.STRING, ['this', 'is'])) |
Simon Glass | b5f0daf | 2019-05-17 22:00:41 -0600 | [diff] [blame] | 156 | |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 157 | class TestNode(unittest.TestCase): |
| 158 | """Test operation of the Node class""" |
| 159 | |
| 160 | @classmethod |
| 161 | def setUpClass(cls): |
Simon Glass | c1aa66e | 2022-01-29 14:14:04 -0700 | [diff] [blame] | 162 | tools.prepare_output_dir(None) |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 163 | |
| 164 | @classmethod |
| 165 | def tearDownClass(cls): |
Simon Glass | c1aa66e | 2022-01-29 14:14:04 -0700 | [diff] [blame] | 166 | tools.finalise_output_dir() |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 167 | |
| 168 | def setUp(self): |
Simon Glass | dff51a5 | 2021-02-03 06:00:56 -0700 | [diff] [blame] | 169 | self.dtb = fdt.FdtScan(find_dtb_file('dtoc_test_simple.dts')) |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 170 | self.node = self.dtb.GetNode('/spl-test') |
Simon Glass | 76677dd | 2021-03-21 18:24:37 +1300 | [diff] [blame] | 171 | self.fdt = self.dtb.GetFdtObj() |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 172 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 173 | def test_offset(self): |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 174 | """Tests that we can obtain the offset of a node""" |
| 175 | self.assertTrue(self.node.Offset() > 0) |
| 176 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 177 | def test_delete(self): |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 178 | """Tests that we can delete a property""" |
| 179 | node2 = self.dtb.GetNode('/spl-test2') |
| 180 | offset1 = node2.Offset() |
| 181 | self.node.DeleteProp('intval') |
| 182 | offset2 = node2.Offset() |
| 183 | self.assertTrue(offset2 < offset1) |
| 184 | self.node.DeleteProp('intarray') |
| 185 | offset3 = node2.Offset() |
| 186 | self.assertTrue(offset3 < offset2) |
Simon Glass | 2a2d91d | 2018-07-06 10:27:28 -0600 | [diff] [blame] | 187 | with self.assertRaises(libfdt.FdtException): |
| 188 | self.node.DeleteProp('missing') |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 189 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 190 | def test_delete_get_offset(self): |
Simon Glass | f9b88b3 | 2018-07-06 10:27:29 -0600 | [diff] [blame] | 191 | """Test that property offset update when properties are deleted""" |
| 192 | self.node.DeleteProp('intval') |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 193 | prop, value = _get_property_value(self.dtb, self.node, 'longbytearray') |
Simon Glass | f9b88b3 | 2018-07-06 10:27:29 -0600 | [diff] [blame] | 194 | self.assertEqual(prop.value, value) |
| 195 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 196 | def test_find_node(self): |
Simon Glass | 1d85888 | 2018-07-17 13:25:41 -0600 | [diff] [blame] | 197 | """Tests that we can find a node using the FindNode() functoin""" |
| 198 | node = self.dtb.GetRoot().FindNode('i2c@0') |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 199 | self.assertEqual('i2c@0', node.name) |
Simon Glass | 1d85888 | 2018-07-17 13:25:41 -0600 | [diff] [blame] | 200 | subnode = node.FindNode('pmic@9') |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 201 | self.assertEqual('pmic@9', subnode.name) |
Simon Glass | 1d85888 | 2018-07-17 13:25:41 -0600 | [diff] [blame] | 202 | self.assertEqual(None, node.FindNode('missing')) |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 203 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 204 | def test_refresh_missing_node(self): |
Simon Glass | f9b88b3 | 2018-07-06 10:27:29 -0600 | [diff] [blame] | 205 | """Test refreshing offsets when an extra node is present in dtb""" |
| 206 | # Delete it from our tables, not the device tree |
| 207 | del self.dtb._root.subnodes[-1] |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 208 | with self.assertRaises(ValueError) as exc: |
Simon Glass | f9b88b3 | 2018-07-06 10:27:29 -0600 | [diff] [blame] | 209 | self.dtb.Refresh() |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 210 | self.assertIn('Internal error, offset', str(exc.exception)) |
Simon Glass | f9b88b3 | 2018-07-06 10:27:29 -0600 | [diff] [blame] | 211 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 212 | def test_refresh_extra_node(self): |
Simon Glass | f9b88b3 | 2018-07-06 10:27:29 -0600 | [diff] [blame] | 213 | """Test refreshing offsets when an expected node is missing""" |
| 214 | # Delete it from the device tre, not our tables |
Simon Glass | 76677dd | 2021-03-21 18:24:37 +1300 | [diff] [blame] | 215 | self.fdt.del_node(self.node.Offset()) |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 216 | with self.assertRaises(ValueError) as exc: |
Simon Glass | f9b88b3 | 2018-07-06 10:27:29 -0600 | [diff] [blame] | 217 | self.dtb.Refresh() |
| 218 | self.assertIn('Internal error, node name mismatch ' |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 219 | 'spl-test != spl-test2', str(exc.exception)) |
Simon Glass | f9b88b3 | 2018-07-06 10:27:29 -0600 | [diff] [blame] | 220 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 221 | def test_refresh_missing_prop(self): |
Simon Glass | f9b88b3 | 2018-07-06 10:27:29 -0600 | [diff] [blame] | 222 | """Test refreshing offsets when an extra property is present in dtb""" |
| 223 | # Delete it from our tables, not the device tree |
| 224 | del self.node.props['notstring'] |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 225 | with self.assertRaises(ValueError) as exc: |
Simon Glass | f9b88b3 | 2018-07-06 10:27:29 -0600 | [diff] [blame] | 226 | self.dtb.Refresh() |
Simon Glass | acd9861 | 2021-03-21 18:24:34 +1300 | [diff] [blame] | 227 | self.assertIn("Internal error, node '/spl-test' property 'notstring' missing, offset ", |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 228 | str(exc.exception)) |
Simon Glass | f9b88b3 | 2018-07-06 10:27:29 -0600 | [diff] [blame] | 229 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 230 | def test_lookup_phandle(self): |
Simon Glass | 94a7c60 | 2018-07-17 13:25:46 -0600 | [diff] [blame] | 231 | """Test looking up a single phandle""" |
Simon Glass | dff51a5 | 2021-02-03 06:00:56 -0700 | [diff] [blame] | 232 | dtb = fdt.FdtScan(find_dtb_file('dtoc_test_phandle.dts')) |
Simon Glass | 94a7c60 | 2018-07-17 13:25:46 -0600 | [diff] [blame] | 233 | node = dtb.GetNode('/phandle-source2') |
| 234 | prop = node.props['clocks'] |
| 235 | target = dtb.GetNode('/phandle-target') |
| 236 | self.assertEqual(target, dtb.LookupPhandle(fdt32_to_cpu(prop.value))) |
| 237 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 238 | def test_add_node_space(self): |
Simon Glass | 76677dd | 2021-03-21 18:24:37 +1300 | [diff] [blame] | 239 | """Test adding a single node when out of space""" |
| 240 | self.fdt.pack() |
| 241 | self.node.AddSubnode('subnode') |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 242 | with self.assertRaises(libfdt.FdtException) as exc: |
Simon Glass | 76677dd | 2021-03-21 18:24:37 +1300 | [diff] [blame] | 243 | self.dtb.Sync(auto_resize=False) |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 244 | self.assertIn('FDT_ERR_NOSPACE', str(exc.exception)) |
Simon Glass | 76677dd | 2021-03-21 18:24:37 +1300 | [diff] [blame] | 245 | |
| 246 | self.dtb.Sync(auto_resize=True) |
| 247 | offset = self.fdt.path_offset('/spl-test/subnode') |
| 248 | self.assertTrue(offset > 0) |
| 249 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 250 | def test_add_nodes(self): |
Simon Glass | 76677dd | 2021-03-21 18:24:37 +1300 | [diff] [blame] | 251 | """Test adding various subnode and properies""" |
| 252 | node = self.dtb.GetNode('/i2c@0') |
| 253 | |
Simon Glass | f617665 | 2021-03-21 18:24:38 +1300 | [diff] [blame] | 254 | # Add one more node next to the pmic one |
| 255 | sn1 = node.AddSubnode('node-one') |
| 256 | sn1.AddInt('integer-a', 12) |
| 257 | sn1.AddInt('integer-b', 23) |
| 258 | |
| 259 | # Sync so that everything is clean |
| 260 | self.dtb.Sync(auto_resize=True) |
| 261 | |
| 262 | # Add two subnodes next to pmic and node-one |
| 263 | sn2 = node.AddSubnode('node-two') |
| 264 | sn2.AddInt('integer-2a', 34) |
| 265 | sn2.AddInt('integer-2b', 45) |
| 266 | |
| 267 | sn3 = node.AddSubnode('node-three') |
| 268 | sn3.AddInt('integer-3', 123) |
| 269 | |
Simon Glass | 76677dd | 2021-03-21 18:24:37 +1300 | [diff] [blame] | 270 | # Add a property to the node after i2c@0 to check that this is not |
| 271 | # disturbed by adding a subnode to i2c@0 |
| 272 | orig_node = self.dtb.GetNode('/orig-node') |
| 273 | orig_node.AddInt('integer-4', 456) |
| 274 | |
| 275 | # Add a property to the pmic node to check that pmic properties are not |
| 276 | # disturbed |
| 277 | pmic = self.dtb.GetNode('/i2c@0/pmic@9') |
| 278 | pmic.AddInt('integer-5', 567) |
| 279 | |
| 280 | self.dtb.Sync(auto_resize=True) |
| 281 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 282 | def test_add_one_node(self): |
Simon Glass | dd857ee | 2022-02-08 11:49:52 -0700 | [diff] [blame] | 283 | """Testing deleting and adding a subnode before syncing""" |
| 284 | subnode = self.node.AddSubnode('subnode') |
| 285 | self.node.AddSubnode('subnode2') |
| 286 | self.dtb.Sync(auto_resize=True) |
| 287 | |
| 288 | # Delete a node and add a new one |
| 289 | subnode.Delete() |
| 290 | self.node.AddSubnode('subnode3') |
| 291 | self.dtb.Sync() |
| 292 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 293 | def test_refresh_name_mismatch(self): |
Simon Glass | 5d1bec3 | 2021-03-21 18:24:39 +1300 | [diff] [blame] | 294 | """Test name mismatch when syncing nodes and properties""" |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 295 | self.node.AddInt('integer-a', 12) |
Simon Glass | 5d1bec3 | 2021-03-21 18:24:39 +1300 | [diff] [blame] | 296 | |
| 297 | wrong_offset = self.dtb.GetNode('/i2c@0')._offset |
| 298 | self.node._offset = wrong_offset |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 299 | with self.assertRaises(ValueError) as exc: |
Simon Glass | 5d1bec3 | 2021-03-21 18:24:39 +1300 | [diff] [blame] | 300 | self.dtb.Sync() |
| 301 | self.assertIn("Internal error, node '/spl-test' name mismatch 'i2c@0'", |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 302 | str(exc.exception)) |
Simon Glass | 5d1bec3 | 2021-03-21 18:24:39 +1300 | [diff] [blame] | 303 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 304 | with self.assertRaises(ValueError) as exc: |
Simon Glass | 5d1bec3 | 2021-03-21 18:24:39 +1300 | [diff] [blame] | 305 | self.node.Refresh(wrong_offset) |
| 306 | self.assertIn("Internal error, node '/spl-test' name mismatch 'i2c@0'", |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 307 | str(exc.exception)) |
Simon Glass | 5d1bec3 | 2021-03-21 18:24:39 +1300 | [diff] [blame] | 308 | |
Simon Glass | 4df457b | 2023-07-18 07:24:02 -0600 | [diff] [blame^] | 309 | def test_copy_node(self): |
| 310 | """Test copy_node() function""" |
| 311 | def do_copy_checks(dtb, dst, expect_none): |
| 312 | self.assertEqual( |
| 313 | ['/dest/base', '/dest/first@0', '/dest/existing'], |
| 314 | [n.path for n in dst.subnodes]) |
| 315 | |
| 316 | chk = dtb.GetNode('/dest/base') |
| 317 | self.assertTrue(chk) |
| 318 | self.assertEqual( |
| 319 | {'compatible', 'bootph-all', '#address-cells', '#size-cells'}, |
| 320 | chk.props.keys()) |
| 321 | |
| 322 | # Check the first property |
| 323 | prop = chk.props['bootph-all'] |
| 324 | self.assertEqual('bootph-all', prop.name) |
| 325 | self.assertEqual(True, prop.value) |
| 326 | self.assertEqual(chk.path, prop._node.path) |
| 327 | |
| 328 | # Check the second property |
| 329 | prop2 = chk.props['compatible'] |
| 330 | self.assertEqual('compatible', prop2.name) |
| 331 | self.assertEqual('sandbox,i2c', prop2.value) |
| 332 | self.assertEqual(chk.path, prop2._node.path) |
| 333 | |
| 334 | base = chk.FindNode('base') |
| 335 | self.assertTrue(chk) |
| 336 | |
| 337 | first = dtb.GetNode('/dest/base/first@0') |
| 338 | self.assertTrue(first) |
| 339 | over = dtb.GetNode('/dest/base/over') |
| 340 | self.assertTrue(over) |
| 341 | |
| 342 | # Make sure that the phandle for 'over' is not copied |
| 343 | self.assertNotIn('phandle', over.props.keys()) |
| 344 | |
| 345 | second = dtb.GetNode('/dest/base/second') |
| 346 | self.assertTrue(second) |
| 347 | self.assertEqual([over.name, first.name, second.name], |
| 348 | [n.name for n in chk.subnodes]) |
| 349 | self.assertEqual(chk, over.parent) |
| 350 | self.assertEqual( |
| 351 | {'bootph-all', 'compatible', 'reg', 'low-power'}, |
| 352 | over.props.keys()) |
| 353 | |
| 354 | if expect_none: |
| 355 | self.assertIsNone(prop._offset) |
| 356 | self.assertIsNone(prop2._offset) |
| 357 | self.assertIsNone(over._offset) |
| 358 | else: |
| 359 | self.assertTrue(prop._offset) |
| 360 | self.assertTrue(prop2._offset) |
| 361 | self.assertTrue(over._offset) |
| 362 | |
| 363 | # Now check ordering of the subnodes |
| 364 | self.assertEqual( |
| 365 | ['second1', 'second2', 'second3', 'second4'], |
| 366 | [n.name for n in second.subnodes]) |
| 367 | |
| 368 | dtb = fdt.FdtScan(find_dtb_file('dtoc_test_copy.dts')) |
| 369 | tmpl = dtb.GetNode('/base') |
| 370 | dst = dtb.GetNode('/dest') |
| 371 | dst.copy_node(tmpl) |
| 372 | |
| 373 | do_copy_checks(dtb, dst, expect_none=True) |
| 374 | |
| 375 | dtb.Sync(auto_resize=True) |
| 376 | |
| 377 | # Now check that the FDT looks correct |
| 378 | new_dtb = fdt.Fdt.FromData(dtb.GetContents()) |
| 379 | new_dtb.Scan() |
| 380 | dst = new_dtb.GetNode('/dest') |
| 381 | do_copy_checks(new_dtb, dst, expect_none=False) |
| 382 | |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 383 | |
| 384 | class TestProp(unittest.TestCase): |
| 385 | """Test operation of the Prop class""" |
| 386 | |
| 387 | @classmethod |
| 388 | def setUpClass(cls): |
Simon Glass | c1aa66e | 2022-01-29 14:14:04 -0700 | [diff] [blame] | 389 | tools.prepare_output_dir(None) |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 390 | |
| 391 | @classmethod |
| 392 | def tearDownClass(cls): |
Simon Glass | c1aa66e | 2022-01-29 14:14:04 -0700 | [diff] [blame] | 393 | tools.finalise_output_dir() |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 394 | |
| 395 | def setUp(self): |
Simon Glass | dff51a5 | 2021-02-03 06:00:56 -0700 | [diff] [blame] | 396 | self.dtb = fdt.FdtScan(find_dtb_file('dtoc_test_simple.dts')) |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 397 | self.node = self.dtb.GetNode('/spl-test') |
| 398 | self.fdt = self.dtb.GetFdtObj() |
| 399 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 400 | def test_missing_node(self): |
| 401 | """Test GetNode() when the node is missing""" |
Simon Glass | b9066ff | 2018-07-06 10:27:30 -0600 | [diff] [blame] | 402 | self.assertEqual(None, self.dtb.GetNode('missing')) |
| 403 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 404 | def test_phandle(self): |
| 405 | """Test GetNode() on a phandle""" |
Simon Glass | dff51a5 | 2021-02-03 06:00:56 -0700 | [diff] [blame] | 406 | dtb = fdt.FdtScan(find_dtb_file('dtoc_test_phandle.dts')) |
Simon Glass | 760b717 | 2018-07-06 10:27:31 -0600 | [diff] [blame] | 407 | node = dtb.GetNode('/phandle-source2') |
| 408 | prop = node.props['clocks'] |
| 409 | self.assertTrue(fdt32_to_cpu(prop.value) > 0) |
Simon Glass | 2a2d91d | 2018-07-06 10:27:28 -0600 | [diff] [blame] | 410 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 411 | def _convert_prop(self, prop_name): |
Simon Glass | 2a2d91d | 2018-07-06 10:27:28 -0600 | [diff] [blame] | 412 | """Helper function to look up a property in self.node and return it |
| 413 | |
| 414 | Args: |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 415 | str: Property name to find |
Simon Glass | 2a2d91d | 2018-07-06 10:27:28 -0600 | [diff] [blame] | 416 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 417 | Returns: |
| 418 | fdt.Prop: object for this property |
Simon Glass | 2a2d91d | 2018-07-06 10:27:28 -0600 | [diff] [blame] | 419 | """ |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 420 | prop = self.fdt.getprop(self.node.Offset(), prop_name) |
| 421 | return fdt.Prop(self.node, -1, prop_name, prop) |
Simon Glass | 2a2d91d | 2018-07-06 10:27:28 -0600 | [diff] [blame] | 422 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 423 | def test_make_prop(self): |
Simon Glass | 2a2d91d | 2018-07-06 10:27:28 -0600 | [diff] [blame] | 424 | """Test we can convert all the the types that are supported""" |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 425 | prop = self._convert_prop('boolval') |
Simon Glass | 5ea9dcc | 2020-11-08 20:36:17 -0700 | [diff] [blame] | 426 | self.assertEqual(Type.BOOL, prop.type) |
Simon Glass | 2a2d91d | 2018-07-06 10:27:28 -0600 | [diff] [blame] | 427 | self.assertEqual(True, prop.value) |
| 428 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 429 | prop = self._convert_prop('intval') |
Simon Glass | 5ea9dcc | 2020-11-08 20:36:17 -0700 | [diff] [blame] | 430 | self.assertEqual(Type.INT, prop.type) |
Simon Glass | 2a2d91d | 2018-07-06 10:27:28 -0600 | [diff] [blame] | 431 | self.assertEqual(1, fdt32_to_cpu(prop.value)) |
| 432 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 433 | prop = self._convert_prop('int64val') |
Simon Glass | d866e62 | 2021-11-23 11:03:39 -0700 | [diff] [blame] | 434 | self.assertEqual(Type.INT, prop.type) |
| 435 | self.assertEqual(0x123456789abcdef0, fdt64_to_cpu(prop.value)) |
| 436 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 437 | prop = self._convert_prop('intarray') |
Simon Glass | 5ea9dcc | 2020-11-08 20:36:17 -0700 | [diff] [blame] | 438 | self.assertEqual(Type.INT, prop.type) |
Simon Glass | 2a2d91d | 2018-07-06 10:27:28 -0600 | [diff] [blame] | 439 | val = [fdt32_to_cpu(val) for val in prop.value] |
| 440 | self.assertEqual([2, 3, 4], val) |
| 441 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 442 | prop = self._convert_prop('byteval') |
Simon Glass | 5ea9dcc | 2020-11-08 20:36:17 -0700 | [diff] [blame] | 443 | self.assertEqual(Type.BYTE, prop.type) |
Simon Glass | 2a2d91d | 2018-07-06 10:27:28 -0600 | [diff] [blame] | 444 | self.assertEqual(5, ord(prop.value)) |
| 445 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 446 | prop = self._convert_prop('longbytearray') |
Simon Glass | 5ea9dcc | 2020-11-08 20:36:17 -0700 | [diff] [blame] | 447 | self.assertEqual(Type.BYTE, prop.type) |
Simon Glass | 2a2d91d | 2018-07-06 10:27:28 -0600 | [diff] [blame] | 448 | val = [ord(val) for val in prop.value] |
| 449 | self.assertEqual([9, 10, 11, 12, 13, 14, 15, 16, 17], val) |
| 450 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 451 | prop = self._convert_prop('stringval') |
Simon Glass | 5ea9dcc | 2020-11-08 20:36:17 -0700 | [diff] [blame] | 452 | self.assertEqual(Type.STRING, prop.type) |
Simon Glass | 2a2d91d | 2018-07-06 10:27:28 -0600 | [diff] [blame] | 453 | self.assertEqual('message', prop.value) |
| 454 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 455 | prop = self._convert_prop('stringarray') |
Simon Glass | 5ea9dcc | 2020-11-08 20:36:17 -0700 | [diff] [blame] | 456 | self.assertEqual(Type.STRING, prop.type) |
Simon Glass | 2a2d91d | 2018-07-06 10:27:28 -0600 | [diff] [blame] | 457 | self.assertEqual(['multi-word', 'message'], prop.value) |
| 458 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 459 | prop = self._convert_prop('notstring') |
Simon Glass | 5ea9dcc | 2020-11-08 20:36:17 -0700 | [diff] [blame] | 460 | self.assertEqual(Type.BYTE, prop.type) |
Simon Glass | 2a2d91d | 2018-07-06 10:27:28 -0600 | [diff] [blame] | 461 | val = [ord(val) for val in prop.value] |
| 462 | self.assertEqual([0x20, 0x21, 0x22, 0x10, 0], val) |
| 463 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 464 | def test_get_empty(self): |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 465 | """Tests the GetEmpty() function for the various supported types""" |
Simon Glass | 5ea9dcc | 2020-11-08 20:36:17 -0700 | [diff] [blame] | 466 | self.assertEqual(True, fdt.Prop.GetEmpty(Type.BOOL)) |
| 467 | self.assertEqual(chr(0), fdt.Prop.GetEmpty(Type.BYTE)) |
Simon Glass | c1aa66e | 2022-01-29 14:14:04 -0700 | [diff] [blame] | 468 | self.assertEqual(tools.get_bytes(0, 4), fdt.Prop.GetEmpty(Type.INT)) |
Simon Glass | 5ea9dcc | 2020-11-08 20:36:17 -0700 | [diff] [blame] | 469 | self.assertEqual('', fdt.Prop.GetEmpty(Type.STRING)) |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 470 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 471 | def test_get_offset(self): |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 472 | """Test we can get the offset of a property""" |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 473 | prop, value = _get_property_value(self.dtb, self.node, 'longbytearray') |
Simon Glass | f9b88b3 | 2018-07-06 10:27:29 -0600 | [diff] [blame] | 474 | self.assertEqual(prop.value, value) |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 475 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 476 | def test_widen(self): |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 477 | """Test widening of values""" |
| 478 | node2 = self.dtb.GetNode('/spl-test2') |
Simon Glass | e144caf | 2020-10-03 11:31:27 -0600 | [diff] [blame] | 479 | node3 = self.dtb.GetNode('/spl-test3') |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 480 | prop = self.node.props['intval'] |
| 481 | |
| 482 | # No action |
| 483 | prop2 = node2.props['intval'] |
| 484 | prop.Widen(prop2) |
Simon Glass | 5ea9dcc | 2020-11-08 20:36:17 -0700 | [diff] [blame] | 485 | self.assertEqual(Type.INT, prop.type) |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 486 | self.assertEqual(1, fdt32_to_cpu(prop.value)) |
| 487 | |
Simon Glass | ca04494 | 2021-07-28 19:23:10 -0600 | [diff] [blame] | 488 | # Convert single value to array |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 489 | prop2 = self.node.props['intarray'] |
| 490 | prop.Widen(prop2) |
Simon Glass | 5ea9dcc | 2020-11-08 20:36:17 -0700 | [diff] [blame] | 491 | self.assertEqual(Type.INT, prop.type) |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 492 | self.assertTrue(isinstance(prop.value, list)) |
| 493 | |
| 494 | # A 4-byte array looks like a single integer. When widened by a longer |
| 495 | # byte array, it should turn into an array. |
| 496 | prop = self.node.props['longbytearray'] |
| 497 | prop2 = node2.props['longbytearray'] |
Simon Glass | e144caf | 2020-10-03 11:31:27 -0600 | [diff] [blame] | 498 | prop3 = node3.props['longbytearray'] |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 499 | self.assertFalse(isinstance(prop2.value, list)) |
| 500 | self.assertEqual(4, len(prop2.value)) |
Simon Glass | e144caf | 2020-10-03 11:31:27 -0600 | [diff] [blame] | 501 | self.assertEqual(b'\x09\x0a\x0b\x0c', prop2.value) |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 502 | prop2.Widen(prop) |
| 503 | self.assertTrue(isinstance(prop2.value, list)) |
| 504 | self.assertEqual(9, len(prop2.value)) |
Simon Glass | e144caf | 2020-10-03 11:31:27 -0600 | [diff] [blame] | 505 | self.assertEqual(['\x09', '\x0a', '\x0b', '\x0c', '\0', |
| 506 | '\0', '\0', '\0', '\0'], prop2.value) |
| 507 | prop3.Widen(prop) |
| 508 | self.assertTrue(isinstance(prop3.value, list)) |
| 509 | self.assertEqual(9, len(prop3.value)) |
| 510 | self.assertEqual(['\x09', '\x0a', '\x0b', '\x0c', '\x0d', |
| 511 | '\x0e', '\x0f', '\x10', '\0'], prop3.value) |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 512 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 513 | def test_widen_more(self): |
| 514 | """More tests of widening values""" |
| 515 | node2 = self.dtb.GetNode('/spl-test2') |
| 516 | node3 = self.dtb.GetNode('/spl-test3') |
| 517 | prop = self.node.props['intval'] |
| 518 | |
| 519 | # Test widening a single string into a string array |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 520 | prop = self.node.props['stringval'] |
| 521 | prop2 = node2.props['stringarray'] |
| 522 | self.assertFalse(isinstance(prop.value, list)) |
| 523 | self.assertEqual(7, len(prop.value)) |
| 524 | prop.Widen(prop2) |
| 525 | self.assertTrue(isinstance(prop.value, list)) |
| 526 | self.assertEqual(3, len(prop.value)) |
| 527 | |
| 528 | # Enlarging an existing array |
| 529 | prop = self.node.props['stringarray'] |
| 530 | prop2 = node2.props['stringarray'] |
| 531 | self.assertTrue(isinstance(prop.value, list)) |
| 532 | self.assertEqual(2, len(prop.value)) |
| 533 | prop.Widen(prop2) |
| 534 | self.assertTrue(isinstance(prop.value, list)) |
| 535 | self.assertEqual(3, len(prop.value)) |
| 536 | |
Simon Glass | ca04494 | 2021-07-28 19:23:10 -0600 | [diff] [blame] | 537 | # Widen an array of ints with an int (should do nothing) |
| 538 | prop = self.node.props['intarray'] |
Simon Glass | e679f39 | 2021-08-02 07:37:54 -0600 | [diff] [blame] | 539 | prop2 = node2.props['intval'] |
Simon Glass | ca04494 | 2021-07-28 19:23:10 -0600 | [diff] [blame] | 540 | self.assertEqual(Type.INT, prop.type) |
| 541 | self.assertEqual(3, len(prop.value)) |
| 542 | prop.Widen(prop2) |
| 543 | self.assertEqual(Type.INT, prop.type) |
| 544 | self.assertEqual(3, len(prop.value)) |
| 545 | |
Simon Glass | eec44c7 | 2021-07-28 19:23:11 -0600 | [diff] [blame] | 546 | # Widen an empty bool to an int |
| 547 | prop = self.node.props['maybe-empty-int'] |
| 548 | prop3 = node3.props['maybe-empty-int'] |
| 549 | self.assertEqual(Type.BOOL, prop.type) |
| 550 | self.assertEqual(True, prop.value) |
| 551 | self.assertEqual(Type.INT, prop3.type) |
| 552 | self.assertFalse(isinstance(prop.value, list)) |
| 553 | self.assertEqual(4, len(prop3.value)) |
| 554 | prop.Widen(prop3) |
| 555 | self.assertEqual(Type.INT, prop.type) |
| 556 | self.assertTrue(isinstance(prop.value, list)) |
| 557 | self.assertEqual(1, len(prop.value)) |
| 558 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 559 | def test_add(self): |
Simon Glass | 116adec | 2018-07-06 10:27:38 -0600 | [diff] [blame] | 560 | """Test adding properties""" |
| 561 | self.fdt.pack() |
| 562 | # This function should automatically expand the device tree |
| 563 | self.node.AddZeroProp('one') |
| 564 | self.node.AddZeroProp('two') |
| 565 | self.node.AddZeroProp('three') |
Simon Glass | fa80c25 | 2018-09-14 04:57:13 -0600 | [diff] [blame] | 566 | self.dtb.Sync(auto_resize=True) |
Simon Glass | 116adec | 2018-07-06 10:27:38 -0600 | [diff] [blame] | 567 | |
| 568 | # Updating existing properties should be OK, since the device-tree size |
| 569 | # does not change |
| 570 | self.fdt.pack() |
| 571 | self.node.SetInt('one', 1) |
| 572 | self.node.SetInt('two', 2) |
| 573 | self.node.SetInt('three', 3) |
Simon Glass | fa80c25 | 2018-09-14 04:57:13 -0600 | [diff] [blame] | 574 | self.dtb.Sync(auto_resize=False) |
Simon Glass | 116adec | 2018-07-06 10:27:38 -0600 | [diff] [blame] | 575 | |
| 576 | # This should fail since it would need to increase the device-tree size |
Simon Glass | fa80c25 | 2018-09-14 04:57:13 -0600 | [diff] [blame] | 577 | self.node.AddZeroProp('four') |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 578 | with self.assertRaises(libfdt.FdtException) as exc: |
Simon Glass | fa80c25 | 2018-09-14 04:57:13 -0600 | [diff] [blame] | 579 | self.dtb.Sync(auto_resize=False) |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 580 | self.assertIn('FDT_ERR_NOSPACE', str(exc.exception)) |
Simon Glass | 6434961 | 2018-09-14 04:57:16 -0600 | [diff] [blame] | 581 | self.dtb.Sync(auto_resize=True) |
Simon Glass | 116adec | 2018-07-06 10:27:38 -0600 | [diff] [blame] | 582 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 583 | def test_add_more(self): |
Simon Glass | 6434961 | 2018-09-14 04:57:16 -0600 | [diff] [blame] | 584 | """Test various other methods for adding and setting properties""" |
| 585 | self.node.AddZeroProp('one') |
| 586 | self.dtb.Sync(auto_resize=True) |
| 587 | data = self.fdt.getprop(self.node.Offset(), 'one') |
| 588 | self.assertEqual(0, fdt32_to_cpu(data)) |
| 589 | |
| 590 | self.node.SetInt('one', 1) |
| 591 | self.dtb.Sync(auto_resize=False) |
| 592 | data = self.fdt.getprop(self.node.Offset(), 'one') |
| 593 | self.assertEqual(1, fdt32_to_cpu(data)) |
| 594 | |
Simon Glass | 6eb9932 | 2021-01-06 21:35:18 -0700 | [diff] [blame] | 595 | val = 1234 |
| 596 | self.node.AddInt('integer', val) |
| 597 | self.dtb.Sync(auto_resize=True) |
| 598 | data = self.fdt.getprop(self.node.Offset(), 'integer') |
| 599 | self.assertEqual(val, fdt32_to_cpu(data)) |
| 600 | |
Simon Glass | 6434961 | 2018-09-14 04:57:16 -0600 | [diff] [blame] | 601 | val = '123' + chr(0) + '456' |
| 602 | self.node.AddString('string', val) |
| 603 | self.dtb.Sync(auto_resize=True) |
| 604 | data = self.fdt.getprop(self.node.Offset(), 'string') |
Simon Glass | c1aa66e | 2022-01-29 14:14:04 -0700 | [diff] [blame] | 605 | self.assertEqual(tools.to_bytes(val) + b'\0', data) |
Simon Glass | 6434961 | 2018-09-14 04:57:16 -0600 | [diff] [blame] | 606 | |
| 607 | self.fdt.pack() |
| 608 | self.node.SetString('string', val + 'x') |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 609 | with self.assertRaises(libfdt.FdtException) as exc: |
Simon Glass | 6434961 | 2018-09-14 04:57:16 -0600 | [diff] [blame] | 610 | self.dtb.Sync(auto_resize=False) |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 611 | self.assertIn('FDT_ERR_NOSPACE', str(exc.exception)) |
Simon Glass | 6434961 | 2018-09-14 04:57:16 -0600 | [diff] [blame] | 612 | self.node.SetString('string', val[:-1]) |
| 613 | |
| 614 | prop = self.node.props['string'] |
Simon Glass | c1aa66e | 2022-01-29 14:14:04 -0700 | [diff] [blame] | 615 | prop.SetData(tools.to_bytes(val)) |
Simon Glass | 6434961 | 2018-09-14 04:57:16 -0600 | [diff] [blame] | 616 | self.dtb.Sync(auto_resize=False) |
| 617 | data = self.fdt.getprop(self.node.Offset(), 'string') |
Simon Glass | c1aa66e | 2022-01-29 14:14:04 -0700 | [diff] [blame] | 618 | self.assertEqual(tools.to_bytes(val), data) |
Simon Glass | 6434961 | 2018-09-14 04:57:16 -0600 | [diff] [blame] | 619 | |
| 620 | self.node.AddEmptyProp('empty', 5) |
| 621 | self.dtb.Sync(auto_resize=True) |
| 622 | prop = self.node.props['empty'] |
Simon Glass | c1aa66e | 2022-01-29 14:14:04 -0700 | [diff] [blame] | 623 | prop.SetData(tools.to_bytes(val)) |
Simon Glass | 6434961 | 2018-09-14 04:57:16 -0600 | [diff] [blame] | 624 | self.dtb.Sync(auto_resize=False) |
| 625 | data = self.fdt.getprop(self.node.Offset(), 'empty') |
Simon Glass | c1aa66e | 2022-01-29 14:14:04 -0700 | [diff] [blame] | 626 | self.assertEqual(tools.to_bytes(val), data) |
Simon Glass | 6434961 | 2018-09-14 04:57:16 -0600 | [diff] [blame] | 627 | |
Simon Glass | f6b6481 | 2019-05-17 22:00:36 -0600 | [diff] [blame] | 628 | self.node.SetData('empty', b'123') |
| 629 | self.assertEqual(b'123', prop.bytes) |
Simon Glass | 6434961 | 2018-09-14 04:57:16 -0600 | [diff] [blame] | 630 | |
Simon Glass | c063917 | 2020-07-09 18:39:44 -0600 | [diff] [blame] | 631 | # Trying adding a lot of data at once |
Simon Glass | c1aa66e | 2022-01-29 14:14:04 -0700 | [diff] [blame] | 632 | self.node.AddData('data', tools.get_bytes(65, 20000)) |
Simon Glass | c063917 | 2020-07-09 18:39:44 -0600 | [diff] [blame] | 633 | self.dtb.Sync(auto_resize=True) |
| 634 | |
Simon Glass | bc11602 | 2022-02-08 11:49:50 -0700 | [diff] [blame] | 635 | def test_string_list(self): |
| 636 | """Test adding string-list property to a node""" |
| 637 | val = ['123', '456'] |
| 638 | self.node.AddStringList('stringlist', val) |
| 639 | self.dtb.Sync(auto_resize=True) |
| 640 | data = self.fdt.getprop(self.node.Offset(), 'stringlist') |
| 641 | self.assertEqual(b'123\x00456\0', data) |
| 642 | |
Simon Glass | 0ded4d4 | 2022-03-05 20:18:56 -0700 | [diff] [blame] | 643 | val = [] |
| 644 | self.node.AddStringList('stringlist', val) |
| 645 | self.dtb.Sync(auto_resize=True) |
| 646 | data = self.fdt.getprop(self.node.Offset(), 'stringlist') |
| 647 | self.assertEqual(b'', data) |
| 648 | |
Simon Glass | a30c39f | 2022-02-08 11:49:51 -0700 | [diff] [blame] | 649 | def test_delete_node(self): |
| 650 | """Test deleting a node""" |
| 651 | old_offset = self.fdt.path_offset('/spl-test') |
| 652 | self.assertGreater(old_offset, 0) |
| 653 | self.node.Delete() |
| 654 | self.dtb.Sync() |
| 655 | new_offset = self.fdt.path_offset('/spl-test', libfdt.QUIET_NOTFOUND) |
| 656 | self.assertEqual(-libfdt.NOTFOUND, new_offset) |
| 657 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 658 | def test_from_data(self): |
| 659 | """Test creating an FDT from data""" |
Simon Glass | 746aee3 | 2018-09-14 04:57:17 -0600 | [diff] [blame] | 660 | dtb2 = fdt.Fdt.FromData(self.dtb.GetContents()) |
| 661 | self.assertEqual(dtb2.GetContents(), self.dtb.GetContents()) |
| 662 | |
| 663 | self.node.AddEmptyProp('empty', 5) |
| 664 | self.dtb.Sync(auto_resize=True) |
| 665 | self.assertTrue(dtb2.GetContents() != self.dtb.GetContents()) |
| 666 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 667 | def test_missing_set_int(self): |
Simon Glass | d9dad10 | 2019-07-20 12:23:37 -0600 | [diff] [blame] | 668 | """Test handling of a missing property with SetInt""" |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 669 | with self.assertRaises(ValueError) as exc: |
Simon Glass | d9dad10 | 2019-07-20 12:23:37 -0600 | [diff] [blame] | 670 | self.node.SetInt('one', 1) |
| 671 | self.assertIn("node '/spl-test': Missing property 'one'", |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 672 | str(exc.exception)) |
Simon Glass | d9dad10 | 2019-07-20 12:23:37 -0600 | [diff] [blame] | 673 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 674 | def test_missing_set_data(self): |
Simon Glass | d9dad10 | 2019-07-20 12:23:37 -0600 | [diff] [blame] | 675 | """Test handling of a missing property with SetData""" |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 676 | with self.assertRaises(ValueError) as exc: |
Simon Glass | d9dad10 | 2019-07-20 12:23:37 -0600 | [diff] [blame] | 677 | self.node.SetData('one', b'data') |
| 678 | self.assertIn("node '/spl-test': Missing property 'one'", |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 679 | str(exc.exception)) |
Simon Glass | d9dad10 | 2019-07-20 12:23:37 -0600 | [diff] [blame] | 680 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 681 | def test_missing_set_string(self): |
Simon Glass | d9dad10 | 2019-07-20 12:23:37 -0600 | [diff] [blame] | 682 | """Test handling of a missing property with SetString""" |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 683 | with self.assertRaises(ValueError) as exc: |
Simon Glass | d9dad10 | 2019-07-20 12:23:37 -0600 | [diff] [blame] | 684 | self.node.SetString('one', 1) |
| 685 | self.assertIn("node '/spl-test': Missing property 'one'", |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 686 | str(exc.exception)) |
Simon Glass | d9dad10 | 2019-07-20 12:23:37 -0600 | [diff] [blame] | 687 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 688 | def test_get_filename(self): |
Simon Glass | f6e0249 | 2019-07-20 12:24:08 -0600 | [diff] [blame] | 689 | """Test the dtb filename can be provided""" |
Simon Glass | c1aa66e | 2022-01-29 14:14:04 -0700 | [diff] [blame] | 690 | self.assertEqual(tools.get_output_filename('source.dtb'), |
Simon Glass | f6e0249 | 2019-07-20 12:24:08 -0600 | [diff] [blame] | 691 | self.dtb.GetFilename()) |
| 692 | |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 693 | |
Simon Glass | 2a2d91d | 2018-07-06 10:27:28 -0600 | [diff] [blame] | 694 | class TestFdtUtil(unittest.TestCase): |
| 695 | """Tests for the fdt_util module |
| 696 | |
| 697 | This module will likely be mostly replaced at some point, once upstream |
| 698 | libfdt has better Python support. For now, this provides tests for current |
| 699 | functionality. |
| 700 | """ |
| 701 | @classmethod |
| 702 | def setUpClass(cls): |
Simon Glass | c1aa66e | 2022-01-29 14:14:04 -0700 | [diff] [blame] | 703 | tools.prepare_output_dir(None) |
Simon Glass | 2a2d91d | 2018-07-06 10:27:28 -0600 | [diff] [blame] | 704 | |
Simon Glass | e0e6275 | 2018-10-01 21:12:41 -0600 | [diff] [blame] | 705 | @classmethod |
| 706 | def tearDownClass(cls): |
Simon Glass | c1aa66e | 2022-01-29 14:14:04 -0700 | [diff] [blame] | 707 | tools.finalise_output_dir() |
Simon Glass | e0e6275 | 2018-10-01 21:12:41 -0600 | [diff] [blame] | 708 | |
Simon Glass | 2a2d91d | 2018-07-06 10:27:28 -0600 | [diff] [blame] | 709 | def setUp(self): |
Simon Glass | dff51a5 | 2021-02-03 06:00:56 -0700 | [diff] [blame] | 710 | self.dtb = fdt.FdtScan(find_dtb_file('dtoc_test_simple.dts')) |
Simon Glass | 2a2d91d | 2018-07-06 10:27:28 -0600 | [diff] [blame] | 711 | self.node = self.dtb.GetNode('/spl-test') |
| 712 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 713 | def test_get_int(self): |
| 714 | """Test getting an int from a node""" |
Simon Glass | 2a2d91d | 2018-07-06 10:27:28 -0600 | [diff] [blame] | 715 | self.assertEqual(1, fdt_util.GetInt(self.node, 'intval')) |
| 716 | self.assertEqual(3, fdt_util.GetInt(self.node, 'missing', 3)) |
| 717 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 718 | with self.assertRaises(ValueError) as exc: |
Simon Glass | d866e62 | 2021-11-23 11:03:39 -0700 | [diff] [blame] | 719 | fdt_util.GetInt(self.node, 'intarray') |
Simon Glass | 2a2d91d | 2018-07-06 10:27:28 -0600 | [diff] [blame] | 720 | self.assertIn("property 'intarray' has list value: expecting a single " |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 721 | 'integer', str(exc.exception)) |
Simon Glass | 2a2d91d | 2018-07-06 10:27:28 -0600 | [diff] [blame] | 722 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 723 | def test_get_int64(self): |
| 724 | """Test getting a 64-bit int from a node""" |
Simon Glass | d866e62 | 2021-11-23 11:03:39 -0700 | [diff] [blame] | 725 | self.assertEqual(0x123456789abcdef0, |
| 726 | fdt_util.GetInt64(self.node, 'int64val')) |
| 727 | self.assertEqual(3, fdt_util.GetInt64(self.node, 'missing', 3)) |
| 728 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 729 | with self.assertRaises(ValueError) as exc: |
Simon Glass | d866e62 | 2021-11-23 11:03:39 -0700 | [diff] [blame] | 730 | fdt_util.GetInt64(self.node, 'intarray') |
| 731 | self.assertIn( |
| 732 | "property 'intarray' should be a list with 2 items for 64-bit values", |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 733 | str(exc.exception)) |
Simon Glass | d866e62 | 2021-11-23 11:03:39 -0700 | [diff] [blame] | 734 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 735 | def test_get_string(self): |
| 736 | """Test getting a string from a node""" |
Simon Glass | 2a2d91d | 2018-07-06 10:27:28 -0600 | [diff] [blame] | 737 | self.assertEqual('message', fdt_util.GetString(self.node, 'stringval')) |
| 738 | self.assertEqual('test', fdt_util.GetString(self.node, 'missing', |
| 739 | 'test')) |
Simon Glass | 0ded4d4 | 2022-03-05 20:18:56 -0700 | [diff] [blame] | 740 | self.assertEqual('', fdt_util.GetString(self.node, 'boolval')) |
Simon Glass | 2a2d91d | 2018-07-06 10:27:28 -0600 | [diff] [blame] | 741 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 742 | with self.assertRaises(ValueError) as exc: |
Simon Glass | 2a2d91d | 2018-07-06 10:27:28 -0600 | [diff] [blame] | 743 | self.assertEqual(3, fdt_util.GetString(self.node, 'stringarray')) |
| 744 | self.assertIn("property 'stringarray' has list value: expecting a " |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 745 | 'single string', str(exc.exception)) |
Simon Glass | 2a2d91d | 2018-07-06 10:27:28 -0600 | [diff] [blame] | 746 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 747 | def test_get_string_list(self): |
| 748 | """Test getting a string list from a node""" |
Simon Glass | 1b5a533 | 2021-11-23 21:09:51 -0700 | [diff] [blame] | 749 | self.assertEqual(['message'], |
| 750 | fdt_util.GetStringList(self.node, 'stringval')) |
| 751 | self.assertEqual( |
| 752 | ['multi-word', 'message'], |
| 753 | fdt_util.GetStringList(self.node, 'stringarray')) |
| 754 | self.assertEqual(['test'], |
| 755 | fdt_util.GetStringList(self.node, 'missing', ['test'])) |
Simon Glass | 0ded4d4 | 2022-03-05 20:18:56 -0700 | [diff] [blame] | 756 | self.assertEqual([], fdt_util.GetStringList(self.node, 'boolval')) |
Simon Glass | 1b5a533 | 2021-11-23 21:09:51 -0700 | [diff] [blame] | 757 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 758 | def test_get_args(self): |
| 759 | """Test getting arguments from a node""" |
Simon Glass | 7e4b66a | 2022-02-08 11:49:53 -0700 | [diff] [blame] | 760 | node = self.dtb.GetNode('/orig-node') |
| 761 | self.assertEqual(['message'], fdt_util.GetArgs(self.node, 'stringval')) |
| 762 | self.assertEqual( |
| 763 | ['multi-word', 'message'], |
| 764 | fdt_util.GetArgs(self.node, 'stringarray')) |
| 765 | self.assertEqual([], fdt_util.GetArgs(self.node, 'boolval')) |
Simon Glass | 6101253 | 2022-03-05 20:18:52 -0700 | [diff] [blame] | 766 | self.assertEqual(['-n first', 'second', '-p', '123,456', '-x'], |
Simon Glass | 7e4b66a | 2022-02-08 11:49:53 -0700 | [diff] [blame] | 767 | fdt_util.GetArgs(node, 'args')) |
Simon Glass | 6101253 | 2022-03-05 20:18:52 -0700 | [diff] [blame] | 768 | self.assertEqual(['a space', 'there'], |
| 769 | fdt_util.GetArgs(node, 'args2')) |
| 770 | self.assertEqual(['-n', 'first', 'second', '-p', '123,456', '-x'], |
| 771 | fdt_util.GetArgs(node, 'args3')) |
Simon Glass | 7e4b66a | 2022-02-08 11:49:53 -0700 | [diff] [blame] | 772 | with self.assertRaises(ValueError) as exc: |
| 773 | fdt_util.GetArgs(self.node, 'missing') |
| 774 | self.assertIn( |
| 775 | "Node '/spl-test': Expected property 'missing'", |
| 776 | str(exc.exception)) |
| 777 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 778 | def test_get_bool(self): |
| 779 | """Test getting a bool from a node""" |
Simon Glass | 2a2d91d | 2018-07-06 10:27:28 -0600 | [diff] [blame] | 780 | self.assertEqual(True, fdt_util.GetBool(self.node, 'boolval')) |
| 781 | self.assertEqual(False, fdt_util.GetBool(self.node, 'missing')) |
| 782 | self.assertEqual(True, fdt_util.GetBool(self.node, 'missing', True)) |
| 783 | self.assertEqual(False, fdt_util.GetBool(self.node, 'missing', False)) |
| 784 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 785 | def test_get_byte(self): |
| 786 | """Test getting a byte from a node""" |
Simon Glass | 3af8e49 | 2018-07-17 13:25:40 -0600 | [diff] [blame] | 787 | self.assertEqual(5, fdt_util.GetByte(self.node, 'byteval')) |
| 788 | self.assertEqual(3, fdt_util.GetByte(self.node, 'missing', 3)) |
| 789 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 790 | with self.assertRaises(ValueError) as exc: |
Simon Glass | 3af8e49 | 2018-07-17 13:25:40 -0600 | [diff] [blame] | 791 | fdt_util.GetByte(self.node, 'longbytearray') |
| 792 | self.assertIn("property 'longbytearray' has list value: expecting a " |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 793 | 'single byte', str(exc.exception)) |
Simon Glass | 3af8e49 | 2018-07-17 13:25:40 -0600 | [diff] [blame] | 794 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 795 | with self.assertRaises(ValueError) as exc: |
Simon Glass | 3af8e49 | 2018-07-17 13:25:40 -0600 | [diff] [blame] | 796 | fdt_util.GetByte(self.node, 'intval') |
| 797 | self.assertIn("property 'intval' has length 4, expecting 1", |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 798 | str(exc.exception)) |
Simon Glass | 3af8e49 | 2018-07-17 13:25:40 -0600 | [diff] [blame] | 799 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 800 | def test_get_bytes(self): |
| 801 | """Test getting multiple bytes from a node""" |
Simon Glass | 40b4d64 | 2021-11-23 11:03:40 -0700 | [diff] [blame] | 802 | self.assertEqual(bytes([5]), fdt_util.GetBytes(self.node, 'byteval', 1)) |
| 803 | self.assertEqual(None, fdt_util.GetBytes(self.node, 'missing', 3)) |
| 804 | self.assertEqual( |
| 805 | bytes([3]), fdt_util.GetBytes(self.node, 'missing', 3, bytes([3]))) |
| 806 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 807 | with self.assertRaises(ValueError) as exc: |
Simon Glass | 40b4d64 | 2021-11-23 11:03:40 -0700 | [diff] [blame] | 808 | fdt_util.GetBytes(self.node, 'longbytearray', 7) |
| 809 | self.assertIn( |
| 810 | "Node 'spl-test' property 'longbytearray' has length 9, expecting 7", |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 811 | str(exc.exception)) |
Simon Glass | 40b4d64 | 2021-11-23 11:03:40 -0700 | [diff] [blame] | 812 | |
| 813 | self.assertEqual( |
| 814 | bytes([0, 0, 0, 1]), fdt_util.GetBytes(self.node, 'intval', 4)) |
| 815 | self.assertEqual( |
| 816 | bytes([3]), fdt_util.GetBytes(self.node, 'missing', 3, bytes([3]))) |
| 817 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 818 | def test_get_phandle_list(self): |
| 819 | """Test getting a list of phandles from a node""" |
Simon Glass | dff51a5 | 2021-02-03 06:00:56 -0700 | [diff] [blame] | 820 | dtb = fdt.FdtScan(find_dtb_file('dtoc_test_phandle.dts')) |
Simon Glass | 94a7c60 | 2018-07-17 13:25:46 -0600 | [diff] [blame] | 821 | node = dtb.GetNode('/phandle-source2') |
| 822 | self.assertEqual([1], fdt_util.GetPhandleList(node, 'clocks')) |
| 823 | node = dtb.GetNode('/phandle-source') |
| 824 | self.assertEqual([1, 2, 11, 3, 12, 13, 1], |
| 825 | fdt_util.GetPhandleList(node, 'clocks')) |
| 826 | self.assertEqual(None, fdt_util.GetPhandleList(node, 'missing')) |
| 827 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 828 | def test_get_data_type(self): |
| 829 | """Test getting a value of a particular type from a node""" |
Simon Glass | 53af22a | 2018-07-17 13:25:32 -0600 | [diff] [blame] | 830 | self.assertEqual(1, fdt_util.GetDatatype(self.node, 'intval', int)) |
| 831 | self.assertEqual('message', fdt_util.GetDatatype(self.node, 'stringval', |
| 832 | str)) |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 833 | with self.assertRaises(ValueError): |
Simon Glass | 53af22a | 2018-07-17 13:25:32 -0600 | [diff] [blame] | 834 | self.assertEqual(3, fdt_util.GetDatatype(self.node, 'boolval', |
| 835 | bool)) |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 836 | def test_fdt_cells_to_cpu(self): |
| 837 | """Test getting cells with the correct endianness""" |
Simon Glass | 2a2d91d | 2018-07-06 10:27:28 -0600 | [diff] [blame] | 838 | val = self.node.props['intarray'].value |
| 839 | self.assertEqual(0, fdt_util.fdt_cells_to_cpu(val, 0)) |
| 840 | self.assertEqual(2, fdt_util.fdt_cells_to_cpu(val, 1)) |
| 841 | |
Simon Glass | dff51a5 | 2021-02-03 06:00:56 -0700 | [diff] [blame] | 842 | dtb2 = fdt.FdtScan(find_dtb_file('dtoc_test_addr64.dts')) |
Simon Glass | e66d318 | 2019-05-17 22:00:40 -0600 | [diff] [blame] | 843 | node1 = dtb2.GetNode('/test1') |
| 844 | val = node1.props['reg'].value |
Simon Glass | 2a2d91d | 2018-07-06 10:27:28 -0600 | [diff] [blame] | 845 | self.assertEqual(0x1234, fdt_util.fdt_cells_to_cpu(val, 2)) |
| 846 | |
Simon Glass | e66d318 | 2019-05-17 22:00:40 -0600 | [diff] [blame] | 847 | node2 = dtb2.GetNode('/test2') |
| 848 | val = node2.props['reg'].value |
| 849 | self.assertEqual(0x1234567890123456, fdt_util.fdt_cells_to_cpu(val, 2)) |
| 850 | self.assertEqual(0x9876543210987654, fdt_util.fdt_cells_to_cpu(val[2:], |
| 851 | 2)) |
| 852 | self.assertEqual(0x12345678, fdt_util.fdt_cells_to_cpu(val, 1)) |
| 853 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 854 | def test_ensure_compiled(self): |
Simon Glass | a004f29 | 2019-07-20 12:23:49 -0600 | [diff] [blame] | 855 | """Test a degenerate case of this function (file already compiled)""" |
Simon Glass | dff51a5 | 2021-02-03 06:00:56 -0700 | [diff] [blame] | 856 | dtb = fdt_util.EnsureCompiled(find_dtb_file('dtoc_test_simple.dts')) |
Simon Glass | 2a2d91d | 2018-07-06 10:27:28 -0600 | [diff] [blame] | 857 | self.assertEqual(dtb, fdt_util.EnsureCompiled(dtb)) |
| 858 | |
Simon Glass | 5d1637a | 2022-07-30 20:57:10 -0600 | [diff] [blame] | 859 | def test_ensure_compiled_tmpdir(self): |
Simon Glass | a004f29 | 2019-07-20 12:23:49 -0600 | [diff] [blame] | 860 | """Test providing a temporary directory""" |
Heinrich Schuchardt | d0d0e77 | 2023-04-20 20:03:43 +0200 | [diff] [blame] | 861 | old_outdir = tools.outdir |
Simon Glass | a004f29 | 2019-07-20 12:23:49 -0600 | [diff] [blame] | 862 | try: |
Simon Glass | a004f29 | 2019-07-20 12:23:49 -0600 | [diff] [blame] | 863 | tools.outdir= None |
| 864 | tmpdir = tempfile.mkdtemp(prefix='test_fdt.') |
Simon Glass | dff51a5 | 2021-02-03 06:00:56 -0700 | [diff] [blame] | 865 | dtb = fdt_util.EnsureCompiled(find_dtb_file('dtoc_test_simple.dts'), |
Simon Glass | a004f29 | 2019-07-20 12:23:49 -0600 | [diff] [blame] | 866 | tmpdir) |
| 867 | self.assertEqual(tmpdir, os.path.dirname(dtb)) |
| 868 | shutil.rmtree(tmpdir) |
| 869 | finally: |
Heinrich Schuchardt | d0d0e77 | 2023-04-20 20:03:43 +0200 | [diff] [blame] | 870 | tools.outdir = old_outdir |
Simon Glass | a004f29 | 2019-07-20 12:23:49 -0600 | [diff] [blame] | 871 | |
Simon Glass | 8f5afe2 | 2023-01-11 16:10:18 -0700 | [diff] [blame] | 872 | def test_get_phandle_name_offset(self): |
| 873 | val = fdt_util.GetPhandleNameOffset(self.node, 'missing') |
| 874 | self.assertIsNone(val) |
| 875 | |
| 876 | dtb = fdt.FdtScan(find_dtb_file('dtoc_test_phandle.dts')) |
| 877 | node = dtb.GetNode('/phandle-source') |
| 878 | node, name, offset = fdt_util.GetPhandleNameOffset(node, |
| 879 | 'phandle-name-offset') |
| 880 | self.assertEqual('phandle3-target', node.name) |
| 881 | self.assertEqual('fred', name) |
| 882 | self.assertEqual(123, offset) |
Simon Glass | 2a2d91d | 2018-07-06 10:27:28 -0600 | [diff] [blame] | 883 | |
Simon Glass | 2598079 | 2022-07-30 20:57:05 -0600 | [diff] [blame] | 884 | def run_test_coverage(build_dir): |
| 885 | """Run the tests and check that we get 100% coverage |
| 886 | |
| 887 | Args: |
| 888 | build_dir (str): Directory containing the build output |
| 889 | """ |
Simon Glass | 5e2ab40 | 2022-01-29 14:14:14 -0700 | [diff] [blame] | 890 | test_util.run_test_coverage('tools/dtoc/test_fdt.py', None, |
Simon Glass | 4583c00 | 2023-02-23 18:18:04 -0700 | [diff] [blame] | 891 | ['tools/patman/*.py', 'tools/u_boot_pylib/*', '*test_fdt.py'], |
| 892 | build_dir) |
Simon Glass | 2a2d91d | 2018-07-06 10:27:28 -0600 | [diff] [blame] | 893 | |
| 894 | |
Simon Glass | 7640b16 | 2022-07-30 20:57:09 -0600 | [diff] [blame] | 895 | def run_tests(names, processes): |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 896 | """Run all the test we have for the fdt model |
| 897 | |
| 898 | Args: |
Simon Glass | 7640b16 | 2022-07-30 20:57:09 -0600 | [diff] [blame] | 899 | names (list of str): List of test names provided. Only the first is used |
Simon Glass | b26dd96 | 2022-07-30 20:57:06 -0600 | [diff] [blame] | 900 | processes (int): Number of processes to use (None means as many as there |
| 901 | are CPUs on the system. This must be set to 1 when running under |
| 902 | the python3-coverage tool |
Simon Glass | 42ae363 | 2022-03-18 18:01:50 -0600 | [diff] [blame] | 903 | |
| 904 | Returns: |
Simon Glass | b26dd96 | 2022-07-30 20:57:06 -0600 | [diff] [blame] | 905 | int: Return code, 0 on success |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 906 | """ |
Simon Glass | 7640b16 | 2022-07-30 20:57:09 -0600 | [diff] [blame] | 907 | test_name = names[0] if names else None |
Alper Nebi Yasak | d8318fe | 2022-04-02 20:06:06 +0300 | [diff] [blame] | 908 | result = test_util.run_test_suites( |
Simon Glass | ad74422 | 2022-07-30 20:57:07 -0600 | [diff] [blame] | 909 | 'test_fdt', False, False, False, processes, test_name, None, |
Simon Glass | 42ae363 | 2022-03-18 18:01:50 -0600 | [diff] [blame] | 910 | [TestFdt, TestNode, TestProp, TestFdtUtil]) |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 911 | |
Alper Nebi Yasak | d8318fe | 2022-04-02 20:06:06 +0300 | [diff] [blame] | 912 | return (0 if result.wasSuccessful() else 1) |
| 913 | |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 914 | |
Simon Glass | a8ad9aa | 2022-07-30 20:57:08 -0600 | [diff] [blame] | 915 | def main(): |
| 916 | """Main program for this tool""" |
Simon Glass | 7640b16 | 2022-07-30 20:57:09 -0600 | [diff] [blame] | 917 | parser = ArgumentParser() |
| 918 | parser.add_argument('-B', '--build-dir', type=str, default='b', |
| 919 | help='Directory containing the build output') |
| 920 | parser.add_argument('-P', '--processes', type=int, |
| 921 | help='set number of processes to use for running tests') |
| 922 | parser.add_argument('-t', '--test', action='store_true', dest='test', |
| 923 | default=False, help='run tests') |
| 924 | parser.add_argument('-T', '--test-coverage', action='store_true', |
| 925 | default=False, |
| 926 | help='run tests and check for 100% coverage') |
| 927 | parser.add_argument('name', nargs='*') |
| 928 | args = parser.parse_args() |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 929 | |
Simon Glass | a8ad9aa | 2022-07-30 20:57:08 -0600 | [diff] [blame] | 930 | # Run our meagre tests |
Simon Glass | 7640b16 | 2022-07-30 20:57:09 -0600 | [diff] [blame] | 931 | if args.test: |
| 932 | ret_code = run_tests(args.name, args.processes) |
Simon Glass | a8ad9aa | 2022-07-30 20:57:08 -0600 | [diff] [blame] | 933 | return ret_code |
Simon Glass | 7640b16 | 2022-07-30 20:57:09 -0600 | [diff] [blame] | 934 | if args.test_coverage: |
| 935 | run_test_coverage(args.build_dir) |
Simon Glass | a8ad9aa | 2022-07-30 20:57:08 -0600 | [diff] [blame] | 936 | return 0 |
Simon Glass | 2ba9875 | 2018-07-06 10:27:24 -0600 | [diff] [blame] | 937 | |
Simon Glass | a8ad9aa | 2022-07-30 20:57:08 -0600 | [diff] [blame] | 938 | if __name__ == '__main__': |
| 939 | sys.exit(main()) |