X-Git-Url: https://osm.etsi.org/gitweb/?a=blobdiff_plain;f=osmclient%2Fcommon%2Ftest%2Ftest_utils.py;h=71346629030d4a4b0f3a163d3d4020af69733e51;hb=refs%2Fchanges%2F68%2F9568%2F1;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..7134662 100644 --- a/osmclient/common/test/test_utils.py +++ b/osmclient/common/test/test_utils.py @@ -18,43 +18,50 @@ 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') - foobar.counter=1 - assert utils.wait_for_value(lambda: foobar(),wait_time=1,catch_exception=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)