Make release return a bool. Improve tests and logging.

[python_utils.git] / tests / zookeeper_test.py

diff --git a/tests/zookeeper_test.py b/tests/zookeeper_test.py
index 6c003147185ab2a0aaa50db9416194a1758e5d38..05749c169f5cfc9d712f31b6ff1c65ef7ff847c0 100644 (file)
--- a/tests/zookeeper_test.py
+++ b/tests/zookeeper_test.py
@@ -20,9 +20,9 @@ class TestZookeeper(unittest.TestCase):
     @zookeeper.obtain_lease(also_pass_lease=True, duration=datetime.timedelta(minutes=1))
     def test_release_lease(self, lease: zookeeper.RenewableReleasableLease):
         self.assertTrue(lease)
-        lease.release()
+        self.assertTrue(lease.release())
         self.assertFalse(lease)
-        lease.release()
+        self.assertFalse(lease.release())
         self.assertFalse(lease)
 
     @zookeeper.obtain_lease(also_pass_lease=True, duration=datetime.timedelta(minutes=1))
@@ -30,7 +30,7 @@ class TestZookeeper(unittest.TestCase):
         self.assertTrue(lease)
         self.assertTrue(lease.try_renew(datetime.timedelta(minutes=2)))
         self.assertTrue(lease)
-        lease.release()
+        self.assertTrue(lease.release())
 
     @zookeeper.obtain_lease(
         also_pass_lease=True,
@@ -38,7 +38,7 @@ class TestZookeeper(unittest.TestCase):
     )
     def test_cant_renew_lease_after_released(self, lease: zookeeper.RenewableReleasableLease):
         self.assertTrue(lease)
-        lease.release()
+        self.assertTrue(lease.release())
         self.assertFalse(lease)
         self.assertFalse(lease.try_renew(datetime.timedelta(minutes=2)))