mirror of
https://github.com/ceph/ceph
synced 2024-12-27 05:53:20 +00:00
73849c1179
I didn't love the way safe_while was encouraged to be used and it didn't
fit right with the new no-raising behavior. Now it's encouraged to be
used like this:
with safe_while() as proceed:
while proceed():
do_things()
Signed-off-by: Zack Cerza <zack.cerza@inktank.com>
73 lines
2.0 KiB
Python
73 lines
2.0 KiB
Python
from pytest import raises
|
|
from teuthology import contextutil
|
|
from logging import ERROR
|
|
|
|
|
|
class TestSafeWhile(object):
|
|
|
|
def setup(self):
|
|
contextutil.log.setLevel(ERROR)
|
|
self.fake_sleep = lambda s: True
|
|
self.s_while = contextutil.safe_while
|
|
|
|
def test_6_5_10_deal(self):
|
|
with raises(contextutil.MaxWhileTries):
|
|
with self.s_while(_sleeper=self.fake_sleep) as proceed:
|
|
while proceed():
|
|
pass
|
|
|
|
def test_6_0_1_deal(self):
|
|
with raises(contextutil.MaxWhileTries) as error:
|
|
with self.s_while(
|
|
tries=1,
|
|
_sleeper=self.fake_sleep
|
|
) as proceed:
|
|
while proceed():
|
|
pass
|
|
|
|
msg = error.value[0]
|
|
assert 'waiting for 6 seconds' in msg
|
|
|
|
def test_1_0_10_deal(self):
|
|
with raises(contextutil.MaxWhileTries) as error:
|
|
with self.s_while(
|
|
sleep=1,
|
|
_sleeper=self.fake_sleep
|
|
) as proceed:
|
|
while proceed():
|
|
pass
|
|
|
|
msg = error.value[0]
|
|
assert 'waiting for 10 seconds' in msg
|
|
|
|
def test_6_1_10_deal(self):
|
|
with raises(contextutil.MaxWhileTries) as error:
|
|
with self.s_while(
|
|
increment=1,
|
|
_sleeper=self.fake_sleep
|
|
) as proceed:
|
|
while proceed():
|
|
pass
|
|
|
|
msg = error.value[0]
|
|
assert 'waiting for 105 seconds' in msg
|
|
|
|
def test_action(self):
|
|
with raises(contextutil.MaxWhileTries) as error:
|
|
with self.s_while(
|
|
action='doing the thing',
|
|
_sleeper=self.fake_sleep
|
|
) as proceed:
|
|
while proceed():
|
|
pass
|
|
|
|
msg = error.value[0]
|
|
assert "'doing the thing'" in msg
|
|
|
|
def test_no_raise(self):
|
|
with self.s_while(_raise=False, _sleeper=self.fake_sleep) as proceed:
|
|
while proceed():
|
|
pass
|
|
|
|
assert True
|