def test_force_unlock_in_same_process(self):
pid = os.getpid()
lockfile = '%s.lock' % pid
os.symlink(lockfile, 'test.lock')
with open(lockfile, 'w+') as f:
f.write(str(os.getpid()))
with FileLock('test', force=True):
assert True
def test_process_killed_force_unlock(self):
pid = os.getpid()
lockfile = '%s.lock' % pid
setup_fake_lock('test.lock')
with open(lockfile, 'w+') as f:
f.write('9999999')
assert os.path.exists(lockfile)
with FileLock('test'):
assert True
def test_exception_after_timeout(self):
pid = os.getpid()
lockfile = '%s.lock' % pid
setup_fake_lock('test.lock')
with open(lockfile, 'w+') as f:
f.write(str(os.getpid()))
try:
with FileLock('test', timeout=1):
assert False
except FileLocked:
assert True
def test_force_after_timeout(self):
pid = os.getpid()
lockfile = '%s.lock' % pid
setup_fake_lock('test.lock')
with open(lockfile, 'w+') as f:
f.write(str(os.getpid()))
timeout = 1
start = time.time()
with FileLock('test', timeout=timeout, force=True):
assert True
end = time.time()
assert end - start > timeout
def handle(self, *args, **options):
logger.info('Acquiring lock for sending queued emails at %s.lock' %
options['lockfile'])
try:
with FileLock(options['lockfile']):
while 1:
try:
send_queued(options['processes'],
options.get('log_level'))
except Exception as e:
logger.error(e,
exc_info=sys.exc_info(),
extra={'status_code': 500})
raise
# Close DB connection to avoid multiprocessing errors
connection.close()
if not OutgoingEmail.objects.filter(status=STATUS.queued) \
.filter(Q(scheduled_time__lte=now()) | Q(scheduled_time=None)).exists():
break
except FileLocked:
logger.info('Failed to acquire lock, terminating now.')
def test_get_lock_pid(self):
"""Ensure get_lock_pid() works properly"""
with FileLock('test', timeout=1, force=True) as lock:
self.assertEqual(lock.get_lock_pid(), int(os.getpid()))