diff options
| author | exprmntr <[email protected]> | 2022-02-10 16:46:50 +0300 | 
|---|---|---|
| committer | Daniil Cherednik <[email protected]> | 2022-02-10 16:46:50 +0300 | 
| commit | a6396ea4e82c2605dcf9a11ebc4c289d0b1c734c (patch) | |
| tree | 49e222ea1c5804306084bb3ae065bb702625360f /library/python/filelock/ut/lib | |
| parent | 078ddb2fb76387cae14e8c3ceb5a896009a5a539 (diff) | |
Restoring authorship annotation for exprmntr <[email protected]>. Commit 2 of 2.
Diffstat (limited to 'library/python/filelock/ut/lib')
| -rw-r--r-- | library/python/filelock/ut/lib/test_filelock.py | 76 | ||||
| -rw-r--r-- | library/python/filelock/ut/lib/ya.make | 18 | 
2 files changed, 47 insertions, 47 deletions
diff --git a/library/python/filelock/ut/lib/test_filelock.py b/library/python/filelock/ut/lib/test_filelock.py index b80e3db8c2d..1b11d891231 100644 --- a/library/python/filelock/ut/lib/test_filelock.py +++ b/library/python/filelock/ut/lib/test_filelock.py @@ -2,59 +2,59 @@ import os  import time  import logging  import multiprocessing -import tempfile  +import tempfile  import threading -import library.python.filelock  +import library.python.filelock  def _acquire_lock(lock_path, out_file_path): -    with library.python.filelock.FileLock(lock_path):  +    with library.python.filelock.FileLock(lock_path):          with open(out_file_path, "a") as out:              out.write("{}:{}\n".format(os.getpid(), time.time()))          time.sleep(2)  def test_filelock(): -    temp_dir = tempfile.mkdtemp()  -    lock_path = os.path.join(temp_dir, "file.lock")  -    out_file_path = os.path.join(temp_dir, "out.txt")  - -    process_count = 5  -    processes = []  -    for i in range(process_count):  -        process = multiprocessing.Process(target=_acquire_lock, args=(lock_path, out_file_path))  -        process.start()  -        processes.append(process)  - -    for process in processes:  -        process.join()  - -    pids = []  -    times = []  -    with open(out_file_path) as out:  -        content = out.read()  -        logging.info("Times:\n%s", content)  -        for line in content.strip().split("\n"):  -            pid, time_val = line.split(":")  -            pids.append(pid)  -            times.append(float(time_val))  - -    assert len(set(pids)) == process_count  -    time1 = times.pop()  -    while times:  -        time2 = times.pop()  -        assert int(time1) - int(time2) >= 2  -        time1 = time2  +    temp_dir = tempfile.mkdtemp() +    lock_path = os.path.join(temp_dir, "file.lock") +    out_file_path = os.path.join(temp_dir, "out.txt") + +    process_count = 5 +    processes = [] +    for i in range(process_count): +        process = multiprocessing.Process(target=_acquire_lock, args=(lock_path, out_file_path)) +        process.start() +        processes.append(process) + +    for process in processes: +        process.join() + +    pids = [] +    times = [] +    with open(out_file_path) as out: +        content = out.read() +        logging.info("Times:\n%s", content) +        for line in content.strip().split("\n"): +            pid, time_val = line.split(":") +            pids.append(pid) +            times.append(float(time_val)) + +    assert len(set(pids)) == process_count +    time1 = times.pop() +    while times: +        time2 = times.pop() +        assert int(time1) - int(time2) >= 2 +        time1 = time2  def test_filelock_init_acquired(): -    temp_dir = tempfile.mkdtemp()  -    lock_path = os.path.join(temp_dir, "file.lock")  +    temp_dir = tempfile.mkdtemp() +    lock_path = os.path.join(temp_dir, "file.lock") -    with library.python.filelock.FileLock(lock_path):  -        sublock = library.python.filelock.FileLock(lock_path)  -        del sublock  +    with library.python.filelock.FileLock(lock_path): +        sublock = library.python.filelock.FileLock(lock_path) +        del sublock  def test_concurrent_lock(): diff --git a/library/python/filelock/ut/lib/ya.make b/library/python/filelock/ut/lib/ya.make index 3faf072d912..f3f9da5a67e 100644 --- a/library/python/filelock/ut/lib/ya.make +++ b/library/python/filelock/ut/lib/ya.make @@ -1,11 +1,11 @@ -OWNER(g:yatool)  -  +OWNER(g:yatool) +  PY23_LIBRARY() -  +  TEST_SRCS(test_filelock.py) -  -PEERDIR(  -    library/python/filelock  -)  -  -END()  + +PEERDIR( +    library/python/filelock +) + +END()  | 
