vbatts/quay
Archived
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

util.failover: re-raise exceptions on failure

Browse source
This commit is contained in:
Jimmy Zelinskie 2017-03-01 00:27:58 -05:00
parent 21b09a7451
commit cba7816caf
2 changed files with 19 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

10
util/test/test_failover.py
View file

@ -2,6 +2,9 @@ import pytest
from util.failover import failover, FailoverException
class FinishedException(Exception):
""" Exception raised at the end of every iteration to force failover. """
class Counter(object):
""" Wraps a counter in an object so that it'll be passed by reference. """
@ -18,7 +21,7 @@ def my_failover_func(i, should_raise=None):
i.increment()
if should_raise is not None:
raise should_raise()
raise FailoverException(None, 'incrementing')
raise FailoverException(FinishedException())
@pytest.mark.parametrize('stop_on,exception', [
@ -40,5 +43,6 @@ def test_readonly_failover(stop_on, exception):
with pytest.raises(exception):
my_failover_func(*arg_sets)
else:
my_failover_func(*arg_sets)
assert counter.calls == stop_on
with pytest.raises(FinishedException):
my_failover_func(*arg_sets)
assert counter.calls == stop_on