mirror of
https://github.com/vale981/ray
synced 2025-03-04 17:41:43 -05:00
req
Signed-off-by: Clarence Ng <clarence.wyng@gmail.com>
This commit is contained in:
parent
9f7879758f
commit
f28beca3a9
1 changed files with 0 additions and 39 deletions
|
@ -1,39 +0,0 @@
|
|||
import sys
|
||||
import pytest
|
||||
from retry import retry
|
||||
|
||||
|
||||
retry_once_counter = 0
|
||||
|
||||
|
||||
@retry((Exception), delay=0, max_delay=20, backoff=2, jitter=2, tries=2)
|
||||
def retry_once():
|
||||
global retry_once_counter
|
||||
retry_once_counter += 1
|
||||
raise Exception()
|
||||
|
||||
|
||||
no_retry_counter = 0
|
||||
|
||||
|
||||
@retry((Exception), delay=0, max_delay=20, backoff=2, jitter=2, tries=1)
|
||||
def no_retry():
|
||||
global no_retry_counter
|
||||
no_retry_counter += 1
|
||||
raise ValueError()
|
||||
|
||||
|
||||
def test_retry():
|
||||
with pytest.raises(Exception):
|
||||
retry_once()
|
||||
assert retry_once_counter == 2
|
||||
|
||||
|
||||
def test_no_retry():
|
||||
with pytest.raises(ValueError):
|
||||
no_retry()
|
||||
assert no_retry_counter == 1
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
sys.exit(pytest.main(["-v", __file__]))
|
Loading…
Add table
Reference in a new issue