Fixes #271: Use parametrize markers params for getting example_kwargs

Author: sliwinski-milosz

pytest_bdd/reporting.py

@@ -7,7 +7,7 @@
 import time
 
 from .feature import force_unicode
-from .utils import get_parametrize_markers_args
+from .utils import get_parametrize_markers_args, get_parametrize_params
 
 
 class StepReport(object):
@@ -73,20 +73,30 @@ def __init__(self, scenario, node):
         """
         self.scenario = scenario
         self.step_reports = []
-        self.param_index = None
+
         parametrize_args = get_parametrize_markers_args(node)
-        if parametrize_args and scenario.examples:
-            param_names = parametrize_args[0] if isinstance(parametrize_args[0], (tuple, list)) else [
-                parametrize_args[0]]
-            param_values = parametrize_args[1]
+        params = get_parametrize_params(parametrize_args)
+
+        self.param_index = self.get_param_index(node, params)
+        self.example_kwargs = self.get_example_kwargs(node, params)
+
+    def get_param_index(self, node, params):
+        if params:
+            param_names = params[0]['names']
+            param_values = params[0]['values']
             node_param_values = [node.funcargs[param_name] for param_name in param_names]
             if node_param_values in param_values:
-                self.param_index = param_values.index(node_param_values)
+                return param_values.index(node_param_values)
             elif tuple(node_param_values) in param_values:
-                self.param_index = param_values.index(tuple(node_param_values))
-        self.example_kwargs = {
-            example_param: force_unicode(node.funcargs[example_param])
-            for example_param in scenario.get_example_params()
+                return param_values.index(tuple(node_param_values))
+        return None
+
+    def get_example_kwargs(self, node, params):
+        params_names = (param['names'] for param in params)
+        all_names = sum(params_names, [])
+        return {
+            example_param_name: force_unicode(node.funcargs[example_param_name])
+            for example_param_name in all_names
         }
 
     @property

pytest_bdd/utils.py

@@ -110,3 +110,33 @@ def get_markers_args_using_iter_markers(node, mark_name):
 def get_markers_args_using_get_marker(node, mark_name):
     """Deprecated on pytest>=3.6"""
     return getattr(node.get_marker(mark_name), 'args', ())
+
+
+def get_parametrize_params(parametrize_args):
+    """Group parametrize markers arguments names and values.
+
+    :param parametrize_args: parametrize markers arguments.
+    :return: `list` of `dict` in the form of:
+        [
+            {
+                "names": ["name1", "name2", ...],
+                "values": [value1, value2, ...],
+            },
+            ...
+        ]
+    """
+    params = []
+    for i in range(0, len(parametrize_args), 2):
+        params.append({
+            'names': _get_param_names(parametrize_args[i]),
+            'values': parametrize_args[i+1]
+        })
+    return params
+
+
+def _get_param_names(names):
+    if not isinstance(names, (tuple, list)):
+        # As pytest.mark.parametrize has only one param name,
+        # it is not returned as a list. Convert it to list:
+        names = [names]
+    return names

tests/feature/outline_feature.feature

@@ -4,6 +4,7 @@ Feature: Outline
     | start | eat | left |
     |  12   |  5  |  7   |
     |  5    |  4  |  1   |
+    |  4    |  2  |  2   |
 
     Scenario Outline: Outlined given, when, thens
         Given there are <start> <fruits>

tests/feature/parametrized.feature

@@ -2,3 +2,7 @@ Scenario: Parametrized given, when, thens
     Given there are <start> cucumbers
     When I eat <eat> cucumbers
     Then I should have <left> cucumbers
+
+
+Scenario: Parametrized given - single param
+    Given there are <start> cucumbers

tests/feature/test_outline.py

@@ -168,7 +168,7 @@ def should_have_left_fruits(start_fruits, start, eat, left, fruits):
 def test_outlined_feature(request):
     assert get_parametrize_markers_args(request.node) == (
         ['start', 'eat', 'left'],
-        [[12, 5.0, '7'], [5, 4.0, '1']],
+        [[12, 5.0, '7'], [5, 4.0, '1'], [4, 2.0, '2']],
         ['fruits'],
         [[u'oranges'], [u'apples']]
     )

tests/feature/test_parametrized.py

@@ -14,6 +14,17 @@ def test_parametrized(request, start, eat, left):
     """Test parametrized scenario."""
 
 
+@pytest.mark.parametrize(
+    'start', [12]
+)
+@scenario(
+    'parametrized.feature',
+    'Parametrized given - single param',
+)
+def test_parametrized_single_param(request, start):
+    """Test parametrized scenario."""
+
+
 @pytest.fixture(params=[1, 2])
 def foo_bar(request):
     return 'bar' * request.param