巧用装饰器，轻轻松松处理异常，让重跑case不再是问题！
2024-02-12 07:33:34
在自动化测试的世界里，异常处理是一个永恒的痛点。面对突如其来的错误，测试脚本往往会戛然中止，留下满屏的报错信息，让人头大不已。为了解决这一顽疾，无数测试工程师绞尽脑汁，想方设法寻找应对之策。
其中，一种广为流传的办法就是编写专门的异常处理代码，在脚本中各个可能出现异常的地方设置try-catch块，对异常进行捕捉和处理。这种方法虽然有效，却也显得笨拙繁琐，不仅会让脚本变得冗长难以维护，而且还容易遗漏异常处理的场景，导致脚本在遇到未知异常时依然束手无策。
为了解决这些痛点，Python社区开发出了一个神奇的工具——装饰器。装饰器是一种特殊类型的函数，它可以对其他函数进行增强和修改，而无需修改原函数的代码。巧妙运用装饰器，我们可以轻松实现异常处理的自动化，让重跑case不再是问题！
装饰器的魔力在于，它可以在函数执行前后执行额外的操作，而无需修改函数本身。例如，我们可以定义一个叫做retry的装饰器，它可以在函数执行过程中自动捕捉和处理异常：
import time
def retry(func):
def wrapper(*args, **kwargs):
max_retries = 3 # 最大重试次数
for i in range(max_retries):
try:
return func(*args, **kwargs)
except Exception as e:
print(f"第{i + 1}次重试失败，错误信息：{str(e)}")
time.sleep(1) # 每次重试前等待1秒
raise e # 超过最大重试次数，抛出异常
return wrapper
有了这个retry装饰器，我们就可以轻松地对任何函数进行异常处理。只需在函数定义前加上@retry装饰器，即可自动实现异常捕捉和重试：
@retry
def test_case():
# 你的测试用例代码
这样一来，当test_case函数执行过程中出现异常时，retry装饰器会自动进行3次重试，并在每次重试失败后打印异常信息和等待1秒。如果重试3次后依然失败，才会抛出异常，让脚本停止执行。
除了自动处理异常，装饰器还可以帮助我们优雅地重跑case。Python unittest框架提供了setUp和tearDown这两个钩子函数，可以在每个测试用例执行前后执行特定的操作。我们可以利用装饰器来自动化这两个钩子函数，实现case的自动重跑：
import unittest
def rerun(func):
def wrapper(*args, **kwargs):
for i in range(2): # 重跑次数
try:
func(*args, **kwargs)
except AssertionError: # 只捕捉断言异常
print(f"第{i + 1}次重跑case失败")
raise # 超过重跑次数，抛出异常
return wrapper
class MyTestCase(unittest.TestCase):
@rerun
def test_case():
# 你的测试用例代码
在这个例子中，我们定义了一个叫做rerun的装饰器，它会在函数执行过程中自动捕捉断言异常（即unittest框架抛出的异常），并在每次捕捉到异常后重跑case一次。如果重跑2次后依然失败，才会抛出异常，让脚本停止执行。
有了这个rerun装饰器，我们就可以轻松地对任何测试case进行重跑，而无需修改原有代码。只需在test_case函数定义前加上@rerun装饰器，即可自动实现case的重跑：
@rerun
def test_case():
# 你的测试用例代码
这样一来，当test_case函数执行过程中出现断言异常时，rerun装饰器会自动进行2次重跑，并在每次重跑失败后打印失败信息。如果重跑2次后依然失败，才会抛出异常，让脚本停止执行。
装饰器作为Python中一种强大的工具，为我们提供了自动化异常处理和重跑case的便捷方式。通过巧妙运用装饰器，我们可以轻松地对脚本进行增强和修改，而无需修改原有代码。
在自动化测试中，装饰器的使用可以极大地提高脚本的健壮性和可维护性。它不仅可以帮助我们自动处理异常，避免脚本因异常而中断执行，还可以优雅地重跑case，保证测试覆盖面和准确性。
如果你还没有尝试过使用装饰器，不妨在你的自动化测试脚本中尝试一下吧。它将为你带来意想不到的惊喜，让你轻松应对异常和重跑case，让自动化测试变得更加轻松高效！
