fdt_test.py 2.9 KB

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