[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

src/pytest_bdd/scenario.py

@@ -162,11 +162,17 @@ def _execute_step_function(
         raise
 
     if context.target_fixture is not None:
-        target_fixture_tokens = [token for token in context.target_fixture.split(',') if token]
-        return_values = (return_value,) if not isinstance(return_value, tuple) else return_value
-        assert len(target_fixture_tokens) == len(return_values), f"Return value count: {len(return_values)} are not matching target_fixture count: {len(target_fixture_tokens)}"
-        for token, value in zip(target_fixture_tokens, return_values):
-            inject_fixture(request, token, value)
+        if isinstance(return_value, Exception):
+            arg = context.target_exception if context.target_exception else "response"
+            inject_fixture(request=request, arg=arg, value=return_value)
+        else:
+            target_fixture_tokens = [token for token in context.target_fixture.split(",") if token]
+            return_values = (return_value,) if not isinstance(return_value, tuple) else return_value
+            assert len(target_fixture_tokens) == len(
+                return_values
+            ), f"Return value count: {len(return_values)} are not matching target_fixture count: {len(target_fixture_tokens)}"
+            for token, value in zip(target_fixture_tokens, return_values):
+                inject_fixture(request=request, arg=token, value=value)
 
     request.config.hook.pytest_bdd_after_step(**kw)
 

src/pytest_bdd/steps.py

@@ -66,6 +66,7 @@ class StepFunctionContext:
     parser: StepParser
     converters: dict[str, Callable[..., Any]] = field(default_factory=dict)
     target_fixture: str | None = None
+    target_exception: str | None = None
 
 
 def get_step_fixture_name(step: Step) -> str:
@@ -96,6 +97,7 @@ def when(
     name: str | StepParser,
     converters: dict[str, Callable] | None = None,
     target_fixture: str | None = None,
+    target_exception: str | None = None,
     stacklevel: int = 1,
 ) -> Callable:
     """When step decorator.
@@ -104,11 +106,12 @@ def when(
     :param converters: Optional `dict` of the argument or parameter converters in form
                        {<param_name>: <converter function>}.
     :param target_fixture: Target fixture name to replace by steps definition function.
+    :param target_exception: Target exception name to receive Exception object
     :param stacklevel: Stack level to find the caller frame. This is used when injecting the step definition fixture.
 
     :return: Decorator function for the step.
     """
-    return step(name, WHEN, converters=converters, target_fixture=target_fixture, stacklevel=stacklevel)
+    return step(name, WHEN, converters=converters, target_fixture=target_fixture, target_exception=target_exception, stacklevel=stacklevel)
 
 
 def then(
@@ -135,6 +138,7 @@ def step(
     type_: Literal["given", "when", "then"] | None = None,
     converters: dict[str, Callable] | None = None,
     target_fixture: str | None = None,
+    target_exception: str | None = None,
     stacklevel: int = 1,
 ) -> Callable[[TCallable], TCallable]:
     """Generic step decorator.
@@ -143,6 +147,7 @@ def step(
     :param type_: Step type ("given", "when" or "then"). If None, this step will work for all the types.
     :param converters: Optional step arguments converters mapping.
     :param target_fixture: Optional fixture name to replace by step definition.
+    :param target_exception: Optional target exception name
     :param stacklevel: Stack level to find the caller frame. This is used when injecting the step definition fixture.
 
     :return: Decorator function for the step.
@@ -165,6 +170,7 @@ def decorator(func: TCallable) -> TCallable:
             parser=parser,
             converters=converters,
             target_fixture=target_fixture,
+            target_exception=target_exception,
         )
 
         def step_function_marker() -> StepFunctionContext:

tests/feature/test_steps.py

@@ -584,3 +584,49 @@ def _(stuff):
             "*Tearing down...*",
         ]
     )
+
+
+def test_when_exception(pytester):
+    pytester.makefile(
+        ".feature",
+        when_exception=textwrap.dedent(
+            """\
+            Feature: When exception
+                Scenario: Test when exception is generated
+                    When I have injected exception
+                    Then Exception object should be received as default 'response' param
+                    When I have injected exception with target_exception='exception_ret'
+                    Then Exception object should be received as target_exception='exception_ret'
+            """
+        ),
+    )
+    pytester.makepyfile(
+        textwrap.dedent(
+            """\
+        import pytest
+        from pytest_bdd import when, then, scenario
+
+        @scenario("when_exception.feature", "Test when exception is generated")
+        def test_when_exception():
+            pass
+        
+        @when("I have injected exception", target_fixture="foo")
+        def _():
+            return Exception("Dummy Exception obj")
+
+        @when("I have injected exception with target_exception='exception_ret'", target_fixture="foo", target_exception="exception_ret")
+        def _():
+            return Exception("Dummy Exception obj")
+
+        @then("Exception object should be received as default 'response' param")
+        def _(response):
+            assert isinstance(response, Exception), "response is not Exception object"
+
+        @then("Exception object should be received as target_exception='exception_ret'")
+        def _(exception_ret):
+            assert isinstance(exception_ret, Exception), "response is not Exception object"
+        """
+        )
+    )
+    result = pytester.runpytest()
+    result.assert_outcomes(passed=1)

tests/steps/test_given.py

@@ -2,7 +2,7 @@
 import textwrap
 
 
-def test_given_injection(pytester):
+def test_given_injection_single_value(pytester):
     pytester.makefile(
         ".feature",
         given=textwrap.dedent(
@@ -40,15 +40,15 @@ def _(foo):
     result.assert_outcomes(passed=1)
 
 
-def test_given_injection_multiple(pytester):
+def test_given_injection_multiple_values(pytester):
     pytester.makefile(
         ".feature",
         given=textwrap.dedent(
             """\
             Feature: Given
                 Scenario: Test given fixture injection
-                    Given I have injecting given
-                    Then foo should be "injected foo"
+                    Given I have injecting given values
+                    Then values should be received
             """
         ),
     )
@@ -62,18 +62,19 @@ def test_given_injection_multiple(pytester):
         def test_given():
             pass
 
-        @given("I have injecting given", target_fixture="foo,bar")
+        @given("I have injecting given values", target_fixture="foo,city,numbers")
         def _():
-            return "injected foo", "injected bar"
+            return ("injected foo", {"city": ["Boston", "Houston"]}, [10,20,30],)
 
 
-        @then('foo should be "injected foo"')
-        def _(foo, bar):
+        @then("values should be received")
+        def _(foo, city, numbers):
             assert foo == "injected foo"
-            assert bar == "injected bar"
+            assert city == {"city": ["Boston", "Houston"]}
+            assert numbers == [10,20,30]
 
         """
         )
     )
     result = pytester.runpytest()
-    result.assert_outcomes(passed=1)
+    result.assert_outcomes(passed=1)