def test_worker_lock_update_from_lockid_count_for_worker(
self, acquire_before, worker_count_before, worker_count_after):
max_count_before = {}
if worker_count_before:
max_count_before = {'worker1': 5}
max_count_after = {}
if worker_count_after:
max_count_after = {'worker1': 7}
lockid = WorkerLock('lock1', maxCount=3,
maxCountForWorker=max_count_before)
lock = RealWorkerLock(lockid)
if acquire_before:
worker_lock = lock.getLockForWorker('worker1')
self.assertEqual(worker_lock.maxCount,
5 if worker_count_before else 3)
lockid = WorkerLock('lock1', maxCount=4,
maxCountForWorker=max_count_after)
lock.updateFromLockId(lockid)
if not acquire_before:
worker_lock = lock.getLockForWorker('worker1')
self.assertEqual(worker_lock.maxCount,
7 if worker_count_after else 4)
def test_worker_lock_init_from_lockid_count_for_worker(self):
lockid = WorkerLock('lock1', maxCount=3,
maxCountForWorker={'worker2': 5})
lock = RealWorkerLock(lockid)
self.assertEqual(lock.name, 'lock1')
self.assertEqual(lock.maxCount, 3)
worker_lock = lock.getLockForWorker('worker1')
self.assertEqual(worker_lock.maxCount, 3)
worker_lock = lock.getLockForWorker('worker2')
self.assertEqual(worker_lock.maxCount, 5)
def test_worker_lock_init_from_lockid(self):
lockid = WorkerLock('lock1', maxCount=3)
lock = RealWorkerLock(lockid)
self.assertEqual(lock.name, 'lock1')
self.assertEqual(lock.maxCount, 3)
self.assertEqual(lock.description, '<WorkerLock(lock1, 3, {})>')
worker_lock = lock.getLockForWorker('worker1')
self.assertEqual(worker_lock.name, 'lock1')
self.assertEqual(worker_lock.maxCount, 3)
self.assertTrue(worker_lock.description.startswith(
'<WorkerLock(lock1, 3)[worker1]'))
def test_worker_lock_update_from_lockid_count_for_worker(
self, acquire_before, worker_count_before, worker_count_after):
max_count_before = {}
if worker_count_before:
max_count_before = {'worker1': 5}
max_count_after = {}
if worker_count_after:
max_count_after = {'worker1': 7}
lock = RealWorkerLock('lock1')
lock.updateFromLockId(
WorkerLock('lock1', maxCount=3,
maxCountForWorker=max_count_before), 0)
if acquire_before:
worker_lock = lock.getLockForWorker('worker1')
self.assertEqual(worker_lock.maxCount,
5 if worker_count_before else 3)
lockid = WorkerLock('lock1',
maxCount=4,
maxCountForWorker=max_count_after)
lock.updateFromLockId(lockid, 0)
if not acquire_before:
worker_lock = lock.getLockForWorker('worker1')
self.assertEqual(worker_lock.maxCount, 7 if worker_count_after else 4)
def test_worker_lock_update_from_lockid(self):
lock = RealWorkerLock('lock1')
lock.updateFromLockId(WorkerLock('lock1', maxCount=3), 0)
worker_lock = lock.getLockForWorker('worker1')
self.assertEqual(worker_lock.maxCount, 3)
lock.updateFromLockId(WorkerLock('lock1', maxCount=5), 0)
self.assertEqual(lock.lockName, 'lock1')
self.assertEqual(lock.maxCount, 5)
self.assertEqual(lock.description, '<WorkerLock(lock1, 5, {})>')
self.assertEqual(worker_lock.lockName, 'lock1')
self.assertEqual(worker_lock.maxCount, 5)
self.assertTrue(worker_lock.description.startswith(
'<WorkerLock(lock1, 5)[worker1]'))
with self.assertRaises(AssertionError):
lock.updateFromLockId(WorkerLock('lock2', maxCount=4), 0)
def test_worker_lock_update_from_lockid(self):
lockid = WorkerLock('lock1', maxCount=3)
lock = RealWorkerLock(lockid)
worker_lock = lock.getLockForWorker('worker1')
self.assertEqual(worker_lock.maxCount, 3)
lockid = WorkerLock('lock1', maxCount=5)
lock.updateFromLockId(lockid)
self.assertEqual(lock.name, 'lock1')
self.assertEqual(lock.maxCount, 5)
self.assertEqual(lock.description, '<WorkerLock(lock1, 5, {})>')
self.assertEqual(worker_lock.name, 'lock1')
self.assertEqual(worker_lock.maxCount, 5)
self.assertTrue(worker_lock.description.startswith(
'<WorkerLock(lock1, 5)[worker1]'))
with self.assertRaises(AssertionError):
lockid = WorkerLock('lock2', maxCount=4)
lock.updateFromLockId(lockid)