python重试重试，禁用unittest [重复]

Question

编辑：由于 Tenacity 是最近的重试，所以这个问题可以被认为是链接问题的重复，解决方案是升级到 Tenacity。

类似于this question，我想在python中对一个具有重试装饰器的函数进行单元测试：

from retrying import retry # from retrying pypi module

WAIT_EXPONENTIAL_MULTIPLIER = 4 * 1000  # ms
STOP_MAX_ATTEMPT_NUMBER = 5

@retry(
   wait_exponential_multiplier=WAIT_EXPONENTIAL_MULTIPLIER,
   stop_max_attempt_number=STOP_MAX_ATTEMPT_NUMBER
)
def get_from_remote(key):
    raise ValueError() # for example

在我的单元测试中，有时我想调用此函数而不重试，有时使用不同的参数。

我尝试在setUp()/tearDown() 中设置变量，但没有成功。我尝试修补重试装饰器，但它也不起作用。

Answer 1

由于这条线，不确定这是否可能

Retrying(*dargs, **dkw).call(f, *args, **kw)

在retrying。 Retrying 实例是在函数运行时即时创建的，无法再更改其属性。

一种方法是保留未修饰的引用 在运行时功能和装饰老式。

from retrying import retry

WAIT_EXPONENTIAL_MULTIPLIER = 1 * 1000  # ms

# the "normal" decorator
retry_dec = retry(wait_exponential_multiplier=WAIT_EXPONENTIAL_MULTIPLIER)

# the undecorated function
def get_from_remote_undecorated(key):
    raise ValueError()

# this is the "normal" decorated function
get_from_remote = retry_dec(get_from_remote_undecorated)

这可以在单元测试中重复：

# some test
retry_dec_test_1 = retry(wait_exponential_multiplier=WAIT_EXPONENTIAL_MULTIPLIER)
get_from_remote_test_1 = retry_dec_test_1(get_from_remote_undecorated)

# another test with different parameters
retry_dec_test_2 = retry(stop_max_attempt_number=STOP_MAX_ATTEMPT_NUMBER)
get_from_remote_test_2 = retry_dec_test_2 (get_from_remote_undecorated)

但我对这种方法并不满意，我会努力寻找更好的解决方案。