]>
Commit | Line | Data |
---|---|---|
bf7fd50b SG |
1 | # |
2 | # Copyright (c) 2016 Google, Inc | |
3 | # Written by Simon Glass <[email protected]> | |
4 | # | |
5 | # SPDX-License-Identifier: GPL-2.0+ | |
6 | # | |
7 | # Test for the fdt modules | |
8 | ||
9 | import os | |
10 | import sys | |
11 | import tempfile | |
12 | import unittest | |
13 | ||
4a28b007 | 14 | import fdt |
99ed4a2e | 15 | from fdt import FdtScan |
bf7fd50b SG |
16 | import fdt_util |
17 | import tools | |
18 | ||
19 | class TestFdt(unittest.TestCase): | |
20 | @classmethod | |
21 | def setUpClass(self): | |
22 | self._binman_dir = os.path.dirname(os.path.realpath(sys.argv[0])) | |
23 | self._indir = tempfile.mkdtemp(prefix='binmant.') | |
24 | tools.PrepareOutputDir(self._indir, True) | |
25 | ||
26 | def TestFile(self, fname): | |
27 | return os.path.join(self._binman_dir, 'test', fname) | |
28 | ||
29 | def GetCompiled(self, fname): | |
30 | return fdt_util.EnsureCompiled(self.TestFile(fname)) | |
31 | ||
1d88ebb2 SG |
32 | def _DeleteProp(self, dt): |
33 | node = dt.GetNode('/microcode/update@0') | |
bf7fd50b SG |
34 | node.DeleteProp('data') |
35 | ||
36 | def testFdtNormal(self): | |
37 | fname = self.GetCompiled('34_x86_ucode.dts') | |
1d88ebb2 SG |
38 | dt = FdtScan(fname) |
39 | self._DeleteProp(dt) | |
bf7fd50b | 40 | |
4a28b007 SG |
41 | def testFdtNormalProp(self): |
42 | fname = self.GetCompiled('45_prop_test.dts') | |
43 | dt = FdtScan(fname) | |
44 | node = dt.GetNode('/binman/intel-me') | |
45 | self.assertEquals('intel-me', node.name) | |
46 | val = fdt_util.GetString(node, 'filename') | |
47 | self.assertEquals(str, type(val)) | |
48 | self.assertEquals('me.bin', val) | |
49 | ||
50 | prop = node.props['intval'] | |
51 | self.assertEquals(fdt.TYPE_INT, prop.type) | |
52 | self.assertEquals(3, fdt_util.GetInt(node, 'intval')) | |
53 | ||
54 | prop = node.props['intarray'] | |
55 | self.assertEquals(fdt.TYPE_INT, prop.type) | |
56 | self.assertEquals(list, type(prop.value)) | |
57 | self.assertEquals(2, len(prop.value)) | |
58 | self.assertEquals([5, 6], | |
59 | [fdt_util.fdt32_to_cpu(val) for val in prop.value]) | |
60 | ||
61 | prop = node.props['byteval'] | |
62 | self.assertEquals(fdt.TYPE_BYTE, prop.type) | |
63 | self.assertEquals(chr(8), prop.value) | |
64 | ||
65 | prop = node.props['bytearray'] | |
66 | self.assertEquals(fdt.TYPE_BYTE, prop.type) | |
67 | self.assertEquals(list, type(prop.value)) | |
68 | self.assertEquals(str, type(prop.value[0])) | |
69 | self.assertEquals(3, len(prop.value)) | |
70 | self.assertEquals([chr(1), '#', '4'], prop.value) | |
71 | ||
72 | prop = node.props['longbytearray'] | |
73 | self.assertEquals(fdt.TYPE_INT, prop.type) | |
74 | self.assertEquals(0x090a0b0c, fdt_util.GetInt(node, 'longbytearray')) | |
75 | ||
76 | prop = node.props['stringval'] | |
77 | self.assertEquals(fdt.TYPE_STRING, prop.type) | |
78 | self.assertEquals('message2', fdt_util.GetString(node, 'stringval')) | |
79 | ||
80 | prop = node.props['stringarray'] | |
81 | self.assertEquals(fdt.TYPE_STRING, prop.type) | |
82 | self.assertEquals(list, type(prop.value)) | |
83 | self.assertEquals(3, len(prop.value)) | |
84 | self.assertEquals(['another', 'multi-word', 'message'], prop.value) |