Fixed ast.With

Author: henrychv

bzt/modules/_apiritif/generator.py

@@ -773,10 +773,11 @@ def _gen_action(self, action_config, parent_request=None, index_label=""):
             label = self._create_action_label(parent_request.label, index_label, action)
 
             return [ast.With(
-                context_expr=ast_call(
-                    func=ast_attr("apiritif.transaction"),
-                    args=[self._gen_expr(label)]),
-                optional_vars=None,
+                items=[ast.withitem(
+                    context_expr=ast_call(
+                        func=ast_attr("apiritif.transaction"),
+                        args=[self._gen_expr(label)]),
+                    optional_vars=None)],
                 body=[ast.Expr(element) for element in action_elements])]
 
         return [ast.Expr(element) for element in action_elements]
@@ -903,10 +904,11 @@ def _gen_loop_mngr(self, action_config, parent_request=None, index_label=""):
                 action_lines = self._gen_action(action)
 
                 transaction = ast.With(
-                    context_expr=ast_call(
-                        func=ast_attr("apiritif.transaction"),
-                        args=[self._gen_expr(label)]),
-                    optional_vars=None,
+                    items=[ast.withitem(
+                        context_expr=ast_call(
+                            func=ast_attr("apiritif.transaction"),
+                            args=[self._gen_expr(label)]),
+                        optional_vars=None)],
                     body=[action_lines])
 
                 body.append([transaction])
@@ -2041,10 +2043,11 @@ def _gen_transaction(self, trans_conf, transaction_class="apiritif.smart_transac
                 body.append(self._gen_http_request(request))
 
         transaction = ast.With(
-            context_expr=ast_call(
-                func=ast_attr(transaction_class),
-                args=[self._gen_expr(trans_conf.label)]),
-            optional_vars=None,
+            items=[ast.withitem(
+                    context_expr=ast_call(
+                        func=ast_attr(transaction_class),
+                        args=[self._gen_expr(trans_conf.label)]),
+                    optional_vars=None)],
             body=body)
 
         return transaction