test/py/tests/test_ut.py

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