27d28b8eef
The default encoding for source files is UTF-8 since Python 3, so we can drop all encoding headers. While we are at it, just replace them with SPDX headers.
44 lines
1.2 KiB
Python
44 lines
1.2 KiB
Python
# SPDX-License-Identifier: MIT
|
|
# Copied from: https://github.com/pycontribs/tendo
|
|
# License: PYTHON SOFTWARE FOUNDATION LICENSE VERSION 2
|
|
# Author: Sorin Sbarnea
|
|
# Changes: changed logging and formatting
|
|
from dosagelib import singleton
|
|
from multiprocessing import Process
|
|
|
|
|
|
def f(flavor_id):
|
|
return singleton.SingleInstance(flavor_id=flavor_id, exit_code=1)
|
|
|
|
|
|
class TestSingleton(object):
|
|
def test_1(self):
|
|
# test in current process
|
|
me = singleton.SingleInstance(flavor_id="test-1")
|
|
del me # now the lock should be removed
|
|
assert True
|
|
|
|
def test_2(self):
|
|
# test in current subprocess
|
|
p = Process(target=f, args=("test-2",))
|
|
p.start()
|
|
p.join()
|
|
# the called function should succeed
|
|
assert p.exitcode == 0
|
|
|
|
def test_3(self):
|
|
# test in current process and subprocess with failure
|
|
# start first instance
|
|
me = f("test-3")
|
|
# second instance
|
|
p = Process(target=f, args=("test-3",))
|
|
p.start()
|
|
p.join()
|
|
assert p.exitcode == 1
|
|
# third instance
|
|
p = Process(target=f, args=("test-3",))
|
|
p.start()
|
|
p.join()
|
|
assert p.exitcode == 1
|
|
del me # now the lock should be removed
|