Explanation variables only defined if failed or passed with plugins implementing the hook.
This commit is contained in:
		
							parent
							
								
									d91a5d3cd7
								
							
						
					
					
						commit
						9a34d88c8d
					
				|  | @ -803,10 +803,12 @@ class AssertionRewriter(ast.NodeVisitor): | |||
|                     top_condition, module_path=self.module_path, lineno=assert_.lineno | ||||
|                 ) | ||||
|             ) | ||||
|         if self.enable_assertion_pass_hook: | ||||
|             ### Experimental pytest_assertion_pass hook | ||||
| 
 | ||||
|         if self.enable_assertion_pass_hook:  # Experimental pytest_assertion_pass hook | ||||
|             negation = ast.UnaryOp(ast.Not(), top_condition) | ||||
|             msg = self.pop_format_context(ast.Str(explanation)) | ||||
| 
 | ||||
|             # Failed | ||||
|             if assert_.msg: | ||||
|                 assertmsg = self.helper("_format_assertmsg", assert_.msg) | ||||
|                 gluestr = "\n>assert " | ||||
|  | @ -817,10 +819,14 @@ class AssertionRewriter(ast.NodeVisitor): | |||
|             err_msg = ast.BinOp(assertmsg, ast.Add(), err_explanation) | ||||
|             err_name = ast.Name("AssertionError", ast.Load()) | ||||
|             fmt = self.helper("_format_explanation", err_msg) | ||||
|             fmt_pass = self.helper("_format_explanation", msg) | ||||
|             exc = ast.Call(err_name, [fmt], []) | ||||
|             raise_ = ast.Raise(exc, None) | ||||
|             # Call to hook when passes | ||||
|             statements_fail = [] | ||||
|             statements_fail.extend(self.expl_stmts) | ||||
|             statements_fail.append(raise_) | ||||
| 
 | ||||
|             # Passed | ||||
|             fmt_pass = self.helper("_format_explanation", msg) | ||||
|             orig = astor.to_source(assert_.test).rstrip("\n").lstrip("(").rstrip(")") | ||||
|             hook_call_pass = ast.Expr( | ||||
|                 self.helper( | ||||
|  | @ -830,14 +836,16 @@ class AssertionRewriter(ast.NodeVisitor): | |||
|                     fmt_pass, | ||||
|                 ) | ||||
|             ) | ||||
| 
 | ||||
|             # If any hooks implement assert_pass hook | ||||
|             hook_impl_test = ast.If( | ||||
|                 self.helper("_check_if_assertionpass_impl"), [hook_call_pass], [] | ||||
|             ) | ||||
|             main_test = ast.If(negation, [raise_], [hook_impl_test]) | ||||
|             statements_pass = [] | ||||
|             statements_pass.extend(self.expl_stmts) | ||||
|             statements_pass.append(hook_impl_test) | ||||
| 
 | ||||
|             self.statements.extend(self.expl_stmts) | ||||
|             # Test for assertion condition | ||||
|             main_test = ast.If(negation, statements_fail, statements_pass) | ||||
|             self.statements.append(main_test) | ||||
|             if self.format_variables: | ||||
|                 variables = [ | ||||
|  | @ -845,8 +853,8 @@ class AssertionRewriter(ast.NodeVisitor): | |||
|                 ] | ||||
|                 clear_format = ast.Assign(variables, _NameConstant(None)) | ||||
|                 self.statements.append(clear_format) | ||||
|         else: | ||||
|             ### Original assertion rewriting | ||||
| 
 | ||||
|         else:  # Original assertion rewriting | ||||
|             # Create failure message. | ||||
|             body = self.expl_stmts | ||||
|             negation = ast.UnaryOp(ast.Not(), top_condition) | ||||
|  | @ -865,6 +873,7 @@ class AssertionRewriter(ast.NodeVisitor): | |||
|             raise_ = ast.Raise(exc, None) | ||||
| 
 | ||||
|             body.append(raise_) | ||||
| 
 | ||||
|         # Clear temporary variables by setting them to None. | ||||
|         if self.variables: | ||||
|             variables = [ast.Name(name, ast.Store()) for name in self.variables] | ||||
|  |  | |||
		Loading…
	
		Reference in New Issue