485 lines
16 KiB
Python
485 lines
16 KiB
Python
# Copyright 2011 Justin Santa Barbara
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
import errno
|
|
import fcntl
|
|
import multiprocessing
|
|
import os
|
|
import shutil
|
|
import sys
|
|
import tempfile
|
|
import threading
|
|
import time
|
|
|
|
from oslo.config import cfg
|
|
from oslotest import base as test_base
|
|
import six
|
|
from six import moves
|
|
|
|
from oslo.concurrency.fixture import lockutils as fixtures
|
|
from oslo.concurrency import lockutils
|
|
from oslo.concurrency.openstack.common.fixture import config
|
|
|
|
|
|
class LockTestCase(test_base.BaseTestCase):
|
|
|
|
def setUp(self):
|
|
super(LockTestCase, self).setUp()
|
|
self.config = self.useFixture(config.Config()).config
|
|
|
|
def test_synchronized_wrapped_function_metadata(self):
|
|
@lockutils.synchronized('whatever', 'test-')
|
|
def foo():
|
|
"""Bar."""
|
|
pass
|
|
|
|
self.assertEqual(foo.__doc__, 'Bar.', "Wrapped function's docstring "
|
|
"got lost")
|
|
self.assertEqual(foo.__name__, 'foo', "Wrapped function's name "
|
|
"got mangled")
|
|
|
|
def test_lock_acquire_release_file_lock(self):
|
|
lock_dir = tempfile.mkdtemp()
|
|
lock_file = os.path.join(lock_dir, 'lock')
|
|
lock = lockutils._FcntlLock(lock_file)
|
|
|
|
def try_lock():
|
|
lock.release() # child co-owns it before fork
|
|
try:
|
|
my_lock = lockutils._FcntlLock(lock_file)
|
|
my_lock.lockfile = open(lock_file, 'w')
|
|
my_lock.trylock()
|
|
my_lock.unlock()
|
|
os._exit(1)
|
|
except IOError:
|
|
os._exit(0)
|
|
|
|
def attempt_acquire(count):
|
|
children = []
|
|
for i in range(count):
|
|
child = multiprocessing.Process(target=try_lock)
|
|
child.start()
|
|
children.append(child)
|
|
exit_codes = []
|
|
for child in children:
|
|
child.join()
|
|
exit_codes.append(child.exitcode)
|
|
return sum(exit_codes)
|
|
|
|
self.assertTrue(lock.acquire())
|
|
try:
|
|
acquired_children = attempt_acquire(10)
|
|
self.assertEqual(0, acquired_children)
|
|
finally:
|
|
lock.release()
|
|
|
|
try:
|
|
acquired_children = attempt_acquire(5)
|
|
self.assertNotEqual(0, acquired_children)
|
|
finally:
|
|
try:
|
|
shutil.rmtree(lock_dir)
|
|
except IOError:
|
|
pass
|
|
|
|
def test_lock_internally(self):
|
|
"""We can lock across multiple threads."""
|
|
saved_sem_num = len(lockutils._semaphores)
|
|
seen_threads = list()
|
|
|
|
def f(_id):
|
|
with lockutils.lock('testlock2', 'test-', external=False):
|
|
for x in range(10):
|
|
seen_threads.append(_id)
|
|
|
|
threads = []
|
|
for i in range(10):
|
|
thread = threading.Thread(target=f, args=(i,))
|
|
threads.append(thread)
|
|
thread.start()
|
|
|
|
for thread in threads:
|
|
thread.join()
|
|
|
|
self.assertEqual(len(seen_threads), 100)
|
|
# Looking at the seen threads, split it into chunks of 10, and verify
|
|
# that the last 9 match the first in each chunk.
|
|
for i in range(10):
|
|
for j in range(9):
|
|
self.assertEqual(seen_threads[i * 10],
|
|
seen_threads[i * 10 + 1 + j])
|
|
|
|
self.assertEqual(saved_sem_num, len(lockutils._semaphores),
|
|
"Semaphore leak detected")
|
|
|
|
def test_nested_synchronized_external_works(self):
|
|
"""We can nest external syncs."""
|
|
tempdir = tempfile.mkdtemp()
|
|
try:
|
|
self.config(lock_path=tempdir)
|
|
sentinel = object()
|
|
|
|
@lockutils.synchronized('testlock1', 'test-', external=True)
|
|
def outer_lock():
|
|
|
|
@lockutils.synchronized('testlock2', 'test-', external=True)
|
|
def inner_lock():
|
|
return sentinel
|
|
return inner_lock()
|
|
|
|
self.assertEqual(sentinel, outer_lock())
|
|
|
|
finally:
|
|
if os.path.exists(tempdir):
|
|
shutil.rmtree(tempdir)
|
|
|
|
def _do_test_lock_externally(self):
|
|
"""We can lock across multiple processes."""
|
|
|
|
def lock_files(handles_dir):
|
|
|
|
with lockutils.lock('external', 'test-', external=True):
|
|
# Open some files we can use for locking
|
|
handles = []
|
|
for n in range(50):
|
|
path = os.path.join(handles_dir, ('file-%s' % n))
|
|
handles.append(open(path, 'w'))
|
|
|
|
# Loop over all the handles and try locking the file
|
|
# without blocking, keep a count of how many files we
|
|
# were able to lock and then unlock. If the lock fails
|
|
# we get an IOError and bail out with bad exit code
|
|
count = 0
|
|
for handle in handles:
|
|
try:
|
|
fcntl.flock(handle, fcntl.LOCK_EX | fcntl.LOCK_NB)
|
|
count += 1
|
|
fcntl.flock(handle, fcntl.LOCK_UN)
|
|
except IOError:
|
|
os._exit(2)
|
|
finally:
|
|
handle.close()
|
|
|
|
# Check if we were able to open all files
|
|
self.assertEqual(50, count)
|
|
|
|
handles_dir = tempfile.mkdtemp()
|
|
try:
|
|
children = []
|
|
for n in range(50):
|
|
pid = os.fork()
|
|
if pid:
|
|
children.append(pid)
|
|
else:
|
|
try:
|
|
lock_files(handles_dir)
|
|
finally:
|
|
os._exit(0)
|
|
|
|
for child in children:
|
|
(pid, status) = os.waitpid(child, 0)
|
|
if pid:
|
|
self.assertEqual(0, status)
|
|
finally:
|
|
if os.path.exists(handles_dir):
|
|
shutil.rmtree(handles_dir, ignore_errors=True)
|
|
|
|
def test_lock_externally(self):
|
|
lock_dir = tempfile.mkdtemp()
|
|
self.config(lock_path=lock_dir)
|
|
|
|
try:
|
|
self._do_test_lock_externally()
|
|
finally:
|
|
if os.path.exists(lock_dir):
|
|
shutil.rmtree(lock_dir, ignore_errors=True)
|
|
|
|
def test_lock_externally_lock_dir_not_exist(self):
|
|
lock_dir = tempfile.mkdtemp()
|
|
os.rmdir(lock_dir)
|
|
self.config(lock_path=lock_dir)
|
|
|
|
try:
|
|
self._do_test_lock_externally()
|
|
finally:
|
|
if os.path.exists(lock_dir):
|
|
shutil.rmtree(lock_dir, ignore_errors=True)
|
|
|
|
def test_synchronized_with_prefix(self):
|
|
lock_name = 'mylock'
|
|
lock_pfix = 'mypfix-'
|
|
|
|
foo = lockutils.synchronized_with_prefix(lock_pfix)
|
|
|
|
@foo(lock_name, external=True)
|
|
def bar(dirpath, pfix, name):
|
|
return True
|
|
|
|
lock_dir = tempfile.mkdtemp()
|
|
self.config(lock_path=lock_dir)
|
|
|
|
self.assertTrue(bar(lock_dir, lock_pfix, lock_name))
|
|
|
|
def test_synchronized_without_prefix(self):
|
|
lock_dir = tempfile.mkdtemp()
|
|
self.config(lock_path=lock_dir)
|
|
|
|
@lockutils.synchronized('lock', external=True)
|
|
def test_without_prefix():
|
|
# We can't check much
|
|
pass
|
|
|
|
try:
|
|
test_without_prefix()
|
|
finally:
|
|
if os.path.exists(lock_dir):
|
|
shutil.rmtree(lock_dir, ignore_errors=True)
|
|
|
|
def test_synchronized_prefix_without_hypen(self):
|
|
lock_dir = tempfile.mkdtemp()
|
|
self.config(lock_path=lock_dir)
|
|
|
|
@lockutils.synchronized('lock', 'hypen', True)
|
|
def test_without_hypen():
|
|
# We can't check much
|
|
pass
|
|
|
|
try:
|
|
test_without_hypen()
|
|
finally:
|
|
if os.path.exists(lock_dir):
|
|
shutil.rmtree(lock_dir, ignore_errors=True)
|
|
|
|
def test_contextlock(self):
|
|
lock_dir = tempfile.mkdtemp()
|
|
self.config(lock_path=lock_dir)
|
|
|
|
try:
|
|
# Note(flaper87): Lock is not external, which means
|
|
# a semaphore will be yielded
|
|
with lockutils.lock("test") as sem:
|
|
if six.PY2:
|
|
self.assertTrue(isinstance(sem, threading._Semaphore))
|
|
else:
|
|
self.assertTrue(isinstance(sem, threading.Semaphore))
|
|
|
|
# NOTE(flaper87): Lock is external so an InterProcessLock
|
|
# will be yielded.
|
|
with lockutils.lock("test2", external=True) as lock:
|
|
self.assertTrue(lock.exists())
|
|
|
|
with lockutils.lock("test1",
|
|
external=True) as lock1:
|
|
self.assertTrue(isinstance(lock1,
|
|
lockutils.InterProcessLock))
|
|
finally:
|
|
if os.path.exists(lock_dir):
|
|
shutil.rmtree(lock_dir, ignore_errors=True)
|
|
|
|
def test_contextlock_unlocks(self):
|
|
lock_dir = tempfile.mkdtemp()
|
|
self.config(lock_path=lock_dir)
|
|
|
|
sem = None
|
|
|
|
try:
|
|
with lockutils.lock("test") as sem:
|
|
if six.PY2:
|
|
self.assertTrue(isinstance(sem, threading._Semaphore))
|
|
else:
|
|
self.assertTrue(isinstance(sem, threading.Semaphore))
|
|
|
|
with lockutils.lock("test2", external=True) as lock:
|
|
self.assertTrue(lock.exists())
|
|
|
|
# NOTE(flaper87): Lock should be free
|
|
with lockutils.lock("test2", external=True) as lock:
|
|
self.assertTrue(lock.exists())
|
|
|
|
# NOTE(flaper87): Lock should be free
|
|
# but semaphore should already exist.
|
|
with lockutils.lock("test") as sem2:
|
|
self.assertEqual(sem, sem2)
|
|
finally:
|
|
if os.path.exists(lock_dir):
|
|
shutil.rmtree(lock_dir, ignore_errors=True)
|
|
|
|
def test_remove_lock_external_file(self):
|
|
lock_name = 'mylock'
|
|
lock_pfix = 'mypfix-remove-lock-test-'
|
|
|
|
lock_dir = tempfile.mkdtemp()
|
|
self.config(lock_path=lock_dir)
|
|
|
|
lockutils.remove_external_lock_file(lock_name, lock_pfix)
|
|
|
|
for ent in os.listdir(lock_dir):
|
|
self.assertRaises(OSError, ent.startswith, lock_pfix)
|
|
|
|
if os.path.exists(lock_dir):
|
|
shutil.rmtree(lock_dir, ignore_errors=True)
|
|
|
|
def test_no_slash_in_b64(self):
|
|
# base64(sha1(foobar)) has a slash in it
|
|
with lockutils.lock("foobar"):
|
|
pass
|
|
|
|
|
|
class BrokenLock(lockutils._FileLock):
|
|
def __init__(self, name, errno_code):
|
|
super(BrokenLock, self).__init__(name)
|
|
self.errno_code = errno_code
|
|
|
|
def unlock(self):
|
|
pass
|
|
|
|
def trylock(self):
|
|
err = IOError()
|
|
err.errno = self.errno_code
|
|
raise err
|
|
|
|
|
|
class FileBasedLockingTestCase(test_base.BaseTestCase):
|
|
def setUp(self):
|
|
super(FileBasedLockingTestCase, self).setUp()
|
|
self.lock_dir = tempfile.mkdtemp()
|
|
|
|
def test_lock_file_exists(self):
|
|
lock_file = os.path.join(self.lock_dir, 'lock-file')
|
|
|
|
@lockutils.synchronized('lock-file', external=True,
|
|
lock_path=self.lock_dir)
|
|
def foo():
|
|
self.assertTrue(os.path.exists(lock_file))
|
|
|
|
foo()
|
|
|
|
def test_bad_acquire(self):
|
|
lock_file = os.path.join(self.lock_dir, 'lock')
|
|
lock = BrokenLock(lock_file, errno.EBUSY)
|
|
|
|
self.assertRaises(threading.ThreadError, lock.acquire)
|
|
|
|
def test_interprocess_lock(self):
|
|
lock_file = os.path.join(self.lock_dir, 'processlock')
|
|
|
|
pid = os.fork()
|
|
if pid:
|
|
# Make sure the child grabs the lock first
|
|
start = time.time()
|
|
while not os.path.exists(lock_file):
|
|
if time.time() - start > 5:
|
|
self.fail('Timed out waiting for child to grab lock')
|
|
time.sleep(0)
|
|
lock1 = lockutils.InterProcessLock('foo')
|
|
lock1.lockfile = open(lock_file, 'w')
|
|
self.assertRaises(IOError, lock1.trylock)
|
|
else:
|
|
try:
|
|
lock2 = lockutils.InterProcessLock('foo')
|
|
lock2.lockfile = open(lock_file, 'w')
|
|
lock2.trylock()
|
|
finally:
|
|
# NOTE(bnemec): This is racy, but I don't want to add any
|
|
# synchronization primitives that might mask a problem
|
|
# with the one we're trying to test here.
|
|
time.sleep(.5)
|
|
os._exit(0)
|
|
|
|
def test_interthread_external_lock(self):
|
|
call_list = []
|
|
|
|
@lockutils.synchronized('foo', external=True, lock_path=self.lock_dir)
|
|
def foo(param):
|
|
"""Simulate a long-running threaded operation."""
|
|
call_list.append(param)
|
|
# NOTE(bnemec): This is racy, but I don't want to add any
|
|
# synchronization primitives that might mask a problem
|
|
# with the one we're trying to test here.
|
|
time.sleep(.5)
|
|
call_list.append(param)
|
|
|
|
def other(param):
|
|
foo(param)
|
|
|
|
thread = threading.Thread(target=other, args=('other',))
|
|
thread.start()
|
|
# Make sure the other thread grabs the lock
|
|
start = time.time()
|
|
while not os.path.exists(os.path.join(self.lock_dir, 'foo')):
|
|
if time.time() - start > 5:
|
|
self.fail('Timed out waiting for thread to grab lock')
|
|
time.sleep(0)
|
|
thread1 = threading.Thread(target=other, args=('main',))
|
|
thread1.start()
|
|
thread1.join()
|
|
thread.join()
|
|
self.assertEqual(call_list, ['other', 'other', 'main', 'main'])
|
|
|
|
|
|
class LockutilsModuleTestCase(test_base.BaseTestCase):
|
|
|
|
def setUp(self):
|
|
super(LockutilsModuleTestCase, self).setUp()
|
|
self.old_env = os.environ.get('OSLO_LOCK_PATH')
|
|
|
|
def tearDown(self):
|
|
if self.old_env is None:
|
|
del os.environ['OSLO_LOCK_PATH']
|
|
else:
|
|
os.environ['OSLO_LOCK_PATH'] = self.old_env
|
|
super(LockutilsModuleTestCase, self).tearDown()
|
|
|
|
def _lock_path_conf_test(self, lock_dir):
|
|
cfg.CONF.unregister_opts(lockutils.util_opts)
|
|
lockutils_ = moves.reload_module(lockutils)
|
|
with lockutils_.lock('test-lock', external=True):
|
|
if not os.path.exists(lock_dir):
|
|
os._exit(2)
|
|
if not os.path.exists(os.path.join(lock_dir, 'test-lock')):
|
|
os._exit(3)
|
|
|
|
def test_main(self):
|
|
script = '\n'.join([
|
|
'import os',
|
|
'lock_path = os.environ.get("OSLO_LOCK_PATH")',
|
|
'assert lock_path is not None',
|
|
'assert os.path.isdir(lock_path)',
|
|
])
|
|
argv = ['', sys.executable, '-c', script]
|
|
retval = lockutils.main(argv)
|
|
self.assertEqual(retval, 0, "Bad OSLO_LOCK_PATH has been set")
|
|
|
|
|
|
class TestLockFixture(test_base.BaseTestCase):
|
|
|
|
def setUp(self):
|
|
super(TestLockFixture, self).setUp()
|
|
self.config = self.useFixture(config.Config()).config
|
|
self.tempdir = tempfile.mkdtemp()
|
|
|
|
def _check_in_lock(self):
|
|
self.assertTrue(self.lock.exists())
|
|
|
|
def tearDown(self):
|
|
self._check_in_lock()
|
|
super(TestLockFixture, self).tearDown()
|
|
|
|
def test_lock_fixture(self):
|
|
# Setup lock fixture to test that teardown is inside the lock
|
|
self.config(lock_path=self.tempdir)
|
|
fixture = fixtures.LockFixture('test-lock')
|
|
self.useFixture(fixture)
|
|
self.lock = fixture.lock
|