mirror of
https://github.com/u-boot/u-boot.git
synced 2025-04-25 23:06:15 +00:00
752c376987
This fixture name is quite long and results in lots of verbose code. We know this is U-Boot so the 'u_boot_' part is not necessary. But it is also a bit of a misnomer, since it provides access to all the information available to tests. It is not just the console. It would be too confusing to use con as it would be confused with config and it is probably too short. So shorten it to 'ubman'. Signed-off-by: Simon Glass <sjg@chromium.org> Link: https://lore.kernel.org/u-boot/CAFLszTgPa4aT_J9h9pqeTtLCVn4x2JvLWRcWRD8NaN3uoSAtyA@mail.gmail.com/
93 lines
3 KiB
Python
93 lines
3 KiB
Python
# SPDX-License-Identifier: GPL-2.0+
|
|
# Copyright 2022 Google LLC
|
|
|
|
"""Common utility functions for FIT tests"""
|
|
|
|
import os
|
|
|
|
import utils as util
|
|
|
|
def make_fname(ubman, basename):
|
|
"""Make a temporary filename
|
|
|
|
Args:
|
|
ubman (ConsoleBase): ubman to use
|
|
basename (str): Base name of file to create (within temporary directory)
|
|
Return:
|
|
Temporary filename
|
|
"""
|
|
|
|
return os.path.join(ubman.config.build_dir, basename)
|
|
|
|
def make_its(ubman, base_its, params, basename='test.its'):
|
|
"""Make a sample .its file with parameters embedded
|
|
|
|
Args:
|
|
ubman (ConsoleBase): ubman to use
|
|
base_its (str): Template text for the .its file, typically containing
|
|
%() references
|
|
params (dict of str): Parameters to embed in the %() strings
|
|
basename (str): base name to write to (will be placed in the temp dir)
|
|
Returns:
|
|
str: Filename of .its file created
|
|
"""
|
|
its = make_fname(ubman, basename)
|
|
with open(its, 'w', encoding='utf-8') as outf:
|
|
print(base_its % params, file=outf)
|
|
return its
|
|
|
|
def make_fit(ubman, mkimage, base_its, params, basename='test.fit', base_fdt=None):
|
|
"""Make a sample .fit file ready for loading
|
|
|
|
This creates a .its script with the selected parameters and uses mkimage to
|
|
turn this into a .fit image.
|
|
|
|
Args:
|
|
ubman (ConsoleBase): ubman to use
|
|
mkimage (str): Filename of 'mkimage' utility
|
|
base_its (str): Template text for the .its file, typically containing
|
|
%() references
|
|
params (dict of str): Parameters to embed in the %() strings
|
|
basename (str): base name to write to (will be placed in the temp dir)
|
|
Return:
|
|
Filename of .fit file created
|
|
"""
|
|
fit = make_fname(ubman, basename)
|
|
its = make_its(ubman, base_its, params)
|
|
util.run_and_log(ubman, [mkimage, '-f', its, fit])
|
|
if base_fdt:
|
|
with open(make_fname(ubman, 'u-boot.dts'), 'w') as fd:
|
|
fd.write(base_fdt)
|
|
return fit
|
|
|
|
def make_kernel(ubman, basename, text):
|
|
"""Make a sample kernel with test data
|
|
|
|
Args:
|
|
ubman (ConsoleBase): ubman to use
|
|
basename (str): base name to write to (will be placed in the temp dir)
|
|
text (str): Contents of the kernel file (will be repeated 100 times)
|
|
Returns:
|
|
str: Full path and filename of the kernel it created
|
|
"""
|
|
fname = make_fname(ubman, basename)
|
|
data = ''
|
|
for i in range(100):
|
|
data += f'this {text} {i} is unlikely to boot\n'
|
|
with open(fname, 'w', encoding='utf-8') as outf:
|
|
print(data, file=outf)
|
|
return fname
|
|
|
|
def make_dtb(ubman, base_fdt, basename):
|
|
"""Make a sample .dts file and compile it to a .dtb
|
|
|
|
Returns:
|
|
ubman (ConsoleBase): ubman to use
|
|
Filename of .dtb file created
|
|
"""
|
|
src = make_fname(ubman, f'{basename}.dts')
|
|
dtb = make_fname(ubman, f'{basename}.dtb')
|
|
with open(src, 'w', encoding='utf-8') as outf:
|
|
outf.write(base_fdt)
|
|
util.run_and_log(ubman, ['dtc', src, '-O', 'dtb', '-o', dtb])
|
|
return dtb
|