| Stephen Warren | 1cd85f5 | 2016-02-08 14:44:16 -0700 | [diff] [blame] | 1 | # SPDX-License-Identifier: GPL-2.0 |
| Tom Rini | 83d290c | 2018-05-06 17:58:06 -0400 | [diff] [blame] | 2 | # Copyright (c) 2016, NVIDIA CORPORATION. All rights reserved. |
| Stephen Warren | 1cd85f5 | 2016-02-08 14:44:16 -0700 | [diff] [blame] | 3 | |
| 4 | import os.path |
| 5 | import pytest |
| 6 | |
| 7 | @pytest.mark.buildconfigspec('ut_dm') |
| 8 | def test_ut_dm_init(u_boot_console): |
| 9 | """Initialize data for ut dm tests.""" |
| 10 | |
| 11 | fn = u_boot_console.config.source_dir + '/testflash.bin' |
| 12 | if not os.path.exists(fn): |
| 13 | data = 'this is a test' |
| 14 | data += '\x00' * ((4 * 1024 * 1024) - len(data)) |
| 15 | with open(fn, 'wb') as fh: |
| 16 | fh.write(data) |
| 17 | |
| 18 | fn = u_boot_console.config.source_dir + '/spi.bin' |
| 19 | if not os.path.exists(fn): |
| 20 | data = '\x00' * (2 * 1024 * 1024) |
| 21 | with open(fn, 'wb') as fh: |
| 22 | fh.write(data) |
| 23 | |
| 24 | def test_ut(u_boot_console, ut_subtest): |
| 25 | """Execute a "ut" subtest.""" |
| 26 | |
| 27 | output = u_boot_console.run_command('ut ' + ut_subtest) |
| 28 | assert output.endswith('Failures: 0') |