diff options
Diffstat (limited to 'tests/zookeeper_test.py')
| -rw-r--r-- | tests/zookeeper_test.py | 8 |
1 files changed, 4 insertions, 4 deletions
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))) |