X-Git-Url: https://wannabe.guru.org/gitweb/?a=blobdiff_plain;f=tests%2Fzookeeper_test.py;fp=tests%2Fzookeeper_test.py;h=05749c169f5cfc9d712f31b6ff1c65ef7ff847c0;hb=f38b4e71714093d2804c18a963f05e85a75a8ffe;hp=6c003147185ab2a0aaa50db9416194a1758e5d38;hpb=a36aa2f6f29ee07198e07b8d2beadc2183d4ea16;p=python_utils.git

diff --git a/tests/zookeeper_test.py b/tests/zookeeper_test.py
index 6c00314..05749c1 100644
--- 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)))