don't remove sockets in kvmd.apps.cleanup

This commit is contained in:
Maxim Devaev
2023-03-07 00:55:55 +02:00
parent 49a1beab37
commit 9ecb7fb94d
2 changed files with 6 additions and 40 deletions

View File

@@ -20,7 +20,6 @@
# ========================================================================== #
import os
import signal
import time
@@ -56,20 +55,6 @@ def _kill_streamer(config: Section) -> None:
logger.exception("Can't send SIGKILL to streamer with pid=%d", proc.pid)
def _remove_sockets(config: Section) -> None:
logger = get_logger(0)
for (owner, unix_path) in [
("KVMD", config.server.unix),
("streamer", config.streamer.unix),
]:
if unix_path and os.path.exists(unix_path):
logger.info("Removing %s socket %r ...", owner, unix_path)
try:
os.remove(unix_path)
except Exception: # pragma: nocover
logger.exception("Can't remove %s socket %r", owner, unix_path)
# =====
def main(argv: (list[str] | None)=None) -> None:
config = init(
@@ -82,13 +67,9 @@ def main(argv: (list[str] | None)=None) -> None:
logger = get_logger(0)
logger.info("Cleaning up ...")
for func in [
_kill_streamer,
_remove_sockets,
]:
try:
func(config)
except Exception:
pass
try:
_kill_streamer(config)
except Exception:
pass
logger.info("Bye-bye")

View File

@@ -20,7 +20,6 @@
# ========================================================================== #
import os
import multiprocessing
import time
@@ -32,15 +31,10 @@ from kvmd.apps.cleanup import main
# =====
def test_ok(tmpdir) -> None: # type: ignore
def test_ok() -> None:
_ = Literal # Makes liters happy
queue: "multiprocessing.Queue[Literal[True]]" = multiprocessing.Queue()
ustreamer_sock_path = os.path.abspath(str(tmpdir.join("ustreamer-fake.sock")))
open(ustreamer_sock_path, "w").close() # pylint: disable=consider-using-with
kvmd_sock_path = os.path.abspath(str(tmpdir.join("kvmd-fake.sock")))
open(kvmd_sock_path, "w").close() # pylint: disable=consider-using-with
def ustreamer_fake() -> None:
setproctitle.setproctitle("kvmd/streamer: /usr/bin/ustreamer")
queue.put(True)
@@ -52,16 +46,7 @@ def test_ok(tmpdir) -> None: # type: ignore
assert queue.get(timeout=5)
assert proc.is_alive()
main([
"kvmd-cleanup",
"--set-options",
f"kvmd/server/unix={kvmd_sock_path}",
f"kvmd/streamer/unix={ustreamer_sock_path}",
"--run",
])
assert not os.path.exists(ustreamer_sock_path)
assert not os.path.exists(kvmd_sock_path)
main(["kvmd-cleanup", "--run"])
assert not proc.is_alive()
proc.join()