pytest-embedded 2.2.0__tar.gz → 2.2.1__tar.gz

{pytest_embedded-2.2.0 → pytest_embedded-2.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pytest-embedded
-Version: 2.2.0
+Version: 2.2.1
 Summary: A pytest plugin that designed for embedded testing.
 Author-email: Fu Hanxi <fuhanxi@espressif.com>
 Requires-Python: >=3.10

{pytest_embedded-2.2.0 → pytest_embedded-2.2.1}/pytest_embedded/__init__.py

@@ -6,4 +6,4 @@ from .dut_factory import DutFactory
 
 __all__ = ['App', 'Dut', 'DutFactory']
 
-__version__ = '2.2.0'
+__version__ = '2.2.1'

{pytest_embedded-2.2.0 → pytest_embedded-2.2.1}/pytest_embedded/dut_factory.py

@@ -838,3 +838,16 @@ class DutFactory:
                 _close_or_terminate(obj)
             del layout
             raise e
+
+    @classmethod
+    def get_all_duts(cls) -> list[Dut]:
+        """Get all DUTs created by DutFactory."""
+
+        duts = []
+        for layout in cls.obj_stack:
+            # The DUT is always the last object in the layout
+            dut = layout[-1]
+            if isinstance(dut, Dut):
+                duts.append(dut)
+
+        return duts

{pytest_embedded-2.2.0 → pytest_embedded-2.2.1}/pytest_embedded/plugin.py

@@ -1372,10 +1372,18 @@ class PytestEmbedded:
 
     @pytest.hookimpl(trylast=True)
     def pytest_runtest_call(self, item: Function):
-        # raise dut failed cases
+        all_duts: list[Dut] = []
+
+        # Check DUTs created by fixture
         if 'dut' in item.funcargs:
-            duts = [dut for dut in to_list(item.funcargs['dut']) if isinstance(dut, Dut)]
-            self._raise_dut_failed_cases_if_exists(duts)  # type: ignore
+            fixture_duts = [dut for dut in to_list(item.funcargs['dut']) if isinstance(dut, Dut)]
+            all_duts.extend(fixture_duts)
+
+        # Check DUTs created by DutFactory
+        factory_duts = DutFactory.get_all_duts()
+        all_duts.extend(factory_duts)
+
+        self._raise_dut_failed_cases_if_exists(all_duts)  # type: ignore
 
     @pytest.hookimpl(trylast=True)  # combine all possible junit reports should be the last step
     def pytest_sessionfinish(self, session: Session, exitstatus: int) -> None: