X-Git-Url: https://osm.etsi.org/gitweb/?a=blobdiff_plain;f=osmclient%2Fcommon%2Ftest%2Ftest_utils.py;h=5ba03f48ebc5116f1cf506570664c210090e6980;hb=cd7cd5eba8cfbb5f1941744ff90cf45e0ebcf422;hp=a14b6363b262b7102c718d5163f8339b7f2278d6;hpb=e84eb31aded0cb62a5523422ddc4524f6e615209;p=osm%2Fosmclient.git diff --git a/osmclient/common/test/test_utils.py b/osmclient/common/test/test_utils.py index a14b636..5ba03f4 100644 --- a/osmclient/common/test/test_utils.py +++ b/osmclient/common/test/test_utils.py @@ -18,43 +18,53 @@ import unittest 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) + + def test_wait_for_method_wait_for_change(self): + def foobar(): + if foobar.counter == 0: + 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") + + assert not utils.wait_for_value( + 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") - 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) - - def test_wait_for_method_wait_for_change(self): - def foobar(): - if foobar.counter == 0: - 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') - assert not utils.wait_for_value(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') - foobar.counter=1 - assert utils.wait_for_value(lambda: foobar(),wait_time=1,catch_exception=Exception) + foobar.counter = 1 + assert utils.wait_for_value( + lambda: foobar(), wait_time=1, catch_exception=Exception + )