class TestTaskSemaphore(unittest.TestCase):
def setUp(self):
self.semaphore = TaskSemaphore(1)
def test_should_block_at_max_capacity(self):
self.semaphore.acquire('a', blocking=False)
with self.assertRaises(NoResourcesAvailable):
self.semaphore.acquire('a', blocking=False)
def test_release_capacity(self):
acquire_token = self.semaphore.acquire('a', blocking=False)
self.semaphore.release('a', acquire_token)
try:
self.semaphore.acquire('a', blocking=False)
except NoResourcesAvailable:
self.fail('The release of the semaphore should have allowed for '
'the second acquire to not be blocked')
class TestTaskSemaphore(unittest.TestCase):
def setUp(self):
self.semaphore = TaskSemaphore(1)
def test_should_block_at_max_capacity(self):
self.semaphore.acquire('a', blocking=False)
with self.assertRaises(NoResourcesAvailable):
self.semaphore.acquire('a', blocking=False)
def test_release_capacity(self):
acquire_token = self.semaphore.acquire('a', blocking=False)
self.semaphore.release('a', acquire_token)
try:
self.semaphore.acquire('a', blocking=False)
except NoResourcesAvailable:
self.fail(
'The release of the semaphore should have allowed for '
'the second acquire to not be blocked'
)
