🧪 tests: add testcase support foreach stage exec concurrent.

Author: korawica

tests/stages/test_stage_foreach.py

@@ -626,56 +626,92 @@ def test_foreach_stage_exec_concurrent_raise():
                 },
             }
         except AssertionError:
-            assert rs.context == {
-                "status": FAILED,
-                "items": [1, 2],
-                "foreach": {
-                    1: {
-                        "status": FAILED,
-                        "item": 1,
-                        "stages": {
-                            "raise-error": {
-                                "outputs": {},
-                                "errors": {
-                                    "name": "ValueError",
-                                    "message": "Testing raise error PyStage!!!",
-                                },
-                                "status": FAILED,
-                            }
-                        },
-                        "errors": {
-                            "name": "StageError",
-                            "message": "Item execution was break because its nested-stage, 'raise-error', failed.",
+            try:
+                assert rs.context == {
+                    "status": FAILED,
+                    "items": [1, 2],
+                    "foreach": {
+                        1: {
+                            "status": FAILED,
+                            "item": 1,
+                            "stages": {
+                                "raise-error": {
+                                    "outputs": {},
+                                    "errors": {
+                                        "name": "ValueError",
+                                        "message": "Testing raise error PyStage!!!",
+                                    },
+                                    "status": FAILED,
+                                }
+                            },
+                            "errors": {
+                                "name": "StageError",
+                                "message": "Item execution was break because its nested-stage, 'raise-error', failed.",
+                            },
                         },
-                    },
-                    2: {
-                        "status": CANCEL,
-                        "item": 2,
-                        "stages": {
-                            "raise-error": {"outputs": {}, "status": SKIP},
-                            "echo": {
-                                "outputs": {},
-                                "errors": {
-                                    "name": "StageCancelError",
-                                    "message": "Execution was canceled from the event before start parallel.",
+                        2: {
+                            "status": CANCEL,
+                            "item": 2,
+                            "stages": {
+                                "raise-error": {"outputs": {}, "status": SKIP},
+                                "echo": {
+                                    "outputs": {},
+                                    "errors": {
+                                        "name": "StageCancelError",
+                                        "message": "Execution was canceled from the event before start parallel.",
+                                    },
+                                    "status": CANCEL,
                                 },
-                                "status": CANCEL,
+                            },
+                            "errors": {
+                                "name": "StageCancelError",
+                                "message": "Item execution was canceled from the event after end item execution.",
                             },
                         },
-                        "errors": {
+                    },
+                    "errors": {
+                        2: {
                             "name": "StageCancelError",
                             "message": "Item execution was canceled from the event after end item execution.",
                         },
+                        1: {
+                            "name": "StageError",
+                            "message": "Item execution was break because its nested-stage, 'raise-error', failed.",
+                        },
                     },
-                },
-                "errors": {
-                    2: {
-                        "name": "StageCancelError",
-                        "message": "Item execution was canceled from the event after end item execution.",
+                }
+            except AssertionError:
+                assert rs.context == {
+                    "status": FAILED,
+                    "items": [1, 2],
+                    "errors": {
+                        1: {
+                            "message": (
+                                "Item execution was break because its "
+                                "nested-stage, 'raise-error', failed."
+                            ),
+                            "name": "StageError",
+                        },
                     },
-                    1: {
-                        "name": "StageError",
-                        "message": "Item execution was break because its nested-stage, 'raise-error', failed.",
+                    "foreach": {
+                        1: {
+                            "errors": {
+                                "message": "Item execution was break because its nested-stage, "
+                                "'raise-error', failed.",
+                                "name": "StageError",
+                            },
+                            "item": 1,
+                            "stages": {
+                                "raise-error": {
+                                    "errors": {
+                                        "message": "Testing raise error PyStage!!!",
+                                        "name": "ValueError",
+                                    },
+                                    "outputs": {},
+                                    "status": FAILED,
+                                },
+                            },
+                            "status": FAILED,
+                        },
                     },
-                },
-            }
+                }