def test_clear(self):
x = BoundedSemaphore(10)
for i in range(11):
x.acquire(Mock())
self.assertTrue(x._waiting)
self.assertEqual(x.value, 0)
x.clear()
self.assertFalse(x._waiting)
self.assertEqual(x.value, x.initial_value)
def test_acquire_release(self):
x = BoundedSemaphore(2)
c1 = Mock()
x.acquire(c1, 1)
self.assertEqual(x.value, 1)
c1.assert_called_with(1)
c2 = Mock()
x.acquire(c2, 2)
self.assertEqual(x.value, 0)
c2.assert_called_with(2)
c3 = Mock()
x.acquire(c3, 3)
self.assertEqual(x.value, 0)
self.assertFalse(c3.called)
x.release()
self.assertEqual(x.value, 1)
c3.assert_called_with(3)
def test_clear(self):
x = BoundedSemaphore(10)
for i in range(11):
x.acquire(Mock())
self.assertTrue(x._waiting)
self.assertEqual(x.value, 0)
x.clear()
self.assertFalse(x._waiting)
self.assertEqual(x.value, x.initial_value)
def test_acquire_release(self):
x = BoundedSemaphore(2)
c1 = Mock()
x.acquire(c1, 1)
self.assertEqual(x.value, 1)
c1.assert_called_with(1)
c2 = Mock()
x.acquire(c2, 2)
self.assertEqual(x.value, 0)
c2.assert_called_with(2)
c3 = Mock()
x.acquire(c3, 3)
self.assertEqual(x.value, 0)
self.assertFalse(c3.called)
x.release()
self.assertEqual(x.value, 1)
c3.assert_called_with(3)
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_bounded(self):
x = BoundedSemaphore(2)
for i in range(100):
x.release()
self.assertEqual(x.value, 2)
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_bounded(self):
x = BoundedSemaphore(2)
for i in range(100):
x.release()
self.assertEqual(x.value, 2)