def test_grow_shrink(self):
x = BoundedSemaphore(1)
self.assertEqual(x.initial_value, 1)
cb1 = Mock()
x.acquire(cb1, 1)
cb1.assert_called_with(1)
self.assertEqual(x.value, 0)
cb2 = Mock()
x.acquire(cb2, 2)
self.assertFalse(cb2.called)
self.assertEqual(x.value, 0)
cb3 = Mock()
x.acquire(cb3, 3)
self.assertFalse(cb3.called)
x.grow(2)
cb2.assert_called_with(2)
cb3.assert_called_with(3)
self.assertEqual(x.value, 3)
self.assertEqual(x.initial_value, 3)
self.assertFalse(x._waiting)
x.grow(3)
for i in range(x.initial_value):
self.assertTrue(x.acquire(Mock()))
self.assertFalse(x.acquire(Mock()))
x.clear()
x.shrink(3)
for i in range(x.initial_value):
self.assertTrue(x.acquire(Mock()))
self.assertFalse(x.acquire(Mock()))
self.assertEqual(x.value, 0)
for i in range(100):
x.release()
self.assertEqual(x.value, x.initial_value)
def test_grow_shrink(self):
x = BoundedSemaphore(1)
self.assertEqual(x.initial_value, 1)
cb1 = Mock()
x.acquire(cb1, 1)
cb1.assert_called_with(1)
self.assertEqual(x.value, 0)
cb2 = Mock()
x.acquire(cb2, 2)
self.assertFalse(cb2.called)
self.assertEqual(x.value, 0)
cb3 = Mock()
x.acquire(cb3, 3)
self.assertFalse(cb3.called)
x.grow(2)
cb2.assert_called_with(2)
cb3.assert_called_with(3)
self.assertEqual(x.value, 3)
self.assertEqual(x.initial_value, 3)
self.assertFalse(x._waiting)
x.grow(3)
for i in range(x.initial_value):
self.assertTrue(x.acquire(Mock()))
self.assertFalse(x.acquire(Mock()))
x.clear()
x.shrink(3)
for i in range(x.initial_value):
self.assertTrue(x.acquire(Mock()))
self.assertFalse(x.acquire(Mock()))
self.assertEqual(x.value, 0)
for i in range(100):
x.release()
self.assertEqual(x.value, x.initial_value)
