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>

setup.py

@@ -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

tests/target_configs.yaml

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

tests/test_target.py

@@ -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):
+    config_file = os.path.join(os.path.dirname(os.path.realpath(__file__)), 'target_configs.yaml')
-        data = {
-            'test1': '1',
-            'test2': '2\n\n',
-            'test3': '3\n\n4\n\n',
-        }
 
-        tempdir = tempfile.mkdtemp(prefix='devlib-test-')
+    target_configs = load_struct_from_yaml(config_file)
-        for key, value in data.items():
+    if target_configs is None:
-            path = os.path.join(tempdir, key)
+        raise ValueError(f'{config_file} looks empty!')
-            with open(path, 'w') as wfh:
-                wfh.write(value)
 
-        t = LocalLinuxTarget(connection_settings={'unrooted': True})
+    targets = []
-        raw_result = t.read_tree_values_flat(tempdir)
-        result = {os.path.basename(k): v for k, v in raw_result.items()}
 
-        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()
+    return targets
-                          for k, v in data.items()},
+
-                         result)
+
+@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