[checkmk-commits] Added tests for store file locking functions

Module: check_mk Branch: master Commit: 62195f7a4186d34b3be9a5f0c3f8accbb2bb8f25 URL: http://git.mathias-kettner.de/git/?p=check_mk.git;a=commit;h=62195f7a4186d3… Author: Lars Michelsen <lm(a)mathias-kettner.de> Date: Mon Sep 26 12:22:22 2016 +0200 Added tests for store file locking functions --- tests/lib/test_store.py | 143 ++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 143 insertions(+) diff --git a/tests/lib/test_store.py b/tests/lib/test_store.py new file mode 100644 index 0000000..03d658c --- /dev/null +++ b/tests/lib/test_store.py @@ -0,0 +1,143 @@ +import cmk.store as store +import imp +import sys +import threading +import time + + +def test_acquire_lock_not_existing(tmpdir): + store.aquire_lock("%s/asd" % tmpdir) + + +def test_acquire_lock(tmpdir): + locked_file = tmpdir.join("locked_file") + locked_file.write("") + + path = "%s" % locked_file + + assert store.have_lock(path) == False + store.aquire_lock(path) + assert store.have_lock(path) == True + + +def test_acquire_lock_twice(tmpdir): + locked_file = tmpdir.join("locked_file") + locked_file.write("") + + path = "%s" % locked_file + + assert store.have_lock(path) == False + store.aquire_lock(path) + assert store.have_lock(path) == True + store.aquire_lock(path) + assert store.have_lock(path) == True + + +def test_release_lock_not_locked(): + store.release_lock("/asdasd/aasdasd") + + +def test_release_lock(tmpdir): + locked_file = tmpdir.join("locked_file") + locked_file.write("") + + path = "%s" % locked_file + + assert store.have_lock(path) == False + store.aquire_lock(path) + assert store.have_lock(path) == True + store.release_lock(path) + assert store.have_lock(path) == False + + +def test_release_all_locks(tmpdir): + locked_file1 = tmpdir.join("locked_file1") + locked_file1.write("") + locked_file2 = tmpdir.join("locked_file2") + locked_file2.write("") + + path1 = "%s" % locked_file1 + path2 = "%s" % locked_file2 + + assert store.have_lock(path1) == False + store.aquire_lock(path1) + assert store.have_lock(path1) == True + + assert store.have_lock(path2) == False + store.aquire_lock(path2) + assert store.have_lock(path2) == True + + store.release_all_locks() + assert store.have_lock(path1) == False + assert store.have_lock(path2) == False + + + +class LockTestThread(threading.Thread): + def __init__(self, store, path): + self.store = store + self.path = path + self.do = None + self.is_locked = False + super(LockTestThread, self).__init__() + self.daemon = True + + + def run(self): + while True: + if self.do == "lock": + assert self.store.have_lock(self.path) == False + self.store.aquire_lock(self.path) + assert self.store.have_lock(self.path) == True + self.do = None + + elif self.do == "unlock": + assert self.store.have_lock(self.path) == True + self.store.release_lock(self.path) + assert self.store.have_lock(self.path) == False + self.do = None + + else: + time.sleep(0.1) + + + +def test_locking(tmpdir): + store1 = sys.modules["cmk.store"] + del sys.modules["cmk.store"] + import cmk.store as store + store2 = sys.modules["cmk.store"] + + assert store1 != store2 + + locked_file = tmpdir.join("locked_file") + locked_file.write("") + path = "%s" % locked_file + + t1 = LockTestThread(store1, path) + t1.start() + t2 = LockTestThread(store2, path) + t2.start() + + # Take lock with store1 + t1.do = "lock" + while range(20): + if store1.have_lock(path): + break + time.sleep(0.01) + assert store1.have_lock(path) == True + + # Now try to get lock with store2 + t2.do = "lock" + time.sleep(0.2) + assert store1.have_lock(path) == True + assert store2.have_lock(path) == False + + # And now unlock store1 and check whether store2 has the lock now + t1.do = "unlock" + while range(20): + if not store1.have_lock(path): + break + time.sleep(0.01) + assert store1.have_lock(path) == False + assert store2.have_lock(path) == True