1
0
mirror of https://github.com/ARM-software/devlib.git synced 2024-10-05 18:30:50 +01:00

tests/test_target: Read target connection settings from a YAML file

This will be useful in automating CI tests without modifying the source
code.

Replace unittest with pytest in order to make parameter passing to test
functions easier.

Move target configuration reading and generating target object outside
of the test function. Because we will run the test function for new
targets and may want to add new test functions.

While we are here, also fix pylint issues.

Signed-off-by: Metin Kaya <metin.kaya@arm.com>
This commit is contained in:
Metin Kaya 2024-02-05 08:36:19 +00:00 committed by Marc Bonnici
parent b5715b6560
commit a1718c3700
3 changed files with 54 additions and 20 deletions

View File

@ -102,6 +102,7 @@ params = dict(
'wrapt', # Basic for construction of decorator functions
'numpy',
'pandas',
'pytest',
'lxml', # More robust xml parsing
'nest_asyncio', # Allows running nested asyncio loops
'future', # for the "past" Python package

View File

@ -0,0 +1,5 @@
LocalLinuxTarget:
entry-0:
connection_settings:
unrooted: True

View File

@ -14,35 +14,63 @@
# limitations under the License.
#
"""Module for testing targets."""
import os
import shutil
import tempfile
from unittest import TestCase
from pprint import pp
import pytest
from devlib import LocalLinuxTarget
from devlib.utils.misc import load_struct_from_yaml
class TestReadTreeValues(TestCase):
def build_targets():
"""Read targets from a YAML formatted config file"""
def test_read_multiline_values(self):
data = {
'test1': '1',
'test2': '2\n\n',
'test3': '3\n\n4\n\n',
}
config_file = os.path.join(os.path.dirname(os.path.realpath(__file__)), 'target_configs.yaml')
tempdir = tempfile.mkdtemp(prefix='devlib-test-')
for key, value in data.items():
path = os.path.join(tempdir, key)
with open(path, 'w') as wfh:
wfh.write(value)
target_configs = load_struct_from_yaml(config_file)
if target_configs is None:
raise ValueError(f'{config_file} looks empty!')
t = LocalLinuxTarget(connection_settings={'unrooted': True})
raw_result = t.read_tree_values_flat(tempdir)
result = {os.path.basename(k): v for k, v in raw_result.items()}
targets = []
shutil.rmtree(tempdir)
if target_configs.get('LocalLinuxTarget') is not None:
print('> LocalLinux targets:')
for entry in target_configs['LocalLinuxTarget'].values():
pp(entry)
ll_target = LocalLinuxTarget(connection_settings=entry['connection_settings'])
targets.append(ll_target)
self.assertEqual({k: v.strip()
for k, v in data.items()},
result)
return targets
@pytest.mark.parametrize("target", build_targets())
def test_read_multiline_values(target):
"""
Test Target.read_tree_values_flat()
:param target: Type of target per :class:`Target` based classes.
:type target: Target
"""
data = {
'test1': '1',
'test2': '2\n\n',
'test3': '3\n\n4\n\n',
}
tempdir = tempfile.mkdtemp(prefix='devlib-test-')
for key, value in data.items():
path = os.path.join(tempdir, key)
with open(path, 'w', encoding='utf-8') as wfh:
wfh.write(value)
raw_result = target.read_tree_values_flat(tempdir)
result = {os.path.basename(k): v for k, v in raw_result.items()}
shutil.rmtree(tempdir)
assert {k: v.strip() for k, v in data.items()} == result