X-Git-Url: https://osm.etsi.org/gitweb/?a=blobdiff_plain;f=osmclient%2Fcommon%2Ftest%2Ftest_utils.py;fp=osmclient%2Fcommon%2Ftest%2Ftest_utils.py;h=5ba03f48ebc5116f1cf506570664c210090e6980;hb=95686bbc69ded243c346f94dceb0bee567572fb7;hp=71346629030d4a4b0f3a163d3d4020af69733e51;hpb=52424a2ae26db69c5a97d01e84454ffdd4d31228;p=osm%2Fosmclient.git diff --git a/osmclient/common/test/test_utils.py b/osmclient/common/test/test_utils.py index 7134662..5ba03f4 100644 --- a/osmclient/common/test/test_utils.py +++ b/osmclient/common/test/test_utils.py @@ -20,20 +20,22 @@ from osmclient.common import utils class TestUtil(unittest.TestCase): - def test_wait_for_method_basic(self): def foobar(): return True + assert utils.wait_for_value(lambda: foobar()) def test_wait_for_method_timeout(self): def foobar(): return False + assert not utils.wait_for_value(lambda: foobar(), wait_time=0) def test_wait_for_method_paramter(self): def foobar(input): return input + assert not utils.wait_for_value(lambda: foobar(False), wait_time=0) assert utils.wait_for_value(lambda: foobar(True), wait_time=0) @@ -43,25 +45,26 @@ class TestUtil(unittest.TestCase): return True foobar.counter -= 1 return False + foobar.counter = 1 assert utils.wait_for_value(lambda: foobar(), wait_time=1) def test_wait_for_method_exception(self): def foobar(): - raise Exception('send exception') + raise Exception("send exception") + assert not utils.wait_for_value( - lambda: foobar(), - wait_time=0, - catch_exception=Exception) + lambda: foobar(), wait_time=0, catch_exception=Exception + ) def test_wait_for_method_first_exception(self): def foobar(): if foobar.counter == 0: return True foobar.counter -= 1 - raise Exception('send exception') + raise Exception("send exception") + foobar.counter = 1 assert utils.wait_for_value( - lambda: foobar(), - wait_time=1, - catch_exception=Exception) + lambda: foobar(), wait_time=1, catch_exception=Exception + )