| #!/usr/bin/env python3 | |
| # SPDX-License-Identifier: GPL-2.0+ | |
| # Copyright (c) 2018 Google, Inc | |
| # Written by Simon Glass <sjg@chromium.org> | |
| # | |
| from __future__ import print_function | |
| from optparse import OptionParser | |
| import glob | |
| import os | |
| import shutil | |
| import sys | |
| import tempfile | |
| import unittest | |
| # Bring in the patman libraries | |
| our_path = os.path.dirname(os.path.realpath(__file__)) | |
| for dirname in ['../patman', '..']: | |
| sys.path.insert(0, os.path.join(our_path, dirname)) | |
| import command | |
| import fdt | |
| from fdt import TYPE_BYTE, TYPE_INT, TYPE_STRING, TYPE_BOOL, BytesToValue | |
| import fdt_util | |
| from fdt_util import fdt32_to_cpu | |
| import libfdt | |
| import test_util | |
| import tools | |
| def _GetPropertyValue(dtb, node, prop_name): | |
| """Low-level function to get the property value based on its offset | |
| This looks directly in the device tree at the property's offset to find | |
| its value. It is useful as a check that the property is in the correct | |
| place. | |
| Args: | |
| node: Node to look in | |
| prop_name: Property name to find | |
| Returns: | |
| Tuple: | |
| Prop object found | |
| Value of property as a string (found using property offset) | |
| """ | |
| prop = node.props[prop_name] | |
| # Add 12, which is sizeof(struct fdt_property), to get to start of data | |
| offset = prop.GetOffset() + 12 | |
| data = dtb.GetContents()[offset:offset + len(prop.value)] | |
| return prop, [tools.ToChar(x) for x in data] | |
| class TestFdt(unittest.TestCase): | |
| """Tests for the Fdt module | |
| This includes unit tests for some functions and functional tests for the fdt | |
| module. | |
| """ | |
| @classmethod | |
| def setUpClass(cls): | |
| tools.PrepareOutputDir(None) | |
| @classmethod | |
| def tearDownClass(cls): | |
| tools.FinaliseOutputDir() | |
| def setUp(self): | |
| self.dtb = fdt.FdtScan('tools/dtoc/dtoc_test_simple.dts') | |
| def testFdt(self): | |
| """Test that we can open an Fdt""" | |
| self.dtb.Scan() | |
| root = self.dtb.GetRoot() | |
| self.assertTrue(isinstance(root, fdt.Node)) | |
| def testGetNode(self): | |
| """Test the GetNode() method""" | |
| node = self.dtb.GetNode('/spl-test') | |
| self.assertTrue(isinstance(node, fdt.Node)) | |
| node = self.dtb.GetNode('/i2c@0/pmic@9') | |
| self.assertTrue(isinstance(node, fdt.Node)) | |
| self.assertEqual('pmic@9', node.name) | |
| self.assertIsNone(self.dtb.GetNode('/i2c@0/pmic@9/missing')) | |
| node = self.dtb.GetNode('/') | |
| self.assertTrue(isinstance(node, fdt.Node)) | |
| self.assertEqual(0, node.Offset()) | |
| def testFlush(self): | |
| """Check that we can flush the device tree out to its file""" | |
| fname = self.dtb._fname | |
| with open(fname, 'rb') as fd: | |
| data = fd.read() | |
| os.remove(fname) | |
| with self.assertRaises(IOError): | |
| open(fname, 'rb') | |
| self.dtb.Flush() | |
| with open(fname, 'rb') as fd: | |
| data = fd.read() | |
| def testPack(self): | |
| """Test that packing a device tree works""" | |
| self.dtb.Pack() | |
| def testGetFdt(self): | |
| """Tetst that we can access the raw device-tree data""" | |
| self.assertTrue(isinstance(self.dtb.GetContents(), bytearray)) | |
| def testGetProps(self): | |
| """Tests obtaining a list of properties""" | |
| node = self.dtb.GetNode('/spl-test') | |
| props = self.dtb.GetProps(node) | |
| self.assertEqual(['boolval', 'bytearray', 'byteval', 'compatible', | |
| 'intarray', 'intval', 'longbytearray', 'notstring', | |
| 'stringarray', 'stringval', 'u-boot,dm-pre-reloc'], | |
| sorted(props.keys())) | |
| def testCheckError(self): | |
| """Tests the ChecKError() function""" | |
| with self.assertRaises(ValueError) as e: | |
| fdt.CheckErr(-libfdt.NOTFOUND, 'hello') | |
| self.assertIn('FDT_ERR_NOTFOUND: hello', str(e.exception)) | |
| def testGetFdt(self): | |
| node = self.dtb.GetNode('/spl-test') | |
| self.assertEqual(self.dtb, node.GetFdt()) | |
| def testBytesToValue(self): | |
| self.assertEqual(BytesToValue(b'this\0is\0'), | |
| (TYPE_STRING, ['this', 'is'])) | |
| class TestNode(unittest.TestCase): | |
| """Test operation of the Node class""" | |
| @classmethod | |
| def setUpClass(cls): | |
| tools.PrepareOutputDir(None) | |
| @classmethod | |
| def tearDownClass(cls): | |
| tools.FinaliseOutputDir() | |
| def setUp(self): | |
| self.dtb = fdt.FdtScan('tools/dtoc/dtoc_test_simple.dts') | |
| self.node = self.dtb.GetNode('/spl-test') | |
| def testOffset(self): | |
| """Tests that we can obtain the offset of a node""" | |
| self.assertTrue(self.node.Offset() > 0) | |
| def testDelete(self): | |
| """Tests that we can delete a property""" | |
| node2 = self.dtb.GetNode('/spl-test2') | |
| offset1 = node2.Offset() | |
| self.node.DeleteProp('intval') | |
| offset2 = node2.Offset() | |
| self.assertTrue(offset2 < offset1) | |
| self.node.DeleteProp('intarray') | |
| offset3 = node2.Offset() | |
| self.assertTrue(offset3 < offset2) | |
| with self.assertRaises(libfdt.FdtException): | |
| self.node.DeleteProp('missing') | |
| def testDeleteGetOffset(self): | |
| """Test that property offset update when properties are deleted""" | |
| self.node.DeleteProp('intval') | |
| prop, value = _GetPropertyValue(self.dtb, self.node, 'longbytearray') | |
| self.assertEqual(prop.value, value) | |
| def testFindNode(self): | |
| """Tests that we can find a node using the FindNode() functoin""" | |
| node = self.dtb.GetRoot().FindNode('i2c@0') | |
| self.assertEqual('i2c@0', node.name) | |
| subnode = node.FindNode('pmic@9') | |
| self.assertEqual('pmic@9', subnode.name) | |
| self.assertEqual(None, node.FindNode('missing')) | |
| def testRefreshMissingNode(self): | |
| """Test refreshing offsets when an extra node is present in dtb""" | |
| # Delete it from our tables, not the device tree | |
| del self.dtb._root.subnodes[-1] | |
| with self.assertRaises(ValueError) as e: | |
| self.dtb.Refresh() | |
| self.assertIn('Internal error, offset', str(e.exception)) | |
| def testRefreshExtraNode(self): | |
| """Test refreshing offsets when an expected node is missing""" | |
| # Delete it from the device tre, not our tables | |
| self.dtb.GetFdtObj().del_node(self.node.Offset()) | |
| with self.assertRaises(ValueError) as e: | |
| self.dtb.Refresh() | |
| self.assertIn('Internal error, node name mismatch ' | |
| 'spl-test != spl-test2', str(e.exception)) | |
| def testRefreshMissingProp(self): | |
| """Test refreshing offsets when an extra property is present in dtb""" | |
| # Delete it from our tables, not the device tree | |
| del self.node.props['notstring'] | |
| with self.assertRaises(ValueError) as e: | |
| self.dtb.Refresh() | |
| self.assertIn("Internal error, property 'notstring' missing, offset ", | |
| str(e.exception)) | |
| def testLookupPhandle(self): | |
| """Test looking up a single phandle""" | |
| dtb = fdt.FdtScan('tools/dtoc/dtoc_test_phandle.dts') | |
| node = dtb.GetNode('/phandle-source2') | |
| prop = node.props['clocks'] | |
| target = dtb.GetNode('/phandle-target') | |
| self.assertEqual(target, dtb.LookupPhandle(fdt32_to_cpu(prop.value))) | |
| class TestProp(unittest.TestCase): | |
| """Test operation of the Prop class""" | |
| @classmethod | |
| def setUpClass(cls): | |
| tools.PrepareOutputDir(None) | |
| @classmethod | |
| def tearDownClass(cls): | |
| tools.FinaliseOutputDir() | |
| def setUp(self): | |
| self.dtb = fdt.FdtScan('tools/dtoc/dtoc_test_simple.dts') | |
| self.node = self.dtb.GetNode('/spl-test') | |
| self.fdt = self.dtb.GetFdtObj() | |
| def testMissingNode(self): | |
| self.assertEqual(None, self.dtb.GetNode('missing')) | |
| def testPhandle(self): | |
| dtb = fdt.FdtScan('tools/dtoc/dtoc_test_phandle.dts') | |
| node = dtb.GetNode('/phandle-source2') | |
| prop = node.props['clocks'] | |
| self.assertTrue(fdt32_to_cpu(prop.value) > 0) | |
| def _ConvertProp(self, prop_name): | |
| """Helper function to look up a property in self.node and return it | |
| Args: | |
| Property name to find | |
| Return fdt.Prop object for this property | |
| """ | |
| p = self.fdt.getprop(self.node.Offset(), prop_name) | |
| return fdt.Prop(self.node, -1, prop_name, p) | |
| def testMakeProp(self): | |
| """Test we can convert all the the types that are supported""" | |
| prop = self._ConvertProp('boolval') | |
| self.assertEqual(fdt.TYPE_BOOL, prop.type) | |
| self.assertEqual(True, prop.value) | |
| prop = self._ConvertProp('intval') | |
| self.assertEqual(fdt.TYPE_INT, prop.type) | |
| self.assertEqual(1, fdt32_to_cpu(prop.value)) | |
| prop = self._ConvertProp('intarray') | |
| self.assertEqual(fdt.TYPE_INT, prop.type) | |
| val = [fdt32_to_cpu(val) for val in prop.value] | |
| self.assertEqual([2, 3, 4], val) | |
| prop = self._ConvertProp('byteval') | |
| self.assertEqual(fdt.TYPE_BYTE, prop.type) | |
| self.assertEqual(5, ord(prop.value)) | |
| prop = self._ConvertProp('longbytearray') | |
| self.assertEqual(fdt.TYPE_BYTE, prop.type) | |
| val = [ord(val) for val in prop.value] | |
| self.assertEqual([9, 10, 11, 12, 13, 14, 15, 16, 17], val) | |
| prop = self._ConvertProp('stringval') | |
| self.assertEqual(fdt.TYPE_STRING, prop.type) | |
| self.assertEqual('message', prop.value) | |
| prop = self._ConvertProp('stringarray') | |
| self.assertEqual(fdt.TYPE_STRING, prop.type) | |
| self.assertEqual(['multi-word', 'message'], prop.value) | |
| prop = self._ConvertProp('notstring') | |
| self.assertEqual(fdt.TYPE_BYTE, prop.type) | |
| val = [ord(val) for val in prop.value] | |
| self.assertEqual([0x20, 0x21, 0x22, 0x10, 0], val) | |
| def testGetEmpty(self): | |
| """Tests the GetEmpty() function for the various supported types""" | |
| self.assertEqual(True, fdt.Prop.GetEmpty(fdt.TYPE_BOOL)) | |
| self.assertEqual(chr(0), fdt.Prop.GetEmpty(fdt.TYPE_BYTE)) | |
| self.assertEqual(tools.GetBytes(0, 4), fdt.Prop.GetEmpty(fdt.TYPE_INT)) | |
| self.assertEqual('', fdt.Prop.GetEmpty(fdt.TYPE_STRING)) | |
| def testGetOffset(self): | |
| """Test we can get the offset of a property""" | |
| prop, value = _GetPropertyValue(self.dtb, self.node, 'longbytearray') | |
| self.assertEqual(prop.value, value) | |
| def testWiden(self): | |
| """Test widening of values""" | |
| node2 = self.dtb.GetNode('/spl-test2') | |
| prop = self.node.props['intval'] | |
| # No action | |
| prop2 = node2.props['intval'] | |
| prop.Widen(prop2) | |
| self.assertEqual(fdt.TYPE_INT, prop.type) | |
| self.assertEqual(1, fdt32_to_cpu(prop.value)) | |
| # Convert singla value to array | |
| prop2 = self.node.props['intarray'] | |
| prop.Widen(prop2) | |
| self.assertEqual(fdt.TYPE_INT, prop.type) | |
| self.assertTrue(isinstance(prop.value, list)) | |
| # A 4-byte array looks like a single integer. When widened by a longer | |
| # byte array, it should turn into an array. | |
| prop = self.node.props['longbytearray'] | |
| prop2 = node2.props['longbytearray'] | |
| self.assertFalse(isinstance(prop2.value, list)) | |
| self.assertEqual(4, len(prop2.value)) | |
| prop2.Widen(prop) | |
| self.assertTrue(isinstance(prop2.value, list)) | |
| self.assertEqual(9, len(prop2.value)) | |
| # Similarly for a string array | |
| prop = self.node.props['stringval'] | |
| prop2 = node2.props['stringarray'] | |
| self.assertFalse(isinstance(prop.value, list)) | |
| self.assertEqual(7, len(prop.value)) | |
| prop.Widen(prop2) | |
| self.assertTrue(isinstance(prop.value, list)) | |
| self.assertEqual(3, len(prop.value)) | |
| # Enlarging an existing array | |
| prop = self.node.props['stringarray'] | |
| prop2 = node2.props['stringarray'] | |
| self.assertTrue(isinstance(prop.value, list)) | |
| self.assertEqual(2, len(prop.value)) | |
| prop.Widen(prop2) | |
| self.assertTrue(isinstance(prop.value, list)) | |
| self.assertEqual(3, len(prop.value)) | |
| def testAdd(self): | |
| """Test adding properties""" | |
| self.fdt.pack() | |
| # This function should automatically expand the device tree | |
| self.node.AddZeroProp('one') | |
| self.node.AddZeroProp('two') | |
| self.node.AddZeroProp('three') | |
| self.dtb.Sync(auto_resize=True) | |
| # Updating existing properties should be OK, since the device-tree size | |
| # does not change | |
| self.fdt.pack() | |
| self.node.SetInt('one', 1) | |
| self.node.SetInt('two', 2) | |
| self.node.SetInt('three', 3) | |
| self.dtb.Sync(auto_resize=False) | |
| # This should fail since it would need to increase the device-tree size | |
| self.node.AddZeroProp('four') | |
| with self.assertRaises(libfdt.FdtException) as e: | |
| self.dtb.Sync(auto_resize=False) | |
| self.assertIn('FDT_ERR_NOSPACE', str(e.exception)) | |
| self.dtb.Sync(auto_resize=True) | |
| def testAddNode(self): | |
| self.fdt.pack() | |
| self.node.AddSubnode('subnode') | |
| with self.assertRaises(libfdt.FdtException) as e: | |
| self.dtb.Sync(auto_resize=False) | |
| self.assertIn('FDT_ERR_NOSPACE', str(e.exception)) | |
| self.dtb.Sync(auto_resize=True) | |
| offset = self.fdt.path_offset('/spl-test/subnode') | |
| self.assertTrue(offset > 0) | |
| def testAddMore(self): | |
| """Test various other methods for adding and setting properties""" | |
| self.node.AddZeroProp('one') | |
| self.dtb.Sync(auto_resize=True) | |
| data = self.fdt.getprop(self.node.Offset(), 'one') | |
| self.assertEqual(0, fdt32_to_cpu(data)) | |
| self.node.SetInt('one', 1) | |
| self.dtb.Sync(auto_resize=False) | |
| data = self.fdt.getprop(self.node.Offset(), 'one') | |
| self.assertEqual(1, fdt32_to_cpu(data)) | |
| val = '123' + chr(0) + '456' | |
| self.node.AddString('string', val) | |
| self.dtb.Sync(auto_resize=True) | |
| data = self.fdt.getprop(self.node.Offset(), 'string') | |
| self.assertEqual(tools.ToBytes(val) + b'\0', data) | |
| self.fdt.pack() | |
| self.node.SetString('string', val + 'x') | |
| with self.assertRaises(libfdt.FdtException) as e: | |
| self.dtb.Sync(auto_resize=False) | |
| self.assertIn('FDT_ERR_NOSPACE', str(e.exception)) | |
| self.node.SetString('string', val[:-1]) | |
| prop = self.node.props['string'] | |
| prop.SetData(tools.ToBytes(val)) | |
| self.dtb.Sync(auto_resize=False) | |
| data = self.fdt.getprop(self.node.Offset(), 'string') | |
| self.assertEqual(tools.ToBytes(val), data) | |
| self.node.AddEmptyProp('empty', 5) | |
| self.dtb.Sync(auto_resize=True) | |
| prop = self.node.props['empty'] | |
| prop.SetData(tools.ToBytes(val)) | |
| self.dtb.Sync(auto_resize=False) | |
| data = self.fdt.getprop(self.node.Offset(), 'empty') | |
| self.assertEqual(tools.ToBytes(val), data) | |
| self.node.SetData('empty', b'123') | |
| self.assertEqual(b'123', prop.bytes) | |
| def testFromData(self): | |
| dtb2 = fdt.Fdt.FromData(self.dtb.GetContents()) | |
| self.assertEqual(dtb2.GetContents(), self.dtb.GetContents()) | |
| self.node.AddEmptyProp('empty', 5) | |
| self.dtb.Sync(auto_resize=True) | |
| self.assertTrue(dtb2.GetContents() != self.dtb.GetContents()) | |
| def testMissingSetInt(self): | |
| """Test handling of a missing property with SetInt""" | |
| with self.assertRaises(ValueError) as e: | |
| self.node.SetInt('one', 1) | |
| self.assertIn("node '/spl-test': Missing property 'one'", | |
| str(e.exception)) | |
| def testMissingSetData(self): | |
| """Test handling of a missing property with SetData""" | |
| with self.assertRaises(ValueError) as e: | |
| self.node.SetData('one', b'data') | |
| self.assertIn("node '/spl-test': Missing property 'one'", | |
| str(e.exception)) | |
| def testMissingSetString(self): | |
| """Test handling of a missing property with SetString""" | |
| with self.assertRaises(ValueError) as e: | |
| self.node.SetString('one', 1) | |
| self.assertIn("node '/spl-test': Missing property 'one'", | |
| str(e.exception)) | |
| def testGetFilename(self): | |
| """Test the dtb filename can be provided""" | |
| self.assertEqual(tools.GetOutputFilename('source.dtb'), | |
| self.dtb.GetFilename()) | |
| class TestFdtUtil(unittest.TestCase): | |
| """Tests for the fdt_util module | |
| This module will likely be mostly replaced at some point, once upstream | |
| libfdt has better Python support. For now, this provides tests for current | |
| functionality. | |
| """ | |
| @classmethod | |
| def setUpClass(cls): | |
| tools.PrepareOutputDir(None) | |
| @classmethod | |
| def tearDownClass(cls): | |
| tools.FinaliseOutputDir() | |
| def setUp(self): | |
| self.dtb = fdt.FdtScan('tools/dtoc/dtoc_test_simple.dts') | |
| self.node = self.dtb.GetNode('/spl-test') | |
| def testGetInt(self): | |
| self.assertEqual(1, fdt_util.GetInt(self.node, 'intval')) | |
| self.assertEqual(3, fdt_util.GetInt(self.node, 'missing', 3)) | |
| with self.assertRaises(ValueError) as e: | |
| self.assertEqual(3, fdt_util.GetInt(self.node, 'intarray')) | |
| self.assertIn("property 'intarray' has list value: expecting a single " | |
| 'integer', str(e.exception)) | |
| def testGetString(self): | |
| self.assertEqual('message', fdt_util.GetString(self.node, 'stringval')) | |
| self.assertEqual('test', fdt_util.GetString(self.node, 'missing', | |
| 'test')) | |
| with self.assertRaises(ValueError) as e: | |
| self.assertEqual(3, fdt_util.GetString(self.node, 'stringarray')) | |
| self.assertIn("property 'stringarray' has list value: expecting a " | |
| 'single string', str(e.exception)) | |
| def testGetBool(self): | |
| self.assertEqual(True, fdt_util.GetBool(self.node, 'boolval')) | |
| self.assertEqual(False, fdt_util.GetBool(self.node, 'missing')) | |
| self.assertEqual(True, fdt_util.GetBool(self.node, 'missing', True)) | |
| self.assertEqual(False, fdt_util.GetBool(self.node, 'missing', False)) | |
| def testGetByte(self): | |
| self.assertEqual(5, fdt_util.GetByte(self.node, 'byteval')) | |
| self.assertEqual(3, fdt_util.GetByte(self.node, 'missing', 3)) | |
| with self.assertRaises(ValueError) as e: | |
| fdt_util.GetByte(self.node, 'longbytearray') | |
| self.assertIn("property 'longbytearray' has list value: expecting a " | |
| 'single byte', str(e.exception)) | |
| with self.assertRaises(ValueError) as e: | |
| fdt_util.GetByte(self.node, 'intval') | |
| self.assertIn("property 'intval' has length 4, expecting 1", | |
| str(e.exception)) | |
| def testGetPhandleList(self): | |
| dtb = fdt.FdtScan('tools/dtoc/dtoc_test_phandle.dts') | |
| node = dtb.GetNode('/phandle-source2') | |
| self.assertEqual([1], fdt_util.GetPhandleList(node, 'clocks')) | |
| node = dtb.GetNode('/phandle-source') | |
| self.assertEqual([1, 2, 11, 3, 12, 13, 1], | |
| fdt_util.GetPhandleList(node, 'clocks')) | |
| self.assertEqual(None, fdt_util.GetPhandleList(node, 'missing')) | |
| def testGetDataType(self): | |
| self.assertEqual(1, fdt_util.GetDatatype(self.node, 'intval', int)) | |
| self.assertEqual('message', fdt_util.GetDatatype(self.node, 'stringval', | |
| str)) | |
| with self.assertRaises(ValueError) as e: | |
| self.assertEqual(3, fdt_util.GetDatatype(self.node, 'boolval', | |
| bool)) | |
| def testFdtCellsToCpu(self): | |
| val = self.node.props['intarray'].value | |
| self.assertEqual(0, fdt_util.fdt_cells_to_cpu(val, 0)) | |
| self.assertEqual(2, fdt_util.fdt_cells_to_cpu(val, 1)) | |
| dtb2 = fdt.FdtScan('tools/dtoc/dtoc_test_addr64.dts') | |
| node1 = dtb2.GetNode('/test1') | |
| val = node1.props['reg'].value | |
| self.assertEqual(0x1234, fdt_util.fdt_cells_to_cpu(val, 2)) | |
| node2 = dtb2.GetNode('/test2') | |
| val = node2.props['reg'].value | |
| self.assertEqual(0x1234567890123456, fdt_util.fdt_cells_to_cpu(val, 2)) | |
| self.assertEqual(0x9876543210987654, fdt_util.fdt_cells_to_cpu(val[2:], | |
| 2)) | |
| self.assertEqual(0x12345678, fdt_util.fdt_cells_to_cpu(val, 1)) | |
| def testEnsureCompiled(self): | |
| """Test a degenerate case of this function (file already compiled)""" | |
| dtb = fdt_util.EnsureCompiled('tools/dtoc/dtoc_test_simple.dts') | |
| self.assertEqual(dtb, fdt_util.EnsureCompiled(dtb)) | |
| def testEnsureCompiledTmpdir(self): | |
| """Test providing a temporary directory""" | |
| try: | |
| old_outdir = tools.outdir | |
| tools.outdir= None | |
| tmpdir = tempfile.mkdtemp(prefix='test_fdt.') | |
| dtb = fdt_util.EnsureCompiled('tools/dtoc/dtoc_test_simple.dts', | |
| tmpdir) | |
| self.assertEqual(tmpdir, os.path.dirname(dtb)) | |
| shutil.rmtree(tmpdir) | |
| finally: | |
| tools.outdir= old_outdir | |
| def RunTestCoverage(): | |
| """Run the tests and check that we get 100% coverage""" | |
| test_util.RunTestCoverage('tools/dtoc/test_fdt.py', None, | |
| ['tools/patman/*.py', '*test_fdt.py'], options.build_dir) | |
| def RunTests(args): | |
| """Run all the test we have for the fdt model | |
| Args: | |
| args: List of positional args provided to fdt. This can hold a test | |
| name to execute (as in 'fdt -t testFdt', for example) | |
| """ | |
| result = unittest.TestResult() | |
| sys.argv = [sys.argv[0]] | |
| test_name = args and args[0] or None | |
| for module in (TestFdt, TestNode, TestProp, TestFdtUtil): | |
| if test_name: | |
| try: | |
| suite = unittest.TestLoader().loadTestsFromName(test_name, module) | |
| except AttributeError: | |
| continue | |
| else: | |
| suite = unittest.TestLoader().loadTestsFromTestCase(module) | |
| suite.run(result) | |
| print(result) | |
| for _, err in result.errors: | |
| print(err) | |
| for _, err in result.failures: | |
| print(err) | |
| if __name__ != '__main__': | |
| sys.exit(1) | |
| parser = OptionParser() | |
| parser.add_option('-B', '--build-dir', type='string', default='b', | |
| help='Directory containing the build output') | |
| parser.add_option('-P', '--processes', type=int, | |
| help='set number of processes to use for running tests') | |
| parser.add_option('-t', '--test', action='store_true', dest='test', | |
| default=False, help='run tests') | |
| parser.add_option('-T', '--test-coverage', action='store_true', | |
| default=False, help='run tests and check for 100% coverage') | |
| (options, args) = parser.parse_args() | |
| # Run our meagre tests | |
| if options.test: | |
| RunTests(args) | |
| elif options.test_coverage: | |
| RunTestCoverage() |