Merge pull request #345 from horw/feat/soc_filtered_targets

feat: get targets filtered by soc attrs

Author: hfudev

pytest-embedded-idf/pytest_embedded_idf/utils.py

@@ -3,6 +3,7 @@
 
 import pytest
 from esp_bool_parser import PREVIEW_TARGETS, SUPPORTED_TARGETS
+from esp_bool_parser.bool_parser import parse_bool_expr
 
 supported_targets = ContextVar('supported_targets', default=SUPPORTED_TARGETS)
 preview_targets = ContextVar('preview_targets', default=PREVIEW_TARGETS)
@@ -90,3 +91,12 @@ def decorator(func):
         return pytest.mark.parametrize(','.join(param_list), processed_values, indirect=indirect)(func)
 
     return decorator
+
+
+def soc_filtered_targets(soc_statement: str) -> t.List[str]:
+    stm = parse_bool_expr(soc_statement)
+    result = []
+    for target in [*supported_targets.get(), *preview_targets.get()]:
+        if stm.get_value(target, ''):
+            result.append(target)
+    return result

pytest-embedded-idf/tests/test_idf.py

@@ -1153,6 +1153,22 @@ def test_skip_if_for_condition():
         run_test_for_condition(c, cf)
 
 
+def test_soc_filtered_targets(testdir, copy_mock_esp_idf, monkeypatch):  # noqa: ARG001
+    monkeypatch.setenv('IDF_PATH', str(testdir))
+    from pytest_embedded_idf.utils import soc_filtered_targets
+
+    assert soc_filtered_targets('SOC_A == 1') == ['esp32c3', 'esp32s3', 'esp32c6', 'esp32c5']
+    assert soc_filtered_targets('SOC_A == 1 or SOC_B == 1') == [
+        'esp32',
+        'esp32c3',
+        'esp32s3',
+        'esp32c2',
+        'esp32c6',
+        'esp32h2',
+        'esp32c5',
+    ]
+
+
 def test_skip_if_soc_target_in_args(testdir, copy_mock_esp_idf, monkeypatch):  # noqa: ARG001
     monkeypatch.setenv('IDF_PATH', str(testdir))