Merge pull request #342 from horw/feat/test-supported-targets

Feat/test supported targets

Author: hfudev

docs/usages/markers.rst

@@ -81,6 +81,27 @@ Target Extension
 
 In scenarios where the supported targets are [esp32, esp32c3, esp32s3], ``idf_parametrize`` simplifies the process of creating parameterized tests by automatically expanding the target list.
 
+By default, the values for ``SUPPORTED_TARGETS`` and ``PREVIEW_TARGETS`` are imported from:
+
+.. code:: python
+
+   from esp_bool_parser import PREVIEW_TARGETS, SUPPORTED_TARGETS
+
+However, you can propagate custom values by using the following:
+
+.. code:: python
+
+   from pytest_embedded_idf.utils import supported_targets, preview_targets
+
+   supported_targets.set(CUSTOM_SUPPORT_TARGETS)
+   preview_targets.set(CUSTOM_SUPPORT_TARGETS)
+
+Another way to override ``supported_targets`` and ``preview_targets`` is by using command-line arguments:
+
+.. code:: sh
+
+   pytest --supported-targets esp32,esp32c3 --preview-targets esp32p4 ...
+
 **Example:**
 
 .. code:: python

pytest-embedded-idf/pytest_embedded_idf/utils.py

@@ -1,8 +1,12 @@
 import typing as t
+from contextvars import ContextVar
 
 import pytest
 from esp_bool_parser import PREVIEW_TARGETS, SUPPORTED_TARGETS
 
+supported_targets = ContextVar('supported_targets', default=SUPPORTED_TARGETS)
+preview_targets = ContextVar('preview_targets', default=PREVIEW_TARGETS)
+
 
 def _expand_target_values(values: t.List[t.List[t.Any]], target_index: int) -> t.List[t.List[t.Any]]:
     """
@@ -13,11 +17,11 @@ def _expand_target_values(values: t.List[t.List[t.Any]], target_index: int) -> t
         target = value[target_index]
         if target == 'supported_targets':
             expanded_values.extend([
-                value[:target_index] + [target] + value[target_index + 1 :] for target in SUPPORTED_TARGETS
+                value[:target_index] + [target] + value[target_index + 1 :] for target in supported_targets.get()
             ])
         elif target == 'preview_targets':
             expanded_values.extend([
-                value[:target_index] + [target] + value[target_index + 1 :] for target in PREVIEW_TARGETS
+                value[:target_index] + [target] + value[target_index + 1 :] for target in preview_targets.get()
             ])
         else:
             expanded_values.append(value)

pytest-embedded/pytest_embedded/plugin.py

@@ -208,6 +208,12 @@ def pytest_addoption(parser):
         help='force mode for esptool',
     )
     idf_group = parser.getgroup('embedded-idf')
+    idf_group.addoption(
+        '--supported-targets', help='Comma-separated list of supported targets for the test case. (Default: None)'
+    )
+    idf_group.addoption(
+        '--preview-targets', help='Comma-separated list of preview targets for the test case. (Default: None)'
+    )
     idf_group.addoption(
         '--part-tool',
         help='Partition tool path, used for parsing partition table. '
@@ -1207,6 +1213,16 @@ def pytest_configure(config: Config) -> None:
     )
     config.stash[_junit_report_path_key] = config.option.xmlpath
 
+    supported_targets_args = config.getoption('supported_targets', None)
+    preview_targets_args = config.getoption('preview_targets', None)
+    if supported_targets_args or preview_targets_args:
+        from pytest_embedded_idf.utils import preview_targets, supported_targets
+
+        if supported_targets_args is not None:
+            supported_targets.set([_t.strip() for _t in supported_targets_args.split(',')])
+        if preview_targets_args:
+            preview_targets.set([_t.strip() for _t in preview_targets_args.split(',')])
+
     config.stash[_pytest_embedded_key] = PytestEmbedded(
         parallel_count=config.getoption('parallel_count'),
         parallel_index=config.getoption('parallel_index'),